use super::*;
#[test]
fn the_shapes_that_broke_routing_are_distinct() {
let (agy, agy_shape) = provider_for_cli(AgentKind::Antigravity);
let (qwen, qwen_shape) = provider_for_cli(AgentKind::Qwen);
assert_eq!(agy_shape, MeteringShape::PerModelFamily);
assert_eq!(qwen_shape, MeteringShape::AccountPool);
assert_ne!(agy, qwen);
}
#[test]
fn a_spend_budget_is_not_a_time_windowed_pool() {
let (_, shape) = provider_for_cli(AgentKind::OpenCode);
assert_eq!(shape, MeteringShape::SpendBudget);
}
#[test]
fn unobserved_providers_are_unknown_not_invented() {
for cli in [AgentKind::Kilo, AgentKind::MiMoCode, AgentKind::Codebuff, AgentKind::Custom] {
let (provider, shape) = provider_for_cli(cli);
assert!(provider.is_unknown(), "{} must not be given an invented provider", cli.as_str());
assert_eq!(shape, MeteringShape::Unknown);
}
}
#[test]
fn a_known_vendor_may_still_have_unknown_metering() {
let (provider, shape) = provider_for_cli(AgentKind::Grok);
assert_eq!(provider.as_str(), "xai");
assert!(!provider.is_unknown());
assert_eq!(shape, MeteringShape::Unknown);
}
#[test]
fn one_vendor_two_cli_routes_are_two_providers() {
let (agy, _) = provider_for_cli(AgentKind::Antigravity);
let (gemini, _) = provider_for_cli(AgentKind::Gemini);
assert_ne!(agy, gemini);
}