1 files changed, 75 insertions, 0 deletions
diff --git a/web/routing_test.go b/web/routing_test.go
new file mode 100644
index 0000000..972b208
--- /dev/null
+++ b/web/routing_test.go
@@ -0,0 +1,75 @@
+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:           "/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)
+				}
+			}
+		})
+	}
+}