package llm

import (
	"testing"

	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model/responses"
)

func TestBuildArkTextFormatJSONObjectDropsSchemaFields(t *testing.T) {
	text := buildArkTextFormat(&ResponseFormat{
		Type:        ResponseFormatTypeJSONObject,
		Name:        "article_outline",
		Description: "outline object",
		SchemaJSON:  []byte(`{"type":"object"}`),
		Strict:      true,
	})

	if text == nil || text.Format == nil {
		t.Fatal("buildArkTextFormat() returned nil")
	}
	if got := text.Format.Type; got != responses.TextType_json_object {
		t.Fatalf("buildArkTextFormat() type = %v, want %v", got, responses.TextType_json_object)
	}
	if text.Format.Name != "" {
		t.Fatalf("buildArkTextFormat() name = %q, want empty for json_object", text.Format.Name)
	}
	if text.Format.Description != nil {
		t.Fatalf("buildArkTextFormat() description = %v, want nil for json_object", *text.Format.Description)
	}
	if text.Format.Schema != nil {
		t.Fatalf("buildArkTextFormat() schema = %v, want nil for json_object", text.Format.Schema)
	}
	if text.Format.Strict != nil {
		t.Fatalf("buildArkTextFormat() strict = %v, want nil for json_object", *text.Format.Strict)
	}
}

func TestBuildArkTextFormatJSONSchemaKeepsSchemaFields(t *testing.T) {
	text := buildArkTextFormat(&ResponseFormat{
		Type:        ResponseFormatTypeJSONSchema,
		Name:        "article_outline",
		Description: "outline object",
		SchemaJSON:  []byte(`{"type":"object"}`),
		Strict:      true,
	})

	if text == nil || text.Format == nil {
		t.Fatal("buildArkTextFormat() returned nil")
	}
	if got := text.Format.Type; got != responses.TextType_json_schema {
		t.Fatalf("buildArkTextFormat() type = %v, want %v", got, responses.TextType_json_schema)
	}
	if text.Format.Name != "article_outline" {
		t.Fatalf("buildArkTextFormat() name = %q, want article_outline", text.Format.Name)
	}
	if text.Format.Description == nil || *text.Format.Description != "outline object" {
		t.Fatalf("buildArkTextFormat() description = %v, want outline object", text.Format.Description)
	}
	if text.Format.Schema == nil || string(text.Format.Schema.Value) != `{"type":"object"}` {
		t.Fatalf("buildArkTextFormat() schema = %v, want schema payload", text.Format.Schema)
	}
	if text.Format.Strict == nil || !*text.Format.Strict {
		t.Fatalf("buildArkTextFormat() strict = %v, want true", text.Format.Strict)
	}
}