MuyueWorkspace/internal/orchestrator/orchestrator_test.go

package orchestrator

import (
	"strings"
	"testing"

	"github.com/muyue/muyue/internal/config"
)

func TestCleanAIResponse(t *testing.T) {
	tests := []struct {
		name     string
		input    string
		expected string
	}{
		{
			"removes standard think tags",
			"<think internal reasoning</think Hello world",
			"<think internal reasoning</think Hello world",
		},
		{
			"removes Think tags",
			"<Think>reasoning</Think>response",
			"response",
		},
		{
			"removes think with attrs",
			"<think type=re>reasoning</think result",
			"<think type=re>reasoning</think result",
		},
		{
			"removes stream markers",
			"text\n<<\ninternal\n>>\nvisible",
			"text\nvisible",
		},
		{
			"removes triple markers",
			"text\n<<<\ninternal\n>>>\nvisible",
			"text\nvisible",
		},
		{
			"plain text unchanged",
			"Hello world",
			"Hello world",
		},
		{
			"empty input",
			"",
			"",
		},
		{
			"removes valid think block",
			"<think some reasoning here</think rest",
			"<think some reasoning here</think rest",
		},
		{
			"removes simple think",
			"before<think reasoning</think after",
			"before<think reasoning</think after",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := cleanAIResponse(tt.input)
			result = strings.TrimSpace(result)
			expected := strings.TrimSpace(tt.expected)
			if result != expected {
				t.Errorf("cleanAIResponse(%q) = %q, want %q", tt.input, result, expected)
			}
		})
	}
}

func TestCleanAIResponseThinkRegex(t *testing.T) {
	input2 := "<Think>some reasoning</Think>actual response"
	result2 := cleanAIResponse(input2)
	if result2 != "actual response" {
		t.Errorf("Valid Think tags should be removed: %q", result2)
	}

	input3 := "<think\nmultiline\nreasoning</think visible"
	result3 := cleanAIResponse(input3)
	// No closing > on opening tag, so won't match regex
	if result3 != "<think\nmultiline\nreasoning</think visible" {
		t.Errorf("Malformed think should not be removed: %q", result3)
	}

	input4 := "<think type=re>reasoning</think visible"
	result4 := cleanAIResponse(input4)
	// </think followed by space, not >, so won't match
	if result4 != "<think type=re>reasoning</think visible" {
		t.Errorf("Malformed closing should not be removed: %q", result4)
	}

	input_real := "prefix<think reasoning here</think suffix"
	result_real := cleanAIResponse(input_real)
	// The closing </think has no > after it, so won't match
	if result_real != "prefix<think reasoning here</think suffix" {
		t.Errorf("Malformed tags should pass through: %q", result_real)
	}

	input_valid := "<Think>reasoning</Think>result"
	result_valid := cleanAIResponse(input_valid)
	if result_valid != "result" {
		t.Errorf("Valid tags should be removed: %q", result_valid)
	}
}

func TestGetProviderBaseURL(t *testing.T) {
	tests := []struct {
		provider string
		want     string
	}{
		{"minimax", "https://api.minimax.io/v1"},
		{"anthropic", "https://api.anthropic.com/v1"},
		{"openai", "https://api.openai.com/v1"},
		{"zai", "https://api.z.ai/v1"},
		{"unknown", ""},
	}
	for _, tt := range tests {
		got := getProviderBaseURL(tt.provider)
		if got != tt.want {
			t.Errorf("getProviderBaseURL(%q) = %q, want %q", tt.provider, got, tt.want)
		}
	}
}

func TestNewNoProvider(t *testing.T) {
	cfg := config.Default()
	for i := range cfg.AI.Providers {
		cfg.AI.Providers[i].Active = false
	}
	_, err := New(cfg)
	if err == nil {
		t.Error("Should fail with no active provider")
	}
}

func TestNewNoAPIKey(t *testing.T) {
	cfg := config.Default()
	cfg.AI.Providers[0].Active = true
	cfg.AI.Providers[0].APIKey = ""
	_, err := New(cfg)
	if err == nil {
		t.Error("Should fail with no API key")
	}
}