mirror of
https://github.com/stefanprodan/podinfo.git
synced 2026-05-23 18:02:46 +00:00
24 lines
502 B
Go
24 lines
502 B
Go
package http
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestVersionMiddleware(t *testing.T) {
|
|
srv := NewMockServer()
|
|
handler := versionMiddleware(http.HandlerFunc(srv.infoHandler))
|
|
|
|
req, _ := http.NewRequest("GET", "/", nil)
|
|
rr := httptest.NewRecorder()
|
|
handler.ServeHTTP(rr, req)
|
|
|
|
if req.Header.Get("X-API-Version") == "" {
|
|
t.Error("X-API-Version not set by middleware")
|
|
}
|
|
if req.Header.Get("X-API-Revision") == "" {
|
|
t.Error("X-API-Revision not set by middleware")
|
|
}
|
|
}
|