Trending, most watched and anticipated movies and series from Trakt

With an empty search box, Movies and Series show one of Trakt's public lists: trending now, most watched this week, anticipated, popular, and for movies the US box office. [ ] switch lists, r reloads. Each entry is looked up in Radarr (tmdb:) or Sonarr (tvdb:), so it behaves like a search result: details, poster, tick when in the library, add, releases. Needs a Trakt client ID: new optional trakt.client_id setting (Config view section "Trakt", ARR_TUI_TRAKT_CLIENT_ID), masked like the API keys and checked by Test. ARR_TUI_TRAKT_URL points at another server for development. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-23 23:05:35 +02:00
parent b71659b357
commit 4dba794b26
18 changed files with 764 additions and 40 deletions
@@ -0,0 +1,87 @@
+package trakt
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestEntries(t *testing.T) {
+	var got http.Header
+	var gotQuery string
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		got, gotQuery = r.Header, r.URL.RawQuery
+		switch r.URL.Path {
+		case "/movies/trending":
+			w.Write([]byte(`[{"watchers":1234,"movie":{"title":"Dune","year":2021,"ids":{"trakt":1,"slug":"dune-2021","imdb":"tt1160419","tmdb":438631}}},
+				{"watchers":7,"movie":{"title":"No Year","year":null,"ids":{"trakt":2,"slug":"x","imdb":null,"tmdb":null}}}]`))
+		case "/movies/popular":
+			w.Write([]byte(`[{"title":"Heat","year":1995,"ids":{"trakt":3,"slug":"heat-1995","tmdb":949}}]`))
+		case "/shows/watched/weekly":
+			w.Write([]byte(`[{"watcher_count":52000,"play_count":1,"collected_count":2,"collector_count":3,"show":{"title":"Andor","year":2022,"ids":{"trakt":4,"slug":"andor","tvdb":393189,"tmdb":83867}}}]`))
+		case "/movies/boxoffice":
+			w.Write([]byte(`[{"revenue":48464322,"movie":{"title":"Big","year":2026,"ids":{"trakt":5,"tmdb":5}}}]`))
+		default:
+			http.NotFound(w, r)
+		}
+	}))
+	defer srv.Close()
+
+	c := NewClient(" my-id ", "arr-tui/test")
+	c.SetBaseURL(srv.URL)
+	movies, err := c.Entries(context.Background(), MovieLists[0], 20)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if got.Get("trakt-api-key") != "my-id" || got.Get("trakt-api-version") != "2" || got.Get("User-Agent") != "arr-tui/test" ||
+		got.Get("Content-Type") != "application/json" {
+		t.Errorf("headers: %v", got)
+	}
+	if gotQuery != "limit=20&page=1" {
+		t.Errorf("query %q", gotQuery)
+	}
+	if len(movies) != 2 || movies[0].Title != "Dune" || movies[0].IDs.TMDB != 438631 || movies[0].Count != 1234 || movies[1].IDs.TMDB != 0 {
+		t.Errorf("trending: %+v", movies)
+	}
+	if note := MovieLists[0].Note(movies[0].Count); note != "1.2k watching" {
+		t.Errorf("note %q", note)
+	}
+
+	popular, err := c.Entries(context.Background(), MovieLists[3], 20)
+	if err != nil || len(popular) != 1 || popular[0].Title != "Heat" || popular[0].IDs.TMDB != 949 || MovieLists[3].Note(popular[0].Count) != "" {
+		t.Errorf("popular (bare items): %+v %v", popular, err)
+	}
+	shows, err := c.Entries(context.Background(), ShowLists[1], 20)
+	if err != nil || len(shows) != 1 || shows[0].IDs.TVDB != 393189 || shows[0].Count != 52000 || ShowLists[1].Note(52000) != "52k watched it" {
+		t.Errorf("most watched shows: %+v %v", shows, err)
+	}
+	box, err := c.Entries(context.Background(), MovieLists[4], 20)
+	if err != nil || box[0].Count != 48464322 || MovieLists[4].Note(box[0].Count) != "$48M last weekend" {
+		t.Errorf("box office: %+v %v", box, err)
+	}
+}
+
+func TestEntriesErrors(t *testing.T) {
+	if _, err := NewClient("", "ua").Entries(context.Background(), MovieLists[0], 5); !errors.Is(err, ErrNotConfigured) {
+		t.Errorf("no client id: %v", err)
+	}
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusForbidden)
+	}))
+	defer srv.Close()
+	c := NewClient("bad", "ua")
+	c.SetBaseURL(srv.URL)
+	if _, err := c.Entries(context.Background(), MovieLists[0], 5); err == nil || err.Error() != "trakt: the client ID was not accepted (HTTP 403)" {
+		t.Errorf("rejected id: %v", err)
+	}
+}
+
+func TestCompact(t *testing.T) {
+	for n, want := range map[int64]string{0: "0", 950: "950", 1000: "1k", 1234: "1.2k", 34567: "34k", 1500000: "1.5M", 48464322: "48M"} {
+		if got := Compact(n); got != want {
+			t.Errorf("Compact(%d) = %q, want %q", n, got, want)
+		}
+	}
+}