重組前檢查點:根目錄 main package

notfound_test.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/go-chi/chi/v5"
+)
+
+func TestNotFoundHandlerRendersPage(t *testing.T) {
+	rec := httptest.NewRecorder()
+	notFoundHandler(rec, httptest.NewRequest(http.MethodGet, "/no-such-page", nil))
+	if rec.Code != http.StatusNotFound {
+		t.Fatalf("status = %d, want 404", rec.Code)
+	}
+	if ct := rec.Header().Get("Content-Type"); !strings.Contains(ct, "text/html") {
+		t.Fatalf("Content-Type = %q, want text/html", ct)
+	}
+	if csp := rec.Header().Get("Content-Security-Policy"); !strings.Contains(csp, "default-src 'none'") || !strings.Contains(csp, "style-src 'self'") {
+		t.Fatalf("Content-Security-Policy = %q, 應停用外部資源載入且樣式僅允許本站", csp)
+	}
+	body := rec.Body.String()
+	for _, want := range []string{"404", "找不到頁面", `href="/login"`, `/static/css/main.css`} {
+		if !strings.Contains(body, want) {
+			t.Errorf("404 頁缺少 %s", want)
+		}
+	}
+	if strings.Contains(body, "<aside") {
+		t.Fatal("404 頁應為獨立版面，不含側邊導覽欄")
+	}
+}
+
+func TestRouterNotFoundUsesCustomPage(t *testing.T) {
+	// chi 的 NotFound 不分方法：GET 與 POST 到未匹配路徑都應輸出自訂頁。
+	r := chi.NewRouter()
+	r.Get("/login", func(w http.ResponseWriter, r *http.Request) {})
+	r.NotFound(notFoundHandler)
+
+	for _, method := range []string{http.MethodGet, http.MethodPost} {
+		rec := httptest.NewRecorder()
+		r.ServeHTTP(rec, httptest.NewRequest(method, "/definitely-not-a-route", nil))
+		if rec.Code != http.StatusNotFound {
+			t.Fatalf("%s status = %d, want 404", method, rec.Code)
+		}
+		if !strings.Contains(rec.Body.String(), "找不到頁面") {
+			t.Fatalf("%s 應輸出自訂 404 頁，body = %s", method, rec.Body.String())
+		}
+	}
+}