func TestAddEndpoint_ValidEndpoint(t *testing.T) {
router := NewRouter()
err := router.AddEndpoint("the-service", "http://localhost/metrics", nil)
if err != nil {
t.Error(helper.AssertError(nil, err))
}
}
func TestRoute_ValidService(t *testing.T) {
router := NewRouter()
router.AddEndpoint("the-service", "http://localhost:8080/metrics", nil)
req, _ := http.NewRequest("GET", "http://localhost/metrics?service=the-service", nil)
endpoint, err := router.Route(req)
if err != nil {
t.Error(helper.AssertError(nil, err))
}
computed := endpoint.URL.String()
expected := "http://localhost:8080/metrics"
if computed != expected {
t.Error(helper.AssertError(expected, computed))
}
}
func TestAddEndpoint_InvalidEndpoint(t *testing.T) {
// All endpoint URLs must be absolute
router := NewRouter()
err := router.AddEndpoint("the-service", "localhost/metrics", nil)
expectedErr := InvalidURLErr{"url must be absolute"}
if err != expectedErr {
t.Error(helper.AssertError(expectedErr, err))
}
}
func TestRoute_UnknownService(t *testing.T) {
router := NewRouter()
router.AddEndpoint("the-service", "http://localhost:8080/metrics", nil)
req, _ := http.NewRequest("GET", "http://localhost/metrics?service=the-other-service", nil)
_, err := router.Route(req)
expectedErr := UnknownService{}
if err != expectedErr {
t.Error(helper.AssertError(expectedErr, err))
}
}
// Testing
func TestServeHTTP_GetUnknownService(t *testing.T) {
// supresses loging.
log.SetOutput(ioutil.Discard)
scraper := NewHTTPScraper()
proxy := NewPromProxy(scraper, NewRouter(), nil, nil)
req, _ := http.NewRequest("GET", "http://fakeproxy?service=unknown", nil)
w := httptest.NewRecorder()
proxy.ServeHTTP(w, req)
if w.Code != 400 {
t.Error(helper.AssertError(400, w.Code))
}
}