Verify prometheus results are mixed into function list

Signed-off-by: Alex Ellis <alexellis2@gmail.com>
This commit is contained in:
Alex Ellis
2017-10-29 09:09:20 +00:00
parent 8363532751
commit 4889c779b7

View File

@ -15,8 +15,10 @@ type FakePrometheusQueryFetcher struct {
}
func (q FakePrometheusQueryFetcher) Fetch(query string) (*metrics.VectorQueryResponse, error) {
return &metrics.VectorQueryResponse{}, nil
val := []byte(`{"status":"success","data":{"resultType":"vector","result":[{"metric":{"code":"200","function_name":"func_echoit"},"value":[1509267827.752,"1"]}]}}`)
queryRes := metrics.VectorQueryResponse{}
err := json.Unmarshal(val, &queryRes)
return &queryRes, err
}
func makeFakePrometheusQueryFetcher() FakePrometheusQueryFetcher {
@ -40,10 +42,18 @@ func Test_PrometheusMetrics_MixedInto_Services(t *testing.T) {
if rr.Header().Get("Content-Type") != "application/json" {
t.Errorf("Want application/json content-type, got: %s", rr.Header().Get("Content-Type"))
}
if len(rr.Body.String()) == 0 {
body := rr.Body.String()
if len(body) == 0 {
t.Errorf("Want content-length > 0, got: %d", len(rr.Body.String()))
}
results := []requests.Function{}
json.Unmarshal([]byte(rr.Body.String()), &results)
if len(results) == 0 {
t.Errorf("Want %d function, got: %d", 1, len(results))
}
if results[0].InvocationCount != 1 {
t.Errorf("InvocationCount want: %d , got: %f", 1, results[0].InvocationCount)
}
}
func Test_FunctionsHandler_ReturnsJSONAndOneFunction(t *testing.T) {
@ -75,7 +85,7 @@ func makeFunctionsHandler() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
functions := []requests.Function{
requests.Function{
Name: "echo",
Name: "func_echoit",
Replicas: 0,
},
}