package api import ( "encoding/json" "net/http" "net/http/httptest" "os" "testing" "github.com/go-chi/chi/v5" "git.fabledsword.com/bvandeusen/minstrel/internal/auth" ) func newForYouRefreshRouter(h *handlers) chi.Router { r := chi.NewRouter() r.Route("/api", func(api chi.Router) { api.Use(auth.RequireUser(h.pool)) api.Post("/playlists/system/for-you/refresh", h.handleForYouRefresh) }) return r } func TestForYouRefresh_AuthenticatedUser_Returns200(t *testing.T) { if os.Getenv("MINSTREL_TEST_DATABASE_URL") == "" { t.Skip("MINSTREL_TEST_DATABASE_URL not set") } h, pool := testHandlers(t) user := seedUser(t, pool, "foryourefresh", "pw", false) req := httptest.NewRequest(http.MethodPost, "/api/playlists/system/for-you/refresh", nil) req = withUser(req, user) rec := httptest.NewRecorder() newForYouRefreshRouter(h).ServeHTTP(rec, req) if rec.Code != http.StatusOK { t.Fatalf("status = %d, want 200; body=%s", rec.Code, rec.Body.String()) } var resp foryouRefreshResp if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil { t.Fatalf("decode: %v", err) } if resp.TrackIDs == nil { t.Errorf("track_ids = nil, want []string (empty array, not null)") } if int(resp.TrackCount) != len(resp.TrackIDs) { // Acceptable mismatch only when some tracks were removed from // the library between build and response. For a fresh test // user the counts should match. t.Errorf("track_count=%d, len(track_ids)=%d — mismatch", resp.TrackCount, len(resp.TrackIDs)) } } func TestForYouRefresh_NoAuth_Returns401(t *testing.T) { if os.Getenv("MINSTREL_TEST_DATABASE_URL") == "" { t.Skip("MINSTREL_TEST_DATABASE_URL not set") } h, _ := testHandlers(t) req := httptest.NewRequest(http.MethodPost, "/api/playlists/system/for-you/refresh", nil) rec := httptest.NewRecorder() newForYouRefreshRouter(h).ServeHTTP(rec, req) if rec.Code != http.StatusUnauthorized { t.Errorf("status = %d, want 401", rec.Code) } }