Rebasing, going about this another way

2026-05-06 07:02:12 -05:00
parent d8d4ac9e5d
commit f4e2475c60
36 changed files with 3040 additions and 466 deletions
@@ -0,0 +1,158 @@
+use super::models::*;
+use crate::settings::ai::BedrockModelConfig;
+
+#[test]
+fn test_cross_region_prefix_us_east() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-3-5-sonnet-20241022-v1:0", "us-east-1"),
+        "us.anthropic.claude-3-5-sonnet-20241022-v1:0"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_eu_west() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-3-5-sonnet-20241022-v1:0", "eu-west-1"),
+        "eu.anthropic.claude-3-5-sonnet-20241022-v1:0"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_ap_northeast_1() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-3-5-sonnet-20241022-v1:0", "ap-northeast-1"),
+        "jp.anthropic.claude-3-5-sonnet-20241022-v1:0"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_ap_southeast_2() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-3-5-sonnet-20241022-v1:0", "ap-southeast-2"),
+        "au.anthropic.claude-3-5-sonnet-20241022-v1:0"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_ap_southeast_1() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-3-5-sonnet-20241022-v1:0", "ap-southeast-1"),
+        "apac.anthropic.claude-3-5-sonnet-20241022-v1:0"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_already_prefixed() {
+    assert_eq!(
+        apply_cross_region_prefix("us.anthropic.claude-sonnet-4-6", "us-east-1"),
+        "us.anthropic.claude-sonnet-4-6"
+    );
+    assert_eq!(
+        apply_cross_region_prefix("global.anthropic.claude-sonnet-4-6", "us-east-1"),
+        "global.anthropic.claude-sonnet-4-6"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_canada_maps_to_us() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-sonnet-4-6", "ca-central-1"),
+        "us.anthropic.claude-sonnet-4-6"
+    );
+}
+
+#[test]
+fn test_cross_region_prefix_unknown_region() {
+    assert_eq!(
+        apply_cross_region_prefix("anthropic.claude-3-5-sonnet-20241022-v1:0", "me-south-1"),
+        "anthropic.claude-3-5-sonnet-20241022-v1:0"
+    );
+}
+
+#[test]
+fn test_get_effective_models_empty_returns_defaults() {
+    let models = get_effective_models(&[]);
+    assert_eq!(models.len(), DEFAULT_BEDROCK_MODELS.len());
+    assert_eq!(models[0].model_id, "anthropic.claude-opus-4-7");
+    assert_eq!(models[0].display_name, "Claude Opus 4.7");
+}
+
+#[test]
+fn test_get_effective_models_custom_overrides() {
+    let custom = vec![BedrockModelConfig {
+        model_id: "custom.model-v1:0".to_string(),
+        display_name: "Custom Model".to_string(),
+        vision_supported: false,
+    }];
+    let models = get_effective_models(&custom);
+    assert_eq!(models.len(), 1);
+    assert_eq!(models[0].model_id, "custom.model-v1:0");
+}
+
+#[test]
+fn test_is_bedrock_model_known_prefix() {
+    assert!(is_bedrock_model("anthropic.claude-sonnet-4-6", &[]));
+    assert!(is_bedrock_model("amazon.nova-pro-v1:0", &[]));
+    assert!(is_bedrock_model("meta.llama3-70b-instruct-v1:0", &[]));
+    assert!(is_bedrock_model("mistral.mistral-large-v1:0", &[]));
+    assert!(is_bedrock_model("deepseek.r1-v1:0", &[]));
+}
+
+#[test]
+fn test_is_bedrock_model_cross_region_prefix() {
+    assert!(is_bedrock_model("us.anthropic.claude-sonnet-4-6", &[]));
+    assert!(is_bedrock_model("eu.anthropic.claude-sonnet-4-6", &[]));
+    assert!(is_bedrock_model("global.anthropic.claude-opus-4-7", &[]));
+}
+
+#[test]
+fn test_is_bedrock_model_unknown() {
+    assert!(!is_bedrock_model("gpt-4o", &[]));
+    assert!(!is_bedrock_model("gemini-pro", &[]));
+}
+
+#[test]
+fn test_is_bedrock_model_custom_config() {
+    let custom = vec![BedrockModelConfig {
+        model_id: "custom.my-model-v1:0".to_string(),
+        display_name: "Custom".to_string(),
+        vision_supported: false,
+    }];
+    assert!(is_bedrock_model("custom.my-model-v1:0", &custom));
+}
+
+#[test]
+fn test_is_bedrock_model_arn() {
+    assert!(is_bedrock_model(
+        "arn:aws:bedrock:us-east-1:156729649053:application-inference-profile/fma5bsw4oxhy",
+        &[],
+    ));
+}
+
+#[test]
+fn test_cross_region_prefix_skips_arn() {
+    let arn =
+        "arn:aws:bedrock:us-east-1:156729649053:application-inference-profile/fma5bsw4oxhy";
+    assert_eq!(apply_cross_region_prefix(arn, "us-east-1"), arn);
+}
+
+#[test]
+fn test_is_bedrock_model_coding_agent_arn() {
+    assert!(is_bedrock_model(
+        "arn:aws:bedrock:us-east-1:156729649053:application-inference-profile/coding-agent-anthropic-claude-opus-4-6-lt2v72",
+        &[],
+    ));
+}
+
+#[test]
+fn test_is_bedrock_model_custom_config_with_arn() {
+    let custom = vec![BedrockModelConfig {
+        model_id: "arn:aws:bedrock:us-east-1:156729649053:application-inference-profile/coding-assistant-inference-profile".to_string(),
+        display_name: "Coding Assistant".to_string(),
+        vision_supported: true,
+    }];
+    assert!(is_bedrock_model(
+        "arn:aws:bedrock:us-east-1:156729649053:application-inference-profile/coding-assistant-inference-profile",
+        &custom,
+    ));
+}