package app

import (
	"context"
	"testing"

	"yingce/backend/internal/model"
)

func TestCanRunProviderTaskRequiresVideoConfig(t *testing.T) {
	tests := []struct {
		name string
		task model.Task
		want bool
	}{
		{
			name: "channel config",
			task: model.Task{Type: "video_generate", InputJSON: `{"mode":"video","config":{"model":"veo","channelId":"channel-1"}}`},
			want: true,
		},
		{
			name: "direct config",
			task: model.Task{Type: "video_generate", InputJSON: `{"mode":"video","config":{"model":"veo","baseUrl":"https://example.test","apiKey":"secret"}}`},
			want: true,
		},
		{
			name: "missing model",
			task: model.Task{Type: "video_generate", InputJSON: `{"mode":"video","config":{"channelId":"channel-1"}}`},
			want: false,
		},
		{
			name: "missing direct credentials",
			task: model.Task{Type: "video_generate", InputJSON: `{"mode":"video","config":{"model":"veo","baseUrl":"https://example.test"}}`},
			want: false,
		},
		{
			name: "non video task",
			task: model.Task{Type: "text_generation", InputJSON: `{"mode":"video","config":{"model":"veo","channelId":"channel-1"}}`},
			want: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := canRunProviderTask(tt.task); got != tt.want {
				t.Fatalf("canRunProviderTask() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestProcessTaskRejectsVideoTaskWithoutProviderConfig(t *testing.T) {
	svc := &Service{}
	_, _, err := svc.processTask(context.Background(), model.Task{
		Type:      "video_generate",
		InputJSON: `{"mode":"video","config":{"model":""}}`,
	})
	if err == nil {
		t.Fatal("processTask() error = nil, want missing provider configuration error")
	}
	if err.Error() != "视频任务缺少可执行的模型配置" {
		t.Fatalf("processTask() error = %q, want explicit provider configuration error", err)
	}
}

func TestValidateTaskType(t *testing.T) {
	tests := []struct {
		name     string
		taskType string
		wantErr  bool
	}{
		{name: "missing", taskType: "", wantErr: true},
		{name: "canvas image", taskType: "canvas_image"},
		{name: "video operation", taskType: "video_image_to_video"},
		{name: "unknown canvas type", taskType: "canvas_unknown", wantErr: true},
		{name: "unknown type", taskType: "workflow_router", wantErr: true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := validateTaskType(tt.taskType); (err != nil) != tt.wantErr {
				t.Fatalf("validateTaskType(%q) error = %v, wantErr %v", tt.taskType, err, tt.wantErr)
			}
		})
	}
}

func TestProcessTaskRejectsUnknownType(t *testing.T) {
	_, _, err := (&Service{}).processTask(context.Background(), model.Task{
		Type:      "workflow_router",
		InputJSON: `{}`,
	})
	if err == nil {
		t.Fatal("processTask() error = nil, want unsupported task type error")
	}
	if err.Error() != "不支持的任务类型：workflow_router" {
		t.Fatalf("processTask() error = %q, want unsupported task type error", err)
	}
}