use serde_json::json;

use crate::integration::IntegrationTest;

fn assert_evaluated_plans(prom: &str, expected: u64) {
    let line = prom
        .lines()
        .find(|line| line.starts_with("apollo_router_query_planning_plan_evaluated_plans_sum"))
        .expect("apollo.router.query_planning.plan.evaluated_plans metric is missing");
    let (_, num) = line
        .split_once(' ')
        .expect("apollo.router.query_planning.plan.evaluated_plans metric has unexpected shape");
    assert_eq!(num, format!("{expected}"));
}

#[tokio::test(flavor = "multi_thread")]
async fn reports_evaluated_plans() {
    let mut router = IntegrationTest::builder()
        .config(
            r#"
            telemetry:
              exporters:
                metrics:
                  prometheus:
                    enabled: true
        "#,
        )
        .supergraph("tests/integration/fixtures/query_planner_max_evaluated_plans.graphql")
        .build()
        .await;
    router.start().await;
    router.assert_started().await;
    router
        .execute_query(&json!({
            "query": r#"{ t { v1 v2 v3 v4 } }"#,
            "variables": {},
        }))
        .await;

    let metrics = router
        .get_metrics_response()
        .await
        .expect("failed to fetch metrics")
        .text()
        .await
        .expect("metrics are not text?!");
    assert_evaluated_plans(&metrics, 16);

    router.graceful_shutdown().await;
}

#[tokio::test(flavor = "multi_thread")]
async fn does_not_exceed_max_evaluated_plans_legacy() {
    let mut router = IntegrationTest::builder()
        .config(
            r#"
            experimental_query_planner_mode: legacy
            telemetry:
              exporters:
                metrics:
                  prometheus:
                    enabled: true
            supergraph:
              query_planning:
                experimental_plans_limit: 4
        "#,
        )
        .supergraph("tests/integration/fixtures/query_planner_max_evaluated_plans.graphql")
        .build()
        .await;
    router.start().await;
    router.assert_started().await;
    router
        .execute_query(&json!({
            "query": r#"{ t { v1 v2 v3 v4 } }"#,
            "variables": {},
        }))
        .await;

    let metrics = router
        .get_metrics_response()
        .await
        .expect("failed to fetch metrics")
        .text()
        .await
        .expect("metrics are not text?!");
    assert_evaluated_plans(&metrics, 4);

    router.graceful_shutdown().await;
}

#[tokio::test(flavor = "multi_thread")]
async fn does_not_exceed_max_evaluated_plans() {
    let mut router = IntegrationTest::builder()
        .config(
            r#"
            experimental_query_planner_mode: new
            telemetry:
              exporters:
                metrics:
                  prometheus:
                    enabled: true
            supergraph:
              query_planning:
                experimental_plans_limit: 4
        "#,
        )
        .supergraph("tests/integration/fixtures/query_planner_max_evaluated_plans.graphql")
        .build()
        .await;
    router.start().await;
    router.assert_started().await;
    router
        .execute_query(&json!({
            "query": r#"{ t { v1 v2 v3 v4 } }"#,
            "variables": {},
        }))
        .await;

    let metrics = router
        .get_metrics_response()
        .await
        .expect("failed to fetch metrics")
        .text()
        .await
        .expect("metrics are not text?!");
    assert_evaluated_plans(&metrics, 4);

    router.graceful_shutdown().await;
}