meutel/oauth2_proxy
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

made gcp healthcheck test better

Browse Source
This commit is contained in:
timothy-spencer 2019-03-25 10:03:22 -07:00
parent 189bda3781
commit e2755624ec
No known key found for this signature in database
GPG Key ID: B9DC267D2CB410CD
1 changed files with 3 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
http_test.go
View File

@ -40,32 +40,17 @@ func TestGCPHealthcheckReadiness(t *testing.T) {
assert.Equal(t, "OK", rw.Body.String()) assert.Equal(t, "OK", rw.Body.String())
} }
func TestGCPHealthcheckNotReadiness(t *testing.T) { func TestGCPHealthcheckNotHealthcheck(t *testing.T) {
handler := func(w http.ResponseWriter, req *http.Request) { handler := func(w http.ResponseWriter, req *http.Request) {
w.Write([]byte("test")) w.Write([]byte("test"))
} }
h := gcpHealthcheck(http.HandlerFunc(handler)) h := gcpHealthcheck(http.HandlerFunc(handler))
rw := httptest.NewRecorder() rw := httptest.NewRecorder()
r, _ := http.NewRequest("GET", "/NOT_readiness_check", nil) r, _ := http.NewRequest("GET", "/NOT_any_check", nil)
r.RemoteAddr = "127.0.0.1" r.RemoteAddr = "127.0.0.1"
r.Host = "test-server" r.Host = "test-server"
h.ServeHTTP(rw, r) h.ServeHTTP(rw, r)
assert.NotEqual(t, "OK", rw.Body.String()) assert.Equal(t, "test", rw.Body.String())
}
func TestGCPHealthcheckNotLiveness(t *testing.T) {
handler := func(w http.ResponseWriter, req *http.Request) {
w.Write([]byte("test"))
}
h := gcpHealthcheck(http.HandlerFunc(handler))
rw := httptest.NewRecorder()
r, _ := http.NewRequest("GET", "/NOT_liveness_check", nil)
r.RemoteAddr = "127.0.0.1"
r.Host = "test-server"
h.ServeHTTP(rw, r)
assert.NotEqual(t, "OK", rw.Body.String())
} }