openserp/bing/bing_test.go

package bing

import (
	"net/url"
	"testing"

	"github.com/karust/openserp/core"
)

func TestBuildURL(t *testing.T) {
	tests := []struct {
		name    string
		query   core.Query
		wantErr bool
		check   func(*testing.T, url.Values, string)
	}{
		{
			name: "combined params with unicode and start zero",
			query: core.Query{
				Text:         "golang тест",
				Site:         "example.com",
				Filetype:     "pdf",
				DateInterval: "20240101..20240131",
				LangCode:     "RU",
				Limit:        30,
				Start:        0,
			},
			check: func(t *testing.T, params url.Values, host string) {
				t.Helper()
				if host != "www.bing.com" {
					t.Fatalf("unexpected host: %s", host)
				}
				if got := params.Get("q"); got != "golang тест site:example.com filetype:pdf after:2024-01-01 before:2024-01-31" {
					t.Fatalf("unexpected q value: %q", got)
				}
				if got := params.Get("pq"); got != params.Get("q") {
					t.Fatalf("pq should match q, got %q vs %q", got, params.Get("q"))
				}
				if got := params.Get("setlang"); got != "ru" {
					t.Fatalf("unexpected setlang value: %q", got)
				}
				if got := params.Get("count"); got != "30" {
					t.Fatalf("unexpected count value: %q", got)
				}
				if got := params.Get("first"); got != "" {
					t.Fatalf("first should be omitted when Start=0, got %q", got)
				}
				if got := params.Get("form"); got != "QBLH" {
					t.Fatalf("unexpected form value: %q", got)
				}
				if got := params.Get("qs"); got != "HS" {
					t.Fatalf("unexpected qs value: %q", got)
				}
				if got := params.Get("sp"); got != "-1" {
					t.Fatalf("unexpected sp value: %q", got)
				}
			},
		},
		{
			name: "very large start",
			query: core.Query{
				Text:  "golang",
				Start: 2147483647,
				Limit: 20,
			},
			check: func(t *testing.T, params url.Values, _ string) {
				t.Helper()
				if got := params.Get("first"); got != "2147483648" {
					t.Fatalf("unexpected first value: %q", got)
				}
				if got := params.Get("count"); got != "" {
					t.Fatalf("count should be omitted when first is used, got %q", got)
				}
			},
		},
		{
			name: "negative start returns error",
			query: core.Query{
				Text:  "golang",
				Start: -1,
			},
			wantErr: true,
		},
		{
			name:    "empty fields return error",
			query:   core.Query{},
			wantErr: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := BuildURL(tt.query)
			if (err != nil) != tt.wantErr {
				t.Fatalf("BuildURL() error = %v, wantErr %v", err, tt.wantErr)
			}
			if tt.wantErr {
				return
			}

			parsed, err := url.Parse(got)
			if err != nil {
				t.Fatalf("BuildURL() returned invalid URL: %v", err)
			}
			if tt.check != nil {
				tt.check(t, parsed.Query(), parsed.Host)
			}
		})
	}
}

func TestBuildImageURL(t *testing.T) {
	tests := []struct {
		name    string
		query   core.Query
		wantErr bool
		check   func(*testing.T, url.Values, string)
	}{
		{
			name:  "basic image query",
			query: core.Query{Text: "test"},
			check: func(t *testing.T, params url.Values, host string) {
				t.Helper()
				if host != "www.bing.com" {
					t.Fatalf("unexpected host: %s", host)
				}
				if got := params.Get("q"); got != "test" {
					t.Fatalf("unexpected q value: %q", got)
				}
			},
		},
		{
			name: "combined params with unicode",
			query: core.Query{
				Text:     "коты",
				Site:     "example.com",
				Filetype: "png",
				LangCode: "EN",
			},
			check: func(t *testing.T, params url.Values, _ string) {
				t.Helper()
				if got := params.Get("q"); got != "коты site:example.com" {
					t.Fatalf("unexpected q value: %q", got)
				}
				if got := params.Get("setlang"); got != "en" {
					t.Fatalf("unexpected setlang value: %q", got)
				}
				if got := params.Get("form"); got != "HDRSC2" {
					t.Fatalf("unexpected form value: %q", got)
				}
				if got := params.Get("first"); got != "1" {
					t.Fatalf("unexpected first value: %q", got)
				}
				if got := params.Get("scenario"); got != "ImageBasicHover" {
					t.Fatalf("unexpected scenario value: %q", got)
				}
			},
		},
		{
			name:    "empty fields return error",
			query:   core.Query{},
			wantErr: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := BuildImageURL(tt.query)
			if (err != nil) != tt.wantErr {
				t.Fatalf("BuildImageURL() error = %v, wantErr %v", err, tt.wantErr)
			}
			if tt.wantErr {
				return
			}

			parsed, err := url.Parse(got)
			if err != nil {
				t.Fatalf("BuildImageURL() returned invalid URL: %v", err)
			}
			if tt.check != nil {
				tt.check(t, parsed.Query(), parsed.Host)
			}
		})
	}
}