From d44f58f0e24b886d9c26c7baab6340c0b4e0867d Mon Sep 17 00:00:00 2001 From: timothy-spencer Date: Mon, 25 Mar 2019 10:47:30 -0700 Subject: [PATCH] found another edge case to test --- http_test.go | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/http_test.go b/http_test.go index 2bc3d5d..b365d6f 100644 --- a/http_test.go +++ b/http_test.go @@ -87,3 +87,19 @@ func TestGCPHealthcheckNotIngress(t *testing.T) { assert.Equal(t, "test", rw.Body.String()) } + +func TestGCPHealthcheckNotIngressPut(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("PUT", "/", nil) + r.RemoteAddr = "127.0.0.1" + r.Host = "test-server" + r.Header.Set(userAgentHeader, googleHealthCheckUserAgent) + h.ServeHTTP(rw, r) + + assert.Equal(t, "test", rw.Body.String()) +}