package provider

import "testing"

func TestSystemWithDeveloperContextSeparatesStableAndDynamicText(t *testing.T) {
	stable, dynamic, messages := systemWithDeveloperContext(Request{
		System:        "stable instructions",
		SystemContext: "legacy host context",
		Messages: []Message{
			{Role: RoleDeveloper, Content: []Content{TextBlock{Text: "dynamic host context"}}},
			{Role: RoleUser, Content: []Content{TextBlock{Text: "user task"}}},
		},
	})
	if stable != "stable instructions" {
		t.Fatalf("stable = %q", stable)
	}
	if dynamic != "legacy host context\n\ndynamic host context" {
		t.Fatalf("dynamic = %q", dynamic)
	}
	if len(messages) != 1 || messages[0].Role != RoleUser {
		t.Fatalf("messages = %#v, want only the user task", messages)
	}
}

func TestAnthropicDeveloperContextFollowsStableCacheBoundary(t *testing.T) {
	req, err := (&anthropicClient{}).buildRequest(Request{
		Model:  "claude-sonnet-4-5",
		System: "stable instructions",
		Messages: []Message{
			{Role: RoleDeveloper, Content: []Content{TextBlock{Text: "dynamic host context"}}},
			{Role: RoleUser, Content: []Content{TextBlock{Text: "user task"}}},
		},
	})
	if err != nil {
		t.Fatal(err)
	}
	if len(req.System) != 2 || req.System[0].Text != "stable instructions" || req.System[0].CacheControl == nil || req.System[1].Text != "dynamic host context" || req.System[1].CacheControl != nil {
		t.Fatalf("system blocks = %#v", req.System)
	}
}

func TestBedrockDeveloperContextFollowsStableCacheBoundary(t *testing.T) {
	req, err := (&bedrockClient{region: "us-east-1"}).buildRequest(Request{
		Model:  "anthropic.claude-sonnet-4-5-20250929-v1:0",
		System: "stable instructions",
		Messages: []Message{
			{Role: RoleDeveloper, Content: []Content{TextBlock{Text: "dynamic host context"}}},
			{Role: RoleUser, Content: []Content{TextBlock{Text: "user task"}}},
		},
	})
	if err != nil {
		t.Fatal(err)
	}
	if len(req.System) != 3 || req.System[0]["text"] != "stable instructions" || req.System[1]["cachePoint"] == nil || req.System[2]["text"] != "dynamic host context" {
		t.Fatalf("system blocks = %#v", req.System)
	}
}