19787a3f51
* Subpackage common -> ariadne * add common * Remove build * only build labrinth * common * set sqlx offline * copy dirs * Fix build
248 lines
9.0 KiB
Rust
248 lines
9.0 KiB
Rust
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::{with_test_environment, TestEnvironment},
|
|
};
|
|
use itertools::Itertools;
|
|
use labrinth::models::teams::ProjectPermissions;
|
|
use labrinth::queue::payouts;
|
|
use rust_decimal::{prelude::ToPrimitive, Decimal};
|
|
|
|
mod common;
|
|
|
|
#[actix_rt::test]
|
|
pub async fn analytics_revenue() {
|
|
with_test_environment(
|
|
None,
|
|
|test_env: TestEnvironment<ApiV3>| 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<Utc>); 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.iter() {
|
|
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.get(&alpha_project_id).unwrap();
|
|
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<i64>, Vec<Decimal>) =
|
|
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.get(&alpha_project_id).unwrap();
|
|
assert_eq!(project_analytics.len(), 9); // and 2 points take place on the same day
|
|
let (sorted_keys, sorted_by_key): (Vec<i64>, Vec<Decimal>) =
|
|
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<Decimal>) -> Vec<f64> {
|
|
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<ApiV3>| 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;
|
|
}
|