snapshot of https://github.com/go-chi/chi.git (master, 2026-09-15; blobs > 131072B omitted)

middleware/sunset_test.go

@@ -0,0 +1,104 @@
+package middleware
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	"github.com/go-chi/chi/v5"
+)
+
+func TestSunset(t *testing.T) {
+
+	t.Run("Sunset without link", func(t *testing.T) {
+		req, _ := http.NewRequest("GET", "/", nil)
+		w := httptest.NewRecorder()
+
+		r := chi.NewRouter()
+
+		sunsetAt := time.Date(2025, 12, 24, 10, 20, 0, 0, time.UTC)
+		r.Use(Sunset(sunsetAt))
+
+		var sunset, deprecation string
+		r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+			clonedHeader := w.Header().Clone()
+			sunset = clonedHeader.Get("Sunset")
+			deprecation = clonedHeader.Get("Deprecation")
+			w.Write([]byte("I'll be unavailable soon"))
+		})
+		r.ServeHTTP(w, req)
+
+		if w.Code != 200 {
+			t.Fatal("Response Code should be 200")
+		}
+
+		if sunset != "Wed, 24 Dec 2025 10:20:00 GMT" {
+			t.Fatal("Test get sunset error.", sunset)
+		}
+
+		if deprecation != "Wed, 24 Dec 2025 10:20:00 GMT" {
+			t.Fatal("Test get deprecation error.")
+		}
+	})
+
+	t.Run("Sunset with link", func(t *testing.T) {
+		req, _ := http.NewRequest("GET", "/", nil)
+		w := httptest.NewRecorder()
+
+		r := chi.NewRouter()
+
+		sunsetAt := time.Date(2025, 12, 24, 10, 20, 0, 0, time.UTC)
+		deprecationLink := "https://example.com/v1/deprecation-details"
+		r.Use(Sunset(sunsetAt, deprecationLink))
+
+		var sunset, deprecation, link string
+		r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+			clonedHeader := w.Header().Clone()
+			sunset = clonedHeader.Get("Sunset")
+			deprecation = clonedHeader.Get("Deprecation")
+			link = clonedHeader.Get("Link")
+
+			w.Write([]byte("I'll be unavailable soon"))
+		})
+
+		r.ServeHTTP(w, req)
+
+		if w.Code != 200 {
+			t.Fatal("Response Code should be 200")
+		}
+
+		if sunset != "Wed, 24 Dec 2025 10:20:00 GMT" {
+			t.Fatal("Test get sunset error.", sunset)
+		}
+
+		if deprecation != "Wed, 24 Dec 2025 10:20:00 GMT" {
+			t.Fatal("Test get deprecation error.")
+		}
+
+		if link != deprecationLink {
+			t.Fatal("Test get deprecation link error.")
+		}
+	})
+
+}
+
+/**
+EXAMPLE USAGES
+func main() {
+	r := chi.NewRouter()
+
+	sunsetAt := time.Date(2025, 12, 24, 10, 20, 0, 0, time.UTC)
+	r.Use(middleware.Sunset(sunsetAt))
+
+	// can provide additional link for updated resource
+	// r.Use(middleware.Sunset(sunsetAt, "https://example.com/v1/deprecation-details"))
+
+	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+		w.Write([]byte("This endpoint will be removed soon"))
+	})
+
+	log.Println("Listening on port: 3000")
+	http.ListenAndServe(":3000", r)
+}
+**/