Пример #1
0
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))
	}
}
Пример #2
0
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))
	}
}
Пример #3
0
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))
	}
}
Пример #4
0
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))
	}
}
Пример #5
0
// 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))
	}
}