neko rss mode

package web

import (
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"adammathes.com/neko/config"
)

func TestRouting(t *testing.T) {
	config.Config.DigestPassword = "secret"
	router := NewRouter(&config.Config)

	tests := []struct {
		name           string
		path           string
		method         string
		cookie         *http.Cookie
		expectedStatus int
		containsBody   string
	}{
		{
			name:           "Root serves new UI",
			path:           "/",
			method:         "GET",
			expectedStatus: http.StatusOK,
			containsBody:   "<!doctype html>", // from React dist/v2
		},
		{
			name:           "/v2/ serves new UI",
			path:           "/v2/",
			method:         "GET",
			expectedStatus: http.StatusOK,
			containsBody:   "<!doctype html>",
		},
		{
			name:           "/v3/ serves v3 UI",
			path:           "/v3/",
			method:         "GET",
			expectedStatus: http.StatusOK,
			containsBody:   "<!doctype html>",
		},
		{
			name:           "/v1/ redirects unauthenticated",
			path:           "/v1/",
			method:         "GET",
			expectedStatus: http.StatusTemporaryRedirect,
		},
		{
			name:           "/v1/ serves legacy UI when authenticated",
			path:           "/v1/",
			method:         "GET",
			cookie:         authCookie(),
			expectedStatus: http.StatusOK,
			containsBody:   "<title>neko rss mode</title>", // from legacy ui.html
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			req := httptest.NewRequest(tt.method, tt.path, nil)
			if tt.cookie != nil {
				req.AddCookie(tt.cookie)
			}
			rr := httptest.NewRecorder()
			router.ServeHTTP(rr, req)

			if rr.Code != tt.expectedStatus {
				t.Errorf("Expected status %d, got %d", tt.expectedStatus, rr.Code)
			}

			if tt.containsBody != "" {
				body := strings.ToLower(rr.Body.String())
				if !strings.Contains(body, strings.ToLower(tt.containsBody)) {
					t.Errorf("Expected body to contain %q, but it didn't", tt.containsBody)
				}
			}
		})
	}
}