diff --git a/core/checker.go b/core/checker.go index 3f03776c..2ec46931 100644 --- a/core/checker.go +++ b/core/checker.go @@ -39,6 +39,7 @@ func (s *Service) Check() *Service { Timeout: 30 * time.Second, } response, err := client.Get(s.Domain) + response.Header.Set("User-Agent", "StatupMonitor") t2 := time.Now() s.Latency = t2.Sub(t1).Seconds() if err != nil { diff --git a/main_test.go b/main_test.go index 05019d3a..bd3f2b56 100644 --- a/main_test.go +++ b/main_test.go @@ -292,7 +292,7 @@ func TestFailingPrometheusHandler(t *testing.T) { assert.Nil(t, err) rr := httptest.NewRecorder() route.ServeHTTP(rr, req) - assert.Equal(t, 55, rr.Result().StatusCode) + assert.Equal(t, 401, rr.Result().StatusCode) } func TestLoginHandler(t *testing.T) {