fix: read imported usage aggregates in dashboards

This commit is contained in:
fawney19
2026-05-25 00:51:46 +08:00
parent 40b4e52508
commit d4ae6e0e64
5 changed files with 1343 additions and 84 deletions
@@ -436,6 +436,30 @@ fn usage_sql_summarize_usage_daily_heatmap_supports_daily_aggregates() {
);
}
#[test]
fn usage_sql_daily_cutoff_falls_back_to_imported_stats_daily() {
let source = include_str!("mod.rs");
assert!(source.contains("FROM stats_summary"));
assert!(source.contains("SELECT MAX(date) AS latest_date"));
assert!(source.contains("FROM stats_daily"));
assert!(source.contains("FROM stats_user_daily"));
assert!(source.contains("FROM stats_daily_api_key"));
assert!(source.contains("value + chrono::Duration::days(1)"));
}
#[test]
fn usage_sql_dashboard_daily_breakdown_falls_back_to_daily_totals() {
let source = include_str!("mod.rs");
assert!(source.contains("list_dashboard_daily_breakdown_aggregate_segments"));
assert!(source.contains("list_dashboard_daily_breakdown_from_daily_totals"));
assert!(source.contains("'aggregate'::TEXT AS model"));
assert!(source.contains("FROM stats_daily"));
assert!(source.contains("FROM stats_user_daily"));
assert!(source.contains("detailed_dates.contains(&item.date)"));
assert!(source.contains("query.tz_offset_minutes != 0"));
assert!(source.contains("aggregate_dates.insert(item.date.clone())"));
}
#[test]
fn usage_sql_summarize_usage_leaderboard_supports_daily_aggregates() {
let source = include_str!("mod.rs");
@@ -533,6 +557,9 @@ fn usage_sql_summarize_usage_totals_by_user_ids_supports_user_summary_aggregates
let source = include_str!("mod.rs");
assert!(source.contains("FROM stats_user_summary"));
assert!(source.contains("all_time_input_tokens"));
assert!(source.contains("FROM stats_user_daily"));
assert!(source.contains("date < $2"));
assert!(source.contains("summary_user_ids.contains(&user_id)"));
}
#[test]