use crate::harness::attributes_to_map;use crate::harness::build_metrics_with_defaults;use crate::harness::find_metric;use crate::harness::latest_metrics;use codex_otel::PLUGIN_INSTALL_ELICITATION_SENT_METRIC;use codex_otel::PLUGIN_INSTALL_SUGGESTION_METRIC;use codex_otel::Result;use codex_otel::SessionTelemetry;use codex_otel::TelemetryAuthMode;use codex_protocol::ThreadId;use codex_protocol::protocol::SessionSource;use opentelemetry_sdk::metrics::data::AggregatedMetrics;use opentelemetry_sdk::metrics::data::MetricData;use pretty_assertions::assert_eq;use std::collections::BTreeMap;// Ensures SessionTelemetry attaches metadata tags when forwarding metrics.#[test]fn manager_attaches_metadata_tags_to_metrics() -> Result<()> {    let (metrics, exporter) = build_metrics_with_defaults(&[("service", "codex-cli")])?;    let manager = SessionTelemetry::new(        ThreadId::new(),        "gpt-5.1",        "gpt-5.1",        Some("account-id".to_string()),        /*account_email*/ None,        Some(TelemetryAuthMode::ApiKey),        "test_originator".to_string(),        /*log_user_prompts*/ true,        "tty".to_string(),        SessionSource::Cli,    )    .with_metrics(metrics);    manager.counter(        "codex.session_started",        /*inc*/ 1,        &[("source", "tui")],    );    manager.shutdown_metrics()?;    let resource_metrics = latest_metrics(&exporter);    let metric =        find_metric(&resource_metrics, "codex.session_started").expect("counter metric missing");    let attrs = match metric.data() {        AggregatedMetrics::U64(data) => match data {            MetricData::Sum(sum) => {                let points: Vec<_> = sum.data_points().collect();                assert_eq!(points.len(), 1);                attributes_to_map(points[0].attributes())            }            _ => panic!("unexpected counter aggregation"),        },        _ => panic!("unexpected counter data type"),    };    let expected = BTreeMap::from([        (            "app.version".to_string(),            env!("CARGO_PKG_VERSION").to_string(),        ),        (            "auth_mode".to_string(),            TelemetryAuthMode::ApiKey.to_string(),        ),        ("model".to_string(), "gpt-5.1".to_string()),        ("originator".to_string(), "test_originator".to_string()),        ("service".to_string(), "codex-cli".to_string()),        ("session_source".to_string(), "cli".to_string()),        ("source".to_string(), "tui".to_string()),    ]);    assert_eq!(attrs, expected);    Ok(())}// Ensures metadata tagging can be disabled when recording via SessionTelemetry.#[test]fn manager_allows_disabling_metadata_tags() -> Result<()> {    let (metrics, exporter) = build_metrics_with_defaults(&[])?;    let manager = SessionTelemetry::new(        ThreadId::new(),        "gpt-4o",        "gpt-4o",        Some("account-id".to_string()),        /*account_email*/ None,        Some(TelemetryAuthMode::ApiKey),        "test_originator".to_string(),        /*log_user_prompts*/ true,        "tty".to_string(),        SessionSource::Cli,    )    .with_metrics_without_metadata_tags(metrics);    manager.counter(        "codex.session_started",        /*inc*/ 1,        &[("source", "tui")],    );    manager.shutdown_metrics()?;    let resource_metrics = latest_metrics(&exporter);    let metric =        find_metric(&resource_metrics, "codex.session_started").expect("counter metric missing");    let attrs = match metric.data() {        AggregatedMetrics::U64(data) => match data {            MetricData::Sum(sum) => {                let points: Vec<_> = sum.data_points().collect();                assert_eq!(points.len(), 1);                attributes_to_map(points[0].attributes())            }            _ => panic!("unexpected counter aggregation"),        },        _ => panic!("unexpected counter data type"),    };    let expected = BTreeMap::from([("source".to_string(), "tui".to_string())]);    assert_eq!(attrs, expected);    Ok(())}#[test]fn manager_attaches_optional_service_name_tag() -> Result<()> {    let (metrics, exporter) = build_metrics_with_defaults(&[])?;    let manager = SessionTelemetry::new(        ThreadId::new(),        "gpt-5.1",        "gpt-5.1",        /*account_id*/ None,        /*account_email*/ None,        /*auth_mode*/ None,        "test_originator".to_string(),        /*log_user_prompts*/ false,        "tty".to_string(),        SessionSource::Cli,    )    .with_metrics_service_name("my_app_server_client")    .with_metrics(metrics);    manager.counter("codex.session_started", /*inc*/ 1, &[]);    manager.shutdown_metrics()?;    let resource_metrics = latest_metrics(&exporter);    let metric =        find_metric(&resource_metrics, "codex.session_started").expect("counter metric missing");    let attrs = match metric.data() {        AggregatedMetrics::U64(data) => match data {            MetricData::Sum(sum) => {                let points: Vec<_> = sum.data_points().collect();                assert_eq!(points.len(), 1);                attributes_to_map(points[0].attributes())            }            _ => panic!("unexpected counter aggregation"),        },        _ => panic!("unexpected counter data type"),    };    assert_eq!(        attrs.get("service_name"),        Some(&"my_app_server_client".to_string())    );    Ok(())}#[test]fn manager_records_plugin_install_suggestion_metric() -> Result<()> {    let (metrics, exporter) = build_metrics_with_defaults(&[])?;    let manager = SessionTelemetry::new(        ThreadId::new(),        "gpt-5.1",        "gpt-5.1",        Some("account-id".to_string()),        /*account_email*/ None,        Some(TelemetryAuthMode::ApiKey),        "test_originator".to_string(),        /*log_user_prompts*/ false,        "tty".to_string(),        SessionSource::Cli,    )    .with_metrics_without_metadata_tags(metrics);    manager.record_plugin_install_suggestion(        "connector",        "connector_calendar",        "Google Calendar",        "accept",        /*user_confirmed*/ true,        /*completed*/ false,    );    manager.shutdown_metrics()?;    let resource_metrics = latest_metrics(&exporter);    let metric = find_metric(&resource_metrics, PLUGIN_INSTALL_SUGGESTION_METRIC)        .expect("plugin install suggestion metric missing");    let attrs = match metric.data() {        AggregatedMetrics::U64(data) => match data {            MetricData::Sum(sum) => {                let points: Vec<_> = sum.data_points().collect();                assert_eq!(points.len(), 1);                attributes_to_map(points[0].attributes())            }            _ => panic!("unexpected counter aggregation"),        },        _ => panic!("unexpected counter data type"),    };    assert_eq!(        attrs,        BTreeMap::from([            ("completed".to_string(), "false".to_string()),            ("response_action".to_string(), "accept".to_string()),            ("tool_type".to_string(), "connector".to_string()),        ])    );    Ok(())}#[test]fn manager_records_plugin_install_elicitation_sent_metric() -> Result<()> {    let (metrics, exporter) = build_metrics_with_defaults(&[])?;    let manager = SessionTelemetry::new(        ThreadId::new(),        "gpt-5.1",        "gpt-5.1",        Some("account-id".to_string()),        /*account_email*/ None,        Some(TelemetryAuthMode::ApiKey),        "test_originator".to_string(),        /*log_user_prompts*/ false,        "tty".to_string(),        SessionSource::Cli,    )    .with_metrics_without_metadata_tags(metrics);    manager.record_plugin_install_elicitation_sent("plugin", "slack@openai-curated", "Slack");    manager.shutdown_metrics()?;    let resource_metrics = latest_metrics(&exporter);    let metric = find_metric(&resource_metrics, PLUGIN_INSTALL_ELICITATION_SENT_METRIC)        .expect("plugin install elicitation sent metric missing");    let attrs = match metric.data() {        AggregatedMetrics::U64(data) => match data {            MetricData::Sum(sum) => {                let points: Vec<_> = sum.data_points().collect();                assert_eq!(points.len(), 1);                attributes_to_map(points[0].attributes())            }            _ => panic!("unexpected counter aggregation"),        },        _ => panic!("unexpected counter data type"),    };    assert_eq!(        attrs,        BTreeMap::from([("tool_type".to_string(), "plugin".to_string())])    );    Ok(())}