Rewrote tests with testutil for several test files (#4086)

* promql: Rewrote tests with testutil for functions_test

Signed-off-by: Elif T. Kuş <elifkus@gmail.com>

* pkg/relabel: Rewrote tests with testutil for relabel_test

Signed-off-by: Elif T. Kuş <elifkus@gmail.com>

* discovery/consul: Rewrote tests with testutil for consul_test

Signed-off-by: Elif T. Kuş <elifkus@gmail.com>

* scrape: Rewrote tests with testutil for manager_test

Signed-off-by: Elif T. Kuş <elifkus@gmail.com>

discovery/consul/consul_test.go

@@ -14,18 +14,18 @@
 package consul
 
 import (
+	"context"
 	"testing"
+	"time"
 
-	"github.com/stretchr/testify/require"
 	"net/http"
 	"net/http/httptest"
 	"net/url"
 
-	"context"
 	"github.com/go-kit/kit/log"
 	"github.com/prometheus/common/model"
 	"github.com/prometheus/prometheus/discovery/targetgroup"
-	"time"
+	"github.com/prometheus/prometheus/util/testutil"
 )
 
 func TestConfiguredService(t *testing.T) {
@@ -127,11 +127,11 @@ func newServer(t *testing.T) (*httptest.Server, *SDConfig) {
 		w.Header().Add("X-Consul-Index", "1")
 		w.Write([]byte(response))
 	}))
-	url, err := url.Parse(stub.URL)
-	require.NoError(t, err)
+	stuburl, err := url.Parse(stub.URL)
+	testutil.Ok(t, err)
 
 	config := &SDConfig{
-		Server:          url.Host,
+		Server:          stuburl.Host,
 		Token:           "fake-token",
 		RefreshInterval: model.Duration(1 * time.Second),
 	}
@@ -141,18 +141,18 @@ func newServer(t *testing.T) (*httptest.Server, *SDConfig) {
 func newDiscovery(t *testing.T, config *SDConfig) *Discovery {
 	logger := log.NewNopLogger()
 	d, err := NewDiscovery(config, logger)
-	require.NoError(t, err)
+	testutil.Ok(t, err)
 	return d
 }
 
 func checkOneTarget(t *testing.T, tg []*targetgroup.Group) {
-	require.Equal(t, 1, len(tg))
+	testutil.Equals(t, 1, len(tg))
 	target := tg[0]
-	require.Equal(t, "test-dc", string(target.Labels["__meta_consul_dc"]))
-	require.Equal(t, target.Source, string(target.Labels["__meta_consul_service"]))
+	testutil.Equals(t, "test-dc", string(target.Labels["__meta_consul_dc"]))
+	testutil.Equals(t, target.Source, string(target.Labels["__meta_consul_service"]))
 	if target.Source == "test" {
 		// test service should have one node.
-		require.NotEmpty(t, target.Targets)
+		testutil.Assert(t, len(target.Targets) > 0, "Test service should have one node")
 	}
 }
 

pkg/relabel/relabel_test.go

@@ -14,13 +14,13 @@
 package relabel
 
 import (
-	"reflect"
 	"testing"
 
 	"github.com/prometheus/common/model"
 
 	"github.com/prometheus/prometheus/config"
 	"github.com/prometheus/prometheus/pkg/labels"
+	"github.com/prometheus/prometheus/util/testutil"
 )
 
 func TestRelabel(t *testing.T) {
@@ -413,11 +413,8 @@ func TestRelabel(t *testing.T) {
 		},
 	}
 
-	for i, test := range tests {
+	for _, test := range tests {
 		res := Process(test.input, test.relabel...)
-
-		if !reflect.DeepEqual(res, test.output) {
-			t.Errorf("Test %d: relabel output mismatch: expected %#v, got %#v", i+1, test.output, res)
-		}
+		testutil.Equals(t, test.output, res)
 	}
 }

promql/functions_test.go

@@ -89,31 +89,22 @@ func TestDeriv(t *testing.T) {
 	engine := NewEngine(nil, nil, 10, 10*time.Second)
 
 	a, err := storage.Appender()
-	if err != nil {
-		t.Fatal(err)
-	}
+	testutil.Ok(t, err)
 
 	metric := labels.FromStrings("__name__", "foo")
 	a.Add(metric, 1493712816939, 1.0)
 	a.Add(metric, 1493712846939, 1.0)
 
-	if err := a.Commit(); err != nil {
-		t.Fatal(err)
-	}
+	err = a.Commit()
+	testutil.Ok(t, err)
 
 	query, err := engine.NewInstantQuery(storage, "deriv(foo[30m])", timestamp.Time(1493712846939))
-	if err != nil {
-		t.Fatalf("Error parsing query: %s", err)
-	}
+	testutil.Ok(t, err)
+
 	result := query.Exec(context.Background())
-	if result.Err != nil {
-		t.Fatalf("Error running query: %s", result.Err)
-	}
+	testutil.Ok(t, result.Err)
+
 	vec, _ := result.Vector()
-	if len(vec) != 1 {
-		t.Fatalf("Expected 1 result, got %d", len(vec))
-	}
-	if vec[0].V != 0.0 {
-		t.Fatalf("Expected 0.0 as value, got %f", vec[0].V)
-	}
+	testutil.Assert(t, len(vec) == 1, "Expected 1 result, got %d", len(vec))
+	testutil.Assert(t, vec[0].V == 0.0, "Expected 0.0 as value, got %f", vec[0].V)
 }

scrape/manager_test.go

@@ -15,7 +15,6 @@ package scrape
 
 import (
 	"fmt"
-	"reflect"
 	"testing"
 	"time"
 
@@ -23,6 +22,7 @@ import (
 	"github.com/prometheus/prometheus/config"
 	"github.com/prometheus/prometheus/discovery/targetgroup"
 	"github.com/prometheus/prometheus/pkg/labels"
+	"github.com/prometheus/prometheus/util/testutil"
 )
 
 func mustNewRegexp(s string) config.Regexp {
@@ -214,22 +214,14 @@ func TestPopulateLabels(t *testing.T) {
 			err:     fmt.Errorf("invalid label value for \"custom\": \"\\xbd\""),
 		},
 	}
-	for i, c := range cases {
+	for _, c := range cases {
 		in := c.in.Copy()
 
 		res, orig, err := populateLabels(c.in, c.cfg)
-		if !reflect.DeepEqual(err, c.err) {
-			t.Fatalf("case %d: wanted %v error, got %v", i, c.err, err)
-		}
-		if !reflect.DeepEqual(c.in, in) {
-			t.Errorf("case %d: input lset was changed was\n\t%+v\n now\n\t%+v", i, in, c.in)
-		}
-		if !reflect.DeepEqual(res, c.res) {
-			t.Errorf("case %d: expected res\n\t%+v\n got\n\t%+v", i, c.res, res)
-		}
-		if !reflect.DeepEqual(orig, c.resOrig) {
-			t.Errorf("case %d: expected resOrig\n\t%+v\n got\n\t%+v", i, c.resOrig, orig)
-		}
+		testutil.Equals(t, c.err, err)
+		testutil.Equals(t, c.in, in)
+		testutil.Equals(t, c.res, res)
+		testutil.Equals(t, c.resOrig, orig)
 	}
 }