inital commit

search_test.go

@@ -0,0 +1,120 @@
+package fa
+
+import (
+	"net/url"
+	"strings"
+	"testing"
+	"time"
+)
+
+// TestBuildSearchURL_DefaultsSendAllRatingsAndTypes asserts that an empty
+// SearchOptions encodes every rating and every type as enabled, matching
+// FA's web-form default state.
+func TestBuildSearchURL_DefaultsSendAllRatingsAndTypes(t *testing.T) {
+	got := buildSearchURL("dragon", 1, SearchOptions{})
+	u, err := url.Parse(got)
+	if err != nil {
+		t.Fatalf("parse: %v", err)
+	}
+	q := u.Query()
+	if q.Get("q") != "dragon" {
+		t.Errorf("q = %q", q.Get("q"))
+	}
+	for _, k := range []string{"rating-general", "rating-mature", "rating-adult"} {
+		if q.Get(k) != "1" {
+			t.Errorf("%s = %q; want 1", k, q.Get(k))
+		}
+	}
+	for _, k := range []string{"type-art", "type-music", "type-flash", "type-story", "type-photo", "type-poetry"} {
+		if q.Get(k) != "1" {
+			t.Errorf("%s = %q; want 1", k, q.Get(k))
+		}
+	}
+	if q.Get("page") != "" {
+		t.Errorf("page should be unset on page 1, got %q", q.Get("page"))
+	}
+}
+
+func TestBuildSearchURL_RatingsSubsetEncodesOnlyRequested(t *testing.T) {
+	got := buildSearchURL("x", 1, SearchOptions{Ratings: []Rating{RatingGeneral}})
+	q := mustQuery(t, got)
+	if q.Get("rating-general") != "1" {
+		t.Errorf("rating-general missing")
+	}
+	if q.Get("rating-mature") != "" || q.Get("rating-adult") != "" {
+		t.Errorf("non-requested ratings should be unset; got mature=%q adult=%q",
+			q.Get("rating-mature"), q.Get("rating-adult"))
+	}
+}
+
+func TestBuildSearchURL_TypesSubsetEncodesOnlyRequested(t *testing.T) {
+	got := buildSearchURL("x", 1, SearchOptions{Types: []SearchType{SearchTypeArt, SearchTypeStory}})
+	q := mustQuery(t, got)
+	for _, k := range []string{"type-art", "type-story"} {
+		if q.Get(k) != "1" {
+			t.Errorf("%s should be set", k)
+		}
+	}
+	for _, k := range []string{"type-music", "type-flash", "type-photo", "type-poetry"} {
+		if q.Get(k) != "" {
+			t.Errorf("%s should be unset; got %q", k, q.Get(k))
+		}
+	}
+}
+
+func TestBuildSearchURL_ManualRangeFormatsDates(t *testing.T) {
+	from := time.Date(2024, 1, 15, 0, 0, 0, 0, time.UTC)
+	to := time.Date(2024, 6, 30, 0, 0, 0, 0, time.UTC)
+	got := buildSearchURL("x", 1, SearchOptions{
+		Range:     SearchRangeManual,
+		RangeFrom: from,
+		RangeTo:   to,
+	})
+	q := mustQuery(t, got)
+	if q.Get("range") != "manual" {
+		t.Errorf("range = %q", q.Get("range"))
+	}
+	if q.Get("range_from") != "2024-01-15" {
+		t.Errorf("range_from = %q", q.Get("range_from"))
+	}
+	if q.Get("range_to") != "2024-06-30" {
+		t.Errorf("range_to = %q", q.Get("range_to"))
+	}
+}
+
+func TestBuildSearchURL_OrderAndPage(t *testing.T) {
+	got := buildSearchURL("x", 3, SearchOptions{
+		OrderBy:  SearchOrderDate,
+		OrderAsc: true,
+		PerPage:  48,
+	})
+	q := mustQuery(t, got)
+	if q.Get("page") != "3" {
+		t.Errorf("page = %q", q.Get("page"))
+	}
+	if q.Get("order-by") != "date" {
+		t.Errorf("order-by = %q", q.Get("order-by"))
+	}
+	if q.Get("order-direction") != "asc" {
+		t.Errorf("order-direction = %q", q.Get("order-direction"))
+	}
+	if q.Get("perpage") != "48" {
+		t.Errorf("perpage = %q", q.Get("perpage"))
+	}
+}
+
+func TestBuildSearchURL_RootPath(t *testing.T) {
+	got := buildSearchURL("x", 1, SearchOptions{})
+	if !strings.HasPrefix(got, "https://www.furaffinity.net/search/?") {
+		t.Errorf("URL prefix wrong: %s", got)
+	}
+}
+
+func mustQuery(t *testing.T, raw string) url.Values {
+	t.Helper()
+	u, err := url.Parse(raw)
+	if err != nil {
+		t.Fatalf("parse %q: %v", raw, err)
+	}
+	return u.Query()
+}