summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/prometheus/client_golang/api/prometheus/v1/api_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/prometheus/client_golang/api/prometheus/v1/api_test.go')
-rw-r--r--vendor/github.com/prometheus/client_golang/api/prometheus/v1/api_test.go424
1 files changed, 0 insertions, 424 deletions
diff --git a/vendor/github.com/prometheus/client_golang/api/prometheus/v1/api_test.go b/vendor/github.com/prometheus/client_golang/api/prometheus/v1/api_test.go
deleted file mode 100644
index e557d68f9..000000000
--- a/vendor/github.com/prometheus/client_golang/api/prometheus/v1/api_test.go
+++ /dev/null
@@ -1,424 +0,0 @@
-// Copyright 2017 The Prometheus Authors
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-// http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// +build go1.7
-
-package v1
-
-import (
- "context"
- "encoding/json"
- "fmt"
- "net/http"
- "net/url"
- "reflect"
- "strings"
- "testing"
- "time"
-
- "github.com/prometheus/common/model"
-)
-
-type apiTest struct {
- do func() (interface{}, error)
- inErr error
- inRes interface{}
-
- reqPath string
- reqParam url.Values
- reqMethod string
- res interface{}
- err error
-}
-
-type apiTestClient struct {
- *testing.T
- curTest apiTest
-}
-
-func (c *apiTestClient) URL(ep string, args map[string]string) *url.URL {
- path := ep
- for k, v := range args {
- path = strings.Replace(path, ":"+k, v, -1)
- }
- u := &url.URL{
- Host: "test:9090",
- Path: path,
- }
- return u
-}
-
-func (c *apiTestClient) Do(ctx context.Context, req *http.Request) (*http.Response, []byte, error) {
-
- test := c.curTest
-
- if req.URL.Path != test.reqPath {
- c.Errorf("unexpected request path: want %s, got %s", test.reqPath, req.URL.Path)
- }
- if req.Method != test.reqMethod {
- c.Errorf("unexpected request method: want %s, got %s", test.reqMethod, req.Method)
- }
-
- b, err := json.Marshal(test.inRes)
- if err != nil {
- c.Fatal(err)
- }
-
- resp := &http.Response{}
- if test.inErr != nil {
- resp.StatusCode = statusAPIError
- } else {
- resp.StatusCode = http.StatusOK
- }
-
- return resp, b, test.inErr
-}
-
-func TestAPIs(t *testing.T) {
-
- testTime := time.Now()
-
- client := &apiTestClient{T: t}
-
- queryAPI := &httpAPI{
- client: client,
- }
-
- doQuery := func(q string, ts time.Time) func() (interface{}, error) {
- return func() (interface{}, error) {
- return queryAPI.Query(context.Background(), q, ts)
- }
- }
-
- doQueryRange := func(q string, rng Range) func() (interface{}, error) {
- return func() (interface{}, error) {
- return queryAPI.QueryRange(context.Background(), q, rng)
- }
- }
-
- doLabelValues := func(label string) func() (interface{}, error) {
- return func() (interface{}, error) {
- return queryAPI.LabelValues(context.Background(), label)
- }
- }
-
- doSeries := func(matcher string, startTime time.Time, endTime time.Time) func() (interface{}, error) {
- return func() (interface{}, error) {
- return queryAPI.Series(context.Background(), []string{matcher}, startTime, endTime)
- }
- }
-
- queryTests := []apiTest{
- {
- do: doQuery("2", testTime),
- inRes: &queryResult{
- Type: model.ValScalar,
- Result: &model.Scalar{
- Value: 2,
- Timestamp: model.TimeFromUnix(testTime.Unix()),
- },
- },
-
- reqMethod: "GET",
- reqPath: "/api/v1/query",
- reqParam: url.Values{
- "query": []string{"2"},
- "time": []string{testTime.Format(time.RFC3339Nano)},
- },
- res: &model.Scalar{
- Value: 2,
- Timestamp: model.TimeFromUnix(testTime.Unix()),
- },
- },
- {
- do: doQuery("2", testTime),
- inErr: fmt.Errorf("some error"),
-
- reqMethod: "GET",
- reqPath: "/api/v1/query",
- reqParam: url.Values{
- "query": []string{"2"},
- "time": []string{testTime.Format(time.RFC3339Nano)},
- },
- err: fmt.Errorf("some error"),
- },
-
- {
- do: doQueryRange("2", Range{
- Start: testTime.Add(-time.Minute),
- End: testTime,
- Step: time.Minute,
- }),
- inErr: fmt.Errorf("some error"),
-
- reqMethod: "GET",
- reqPath: "/api/v1/query_range",
- reqParam: url.Values{
- "query": []string{"2"},
- "start": []string{testTime.Add(-time.Minute).Format(time.RFC3339Nano)},
- "end": []string{testTime.Format(time.RFC3339Nano)},
- "step": []string{time.Minute.String()},
- },
- err: fmt.Errorf("some error"),
- },
-
- {
- do: doLabelValues("mylabel"),
- inRes: []string{"val1", "val2"},
- reqMethod: "GET",
- reqPath: "/api/v1/label/mylabel/values",
- res: model.LabelValues{"val1", "val2"},
- },
-
- {
- do: doLabelValues("mylabel"),
- inErr: fmt.Errorf("some error"),
- reqMethod: "GET",
- reqPath: "/api/v1/label/mylabel/values",
- err: fmt.Errorf("some error"),
- },
-
- {
- do: doSeries("up", testTime.Add(-time.Minute), testTime),
- inRes: []map[string]string{
- {
- "__name__": "up",
- "job": "prometheus",
- "instance": "localhost:9090"},
- },
- reqMethod: "GET",
- reqPath: "/api/v1/series",
- reqParam: url.Values{
- "match": []string{"up"},
- "start": []string{testTime.Add(-time.Minute).Format(time.RFC3339Nano)},
- "end": []string{testTime.Format(time.RFC3339Nano)},
- },
- res: []model.LabelSet{
- model.LabelSet{
- "__name__": "up",
- "job": "prometheus",
- "instance": "localhost:9090",
- },
- },
- },
-
- {
- do: doSeries("up", testTime.Add(-time.Minute), testTime),
- inErr: fmt.Errorf("some error"),
- reqMethod: "GET",
- reqPath: "/api/v1/series",
- reqParam: url.Values{
- "match": []string{"up"},
- "start": []string{testTime.Add(-time.Minute).Format(time.RFC3339Nano)},
- "end": []string{testTime.Format(time.RFC3339Nano)},
- },
- err: fmt.Errorf("some error"),
- },
- }
-
- var tests []apiTest
- tests = append(tests, queryTests...)
-
- for _, test := range tests {
- client.curTest = test
-
- res, err := test.do()
-
- if test.err != nil {
- if err == nil {
- t.Errorf("expected error %q but got none", test.err)
- continue
- }
- if err.Error() != test.err.Error() {
- t.Errorf("unexpected error: want %s, got %s", test.err, err)
- }
- continue
- }
- if err != nil {
- t.Errorf("unexpected error: %s", err)
- continue
- }
-
- if !reflect.DeepEqual(res, test.res) {
- t.Errorf("unexpected result: want %v, got %v", test.res, res)
- }
- }
-}
-
-type testClient struct {
- *testing.T
-
- ch chan apiClientTest
- req *http.Request
-}
-
-type apiClientTest struct {
- code int
- response interface{}
- expected string
- err *Error
-}
-
-func (c *testClient) URL(ep string, args map[string]string) *url.URL {
- return nil
-}
-
-func (c *testClient) Do(ctx context.Context, req *http.Request) (*http.Response, []byte, error) {
- if ctx == nil {
- c.Fatalf("context was not passed down")
- }
- if req != c.req {
- c.Fatalf("request was not passed down")
- }
-
- test := <-c.ch
-
- var b []byte
- var err error
-
- switch v := test.response.(type) {
- case string:
- b = []byte(v)
- default:
- b, err = json.Marshal(v)
- if err != nil {
- c.Fatal(err)
- }
- }
-
- resp := &http.Response{
- StatusCode: test.code,
- }
-
- return resp, b, nil
-}
-
-func TestAPIClientDo(t *testing.T) {
- tests := []apiClientTest{
- {
- response: &apiResponse{
- Status: "error",
- Data: json.RawMessage(`null`),
- ErrorType: ErrBadData,
- Error: "failed",
- },
- err: &Error{
- Type: ErrBadData,
- Msg: "failed",
- },
- code: statusAPIError,
- expected: `null`,
- },
- {
- response: &apiResponse{
- Status: "error",
- Data: json.RawMessage(`"test"`),
- ErrorType: ErrTimeout,
- Error: "timed out",
- },
- err: &Error{
- Type: ErrTimeout,
- Msg: "timed out",
- },
- code: statusAPIError,
- expected: `test`,
- },
- {
- response: "bad json",
- err: &Error{
- Type: ErrBadResponse,
- Msg: "bad response code 400",
- },
- code: http.StatusBadRequest,
- },
- {
- response: "bad json",
- err: &Error{
- Type: ErrBadResponse,
- Msg: "invalid character 'b' looking for beginning of value",
- },
- code: statusAPIError,
- },
- {
- response: &apiResponse{
- Status: "success",
- Data: json.RawMessage(`"test"`),
- },
- err: &Error{
- Type: ErrBadResponse,
- Msg: "inconsistent body for response code",
- },
- code: statusAPIError,
- },
- {
- response: &apiResponse{
- Status: "success",
- Data: json.RawMessage(`"test"`),
- ErrorType: ErrTimeout,
- Error: "timed out",
- },
- err: &Error{
- Type: ErrBadResponse,
- Msg: "inconsistent body for response code",
- },
- code: statusAPIError,
- },
- {
- response: &apiResponse{
- Status: "error",
- Data: json.RawMessage(`"test"`),
- ErrorType: ErrTimeout,
- Error: "timed out",
- },
- err: &Error{
- Type: ErrBadResponse,
- Msg: "inconsistent body for response code",
- },
- code: http.StatusOK,
- },
- }
-
- tc := &testClient{
- T: t,
- ch: make(chan apiClientTest, 1),
- req: &http.Request{},
- }
- client := &apiClient{tc}
-
- for _, test := range tests {
-
- tc.ch <- test
-
- _, body, err := client.Do(context.Background(), tc.req)
-
- if test.err != nil {
- if err == nil {
- t.Errorf("expected error %q but got none", test.err)
- continue
- }
- if test.err.Error() != err.Error() {
- t.Errorf("unexpected error: want %q, got %q", test.err, err)
- }
- continue
- }
- if err != nil {
- t.Errorf("unexpeceted error %s", err)
- continue
- }
-
- want, got := test.expected, string(body)
- if want != got {
- t.Errorf("unexpected body: want %q, got %q", want, got)
- }
- }
-}
generated by cgit v1.2.3-1-g7c22 (git 2.25.1) at 2024-06-13 20:58:45 +0000