package websearch

import (
	"strings"
	"testing"
)

func TestParseDuckDuckGoResults(t *testing.T) {
	body := `
<a rel="nofollow" class="result__a" href="//duckduckgo.com/l/?uddg=https%3A%2F%2Fexample.com%2Ffashion&amp;rut=abc">Fashion Website Design</a>
<a class="result__snippet" href="//duckduckgo.com/l/?uddg=https%3A%2F%2Fexample.com%2Ffashion&amp;rut=abc">A <b>fashion</b> ecommerce homepage reference.</a>
<a rel="nofollow" class="result__a" href="//duckduckgo.com/l/?uddg=https%3A%2F%2Fexample.org%2Fhoodie&amp;rut=def">Hoodie Brand System</a>
<a class="result__snippet" href="//duckduckgo.com/l/?uddg=https%3A%2F%2Fexample.org%2Fhoodie&amp;rut=def">Brand and hoodie visual language.</a>`

	results := parseDuckDuckGoResults(body, 5)
	if len(results) != 2 {
		t.Fatalf("expected 2 results, got %d", len(results))
	}
	if results[0].Title != "Fashion Website Design" {
		t.Fatalf("unexpected title: %s", results[0].Title)
	}
	if results[0].URL != "https://example.com/fashion" {
		t.Fatalf("unexpected url: %s", results[0].URL)
	}
	if results[0].Snippet != "A fashion ecommerce homepage reference." {
		t.Fatalf("unexpected snippet: %s", results[0].Snippet)
	}
}

func TestBuildVisualSearchQueryForLogoReferencesAvoidsMakerTools(t *testing.T) {
	query := BuildVisualSearchQuery("Moteva 让设计变简单懂你的设计 Agent，帮我搞定一切设计需求，制作一个 logo")

	for _, want := range []string{"AI design agent", "logo 设计稿", "设计思路", "brand identity", "-maker", "-generator", "-template"} {
		if !strings.Contains(query, want) {
			t.Fatalf("expected query to contain %q, got %q", want, query)
		}
	}
	if strings.Contains(query, "制作一个 logo") {
		t.Fatalf("expected query to avoid raw prompt wording, got %q", query)
	}
}