72c9f2ee75
Some checks failed
Test / test (1.24.x, ubuntu-latest) (push) Successful in 2m56s
Test / test (1.25.x, ubuntu-latest) (push) Successful in 2m15s
Test / test (1.26.x, ubuntu-latest) (push) Successful in 2m23s
Test / test (1.27.x, ubuntu-latest) (push) Failing after 3m56s
Test / test (1.24.x, windows-latest) (push) Has been cancelled
Test / test (1.25.x, windows-latest) (push) Has been cancelled
Test / test (1.26.x, windows-latest) (push) Has been cancelled
Test / test (1.27.x, windows-latest) (push) Has been cancelled
105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
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)
|
|
}
|
|
**/
|