ds2api/internal/assistantturn/turn_test.go

package assistantturn

import (
	"testing"

	"ds2api/internal/promptcompat"
	"ds2api/internal/sse"
)

func TestBuildTurnFromCollectedTextCitation(t *testing.T) {
	turn := BuildTurnFromCollected(sse.CollectResult{
		Text:          "See [citation:1]",
		CitationLinks: map[int]string{1: "https://example.com"},
	}, BuildOptions{Model: "deepseek-v4-flash", Prompt: "prompt", SearchEnabled: true, StripReferenceMarkers: true})
	if turn.Text != "See [1](https://example.com)" {
		t.Fatalf("text mismatch: %q", turn.Text)
	}
	if turn.StopReason != StopReasonStop {
		t.Fatalf("stop reason mismatch: %q", turn.StopReason)
	}
	if turn.Error != nil {
		t.Fatalf("unexpected error: %#v", turn.Error)
	}
}

func TestBuildTurnFromCollectedToolCall(t *testing.T) {
	turn := BuildTurnFromCollected(sse.CollectResult{
		Text: `<tool_calls><invoke name="Write"><parameter name="content">{"x":1}</parameter></invoke></tool_calls>`,
	}, BuildOptions{
		ToolNames: []string{"Write"},
		ToolsRaw: []any{map[string]any{
			"name": "Write",
			"input_schema": map[string]any{
				"type": "object",
				"properties": map[string]any{
					"content": map[string]any{"type": "string"},
				},
			},
		}},
	})
	if len(turn.ToolCalls) != 1 {
		t.Fatalf("expected one tool call, got %d", len(turn.ToolCalls))
	}
	if turn.StopReason != StopReasonToolCalls {
		t.Fatalf("stop reason mismatch: %q", turn.StopReason)
	}
	if _, ok := turn.ToolCalls[0].Input["content"].(string); !ok {
		t.Fatalf("expected content coerced to string, got %#v", turn.ToolCalls[0].Input["content"])
	}
}

func TestBuildTurnFromCollectedThinkingOnlyIsEmptyOutput(t *testing.T) {
	turn := BuildTurnFromCollected(sse.CollectResult{Thinking: "hidden"}, BuildOptions{})
	if turn.Error == nil || turn.Error.Code != "upstream_empty_output" {
		t.Fatalf("expected empty output error, got %#v", turn.Error)
	}
}

func TestBuildTurnFromCollectedToolChoiceRequired(t *testing.T) {
	turn := BuildTurnFromCollected(sse.CollectResult{Text: "hello"}, BuildOptions{
		ToolChoice: promptcompat.ToolChoicePolicy{Mode: promptcompat.ToolChoiceRequired},
	})
	if turn.Error == nil || turn.Error.Code != "tool_choice_violation" {
		t.Fatalf("expected tool choice violation, got %#v", turn.Error)
	}
}

func TestBuildTurnFromStreamSnapshotUsesVisibleTextAndRawToolDetection(t *testing.T) {
	turn := BuildTurnFromStreamSnapshot(StreamSnapshot{
		RawText:     `<tool_calls><invoke name="Write"><parameter name="content">{"x":1}</parameter></invoke></tool_calls>`,
		VisibleText: "",
	}, BuildOptions{
		ToolNames: []string{"Write"},
		ToolsRaw: []any{map[string]any{
			"name": "Write",
			"schema": map[string]any{
				"type": "object",
				"properties": map[string]any{
					"content": map[string]any{"type": "string"},
				},
			},
		}},
	})
	if len(turn.ToolCalls) != 1 {
		t.Fatalf("expected stream snapshot tool call, got %d", len(turn.ToolCalls))
	}
	if _, ok := turn.ToolCalls[0].Input["content"].(string); !ok {
		t.Fatalf("expected stream snapshot schema coercion, got %#v", turn.ToolCalls[0].Input["content"])
	}
}

func TestBuildTurnFromStreamSnapshotAlreadyEmittedToolAvoidsEmptyError(t *testing.T) {
	turn := BuildTurnFromStreamSnapshot(StreamSnapshot{AlreadyEmittedCalls: true}, BuildOptions{})
	if turn.Error != nil {
		t.Fatalf("unexpected empty-output error after emitted tool call: %#v", turn.Error)
	}
	if turn.StopReason != StopReasonToolCalls {
		t.Fatalf("stop reason mismatch: %q", turn.StopReason)
	}
}