use ariadne::ids::base62_impl::parse_base62; use chrono::{DateTime, Duration, Utc}; use common::permissions::PermissionsTest; use common::permissions::PermissionsTestContext; use common::{ api_v3::ApiV3, database::*, environment::{TestEnvironment, with_test_environment}, }; use itertools::Itertools; use labrinth::models::teams::ProjectPermissions; use labrinth::queue::payouts; use rust_decimal::{Decimal, prelude::ToPrimitive}; pub mod common; #[actix_rt::test] pub async fn analytics_revenue() { with_test_environment( None, |test_env: TestEnvironment| async move { let api = &test_env.api; let alpha_project_id = test_env.dummy.project_alpha.project_id.clone(); let pool = test_env.db.pool.clone(); // Generate sample revenue data- directly insert into sql let ( mut insert_user_ids, mut insert_project_ids, mut insert_payouts, mut insert_starts, mut insert_availables, ) = (Vec::new(), Vec::new(), Vec::new(), Vec::new(), Vec::new()); // Note: these go from most recent to least recent let money_time_pairs: [(f64, DateTime); 10] = [ (50.0, Utc::now() - Duration::minutes(5)), (50.1, Utc::now() - Duration::minutes(10)), (101.0, Utc::now() - Duration::days(1)), (200.0, Utc::now() - Duration::days(2)), (311.0, Utc::now() - Duration::days(3)), (400.0, Utc::now() - Duration::days(4)), (526.0, Utc::now() - Duration::days(5)), (633.0, Utc::now() - Duration::days(6)), (800.0, Utc::now() - Duration::days(14)), (800.0, Utc::now() - Duration::days(800)), ]; let project_id = parse_base62(&alpha_project_id).unwrap() as i64; for (money, time) in &money_time_pairs { insert_user_ids.push(USER_USER_ID_PARSED); insert_project_ids.push(project_id); insert_payouts.push(Decimal::from_f64_retain(*money).unwrap()); insert_starts.push(*time); insert_availables.push(*time); } let mut transaction = pool.begin().await.unwrap(); payouts::insert_payouts( insert_user_ids, insert_project_ids, insert_payouts, insert_starts, insert_availables, &mut transaction, ) .await .unwrap(); transaction.commit().await.unwrap(); let day = 86400; // Test analytics endpoint with default values // - all time points in the last 2 weeks // - 1 day resolution let analytics = api .get_analytics_revenue_deserialized( vec![&alpha_project_id], false, None, None, None, USER_USER_PAT, ) .await; assert_eq!(analytics.len(), 1); // 1 project let project_analytics = &analytics[&alpha_project_id]; assert_eq!(project_analytics.len(), 8); // 1 days cut off, and 2 points take place on the same day. note that the day exactly 14 days ago is included // sorted_by_key, values in the order of smallest to largest key let (sorted_keys, sorted_by_key): (Vec, Vec) = project_analytics .iter() .sorted_by_key(|(k, _)| *k) .rev() .unzip(); assert_eq!( vec![100.1, 101.0, 200.0, 311.0, 400.0, 526.0, 633.0, 800.0], to_f64_vec_rounded_up(sorted_by_key) ); // Ensure that the keys are in multiples of 1 day for k in sorted_keys { assert_eq!(k % day, 0); } // Test analytics with last 900 days to include all data // keep resolution at default let analytics = api .get_analytics_revenue_deserialized( vec![&alpha_project_id], false, Some(Utc::now() - Duration::days(801)), None, None, USER_USER_PAT, ) .await; let project_analytics = &analytics[&alpha_project_id]; assert_eq!(project_analytics.len(), 9); // and 2 points take place on the same day let (sorted_keys, sorted_by_key): (Vec, Vec) = project_analytics .iter() .sorted_by_key(|(k, _)| *k) .rev() .unzip(); assert_eq!( vec![ 100.1, 101.0, 200.0, 311.0, 400.0, 526.0, 633.0, 800.0, 800.0 ], to_f64_vec_rounded_up(sorted_by_key) ); for k in sorted_keys { assert_eq!(k % day, 0); } }, ) .await; } fn to_f64_rounded_up(d: Decimal) -> f64 { d.round_dp_with_strategy( 1, rust_decimal::RoundingStrategy::MidpointAwayFromZero, ) .to_f64() .unwrap() } fn to_f64_vec_rounded_up(d: Vec) -> Vec { d.into_iter().map(to_f64_rounded_up).collect_vec() } #[actix_rt::test] pub async fn permissions_analytics_revenue() { with_test_environment( None, |test_env: TestEnvironment| async move { let alpha_project_id = test_env.dummy.project_alpha.project_id.clone(); let alpha_version_id = test_env.dummy.project_alpha.version_id.clone(); let alpha_team_id = test_env.dummy.project_alpha.team_id.clone(); let api = &test_env.api; let view_analytics = ProjectPermissions::VIEW_ANALYTICS; // first, do check with a project let req_gen = |ctx: PermissionsTestContext| async move { let project_id = ctx.project_id.unwrap(); let ids_or_slugs = vec![project_id.as_str()]; api.get_analytics_revenue( ids_or_slugs, false, None, None, Some(5), ctx.test_pat.as_deref(), ) .await }; PermissionsTest::new(&test_env) .with_failure_codes(vec![200, 401]) .with_200_json_checks( // On failure, should have 0 projects returned |value: &serde_json::Value| { let value = value.as_object().unwrap(); assert_eq!(value.len(), 0); }, // On success, should have 1 project returned |value: &serde_json::Value| { let value = value.as_object().unwrap(); assert_eq!(value.len(), 1); }, ) .simple_project_permissions_test(view_analytics, req_gen) .await .unwrap(); // Now with a version // Need to use alpha let req_gen = |ctx: PermissionsTestContext| { let alpha_version_id = alpha_version_id.clone(); async move { let ids_or_slugs = vec![alpha_version_id.as_str()]; api.get_analytics_revenue( ids_or_slugs, true, None, None, Some(5), ctx.test_pat.as_deref(), ) .await } }; PermissionsTest::new(&test_env) .with_failure_codes(vec![200, 401]) .with_existing_project(&alpha_project_id, &alpha_team_id) .with_user(FRIEND_USER_ID, FRIEND_USER_PAT, true) .with_200_json_checks( // On failure, should have 0 versions returned |value: &serde_json::Value| { let value = value.as_object().unwrap(); assert_eq!(value.len(), 0); }, // On success, should have 1 versions returned |value: &serde_json::Value| { let value = value.as_object().unwrap(); assert_eq!(value.len(), 0); }, ) .simple_project_permissions_test(view_analytics, req_gen) .await .unwrap(); // Cleanup test db test_env.cleanup().await; }, ) .await; }