@@ -55,6 +55,9 @@ func TestKeyValuesAndSizes(t *testing.T) {
5555}
5656
5757func TestKeyValuesAsLabel (t * testing.T ) {
58+ setupDBKeys (t , os .Getenv ("TEST_REDIS_URI" ))
59+ defer deleteKeysFromDB (t , os .Getenv ("TEST_REDIS_URI" ))
60+
5861 for _ , exc := range []bool {true , false } {
5962 e , _ := NewRedisExporter (
6063 os .Getenv ("TEST_REDIS_URI" ),
@@ -65,10 +68,6 @@ func TestKeyValuesAsLabel(t *testing.T) {
6568 Registry : prometheus .NewRegistry ()},
6669 )
6770 ts := httptest .NewServer (e )
68- defer ts .Close ()
69-
70- setupDBKeys (t , os .Getenv ("TEST_REDIS_URI" ))
71- defer deleteKeysFromDB (t , os .Getenv ("TEST_REDIS_URI" ))
7271
7372 chM := make (chan prometheus.Metric , 10000 )
7473 go func () {
@@ -87,6 +86,7 @@ func TestKeyValuesAsLabel(t *testing.T) {
8786 t .Fatalf ("didn't find %s with DisableExportingKeyValues disabled, body: %s" , match , body )
8887 }
8988 }
89+ ts .Close ()
9090 }
9191}
9292
@@ -95,9 +95,10 @@ func TestClusterKeyValuesAndSizes(t *testing.T) {
9595 if clusterUri == "" {
9696 t .Skipf ("Skipping TestClusterKeyValuesAndSizes, don't have env var TEST_REDIS_CLUSTER_MASTER_URI" )
9797 }
98+ setupDBKeysCluster (t , clusterUri )
99+ defer deleteKeysFromDBCluster (clusterUri )
98100
99101 for _ , exc := range []bool {true , false } {
100-
101102 e , _ := NewRedisExporter (
102103 clusterUri ,
103104 Options {
@@ -111,11 +112,6 @@ func TestClusterKeyValuesAndSizes(t *testing.T) {
111112 },
112113 )
113114
114- if err := setupDBKeysCluster (t , clusterUri ); err != nil {
115- t .Fatalf ("setupDBKeysCluster() err: %s" , err )
116- }
117- defer deleteKeysFromDBCluster (clusterUri )
118-
119115 chM := make (chan prometheus.Metric )
120116 go func () {
121117 e .Collect (chM )
@@ -531,13 +527,13 @@ func TestGetKeyInfo(t *testing.T) {
531527 for _ , f := range fixtures {
532528 info , err := getKeyInfo (c , f .key , false )
533529 if err != nil {
534- t .Errorf ("Error getting key info for %#v." , f .key )
530+ t .Fatalf ("Error getting key info for %#v." , f .key )
535531 }
536532
537533 expected := expectedSizes [f .key ]
538534 if info .size != expected {
539- t .Logf ("%#v" , info )
540535 t .Errorf ("Wrong size for key: %#v. Expected: %#v; Actual: %#v" , f .key , expected , info .size )
536+ t .Logf ("info: %#v" , info )
541537 }
542538 }
543539
@@ -672,9 +668,7 @@ func TestClusterGetKeyInfo(t *testing.T) {
672668 ts := httptest .NewServer (e )
673669 defer ts .Close ()
674670
675- if err := setupDBKeysCluster (t , clusterUri ); err != nil {
676- t .Fatalf ("setupDBKeysCluster() err: %s" , err )
677- }
671+ setupDBKeysCluster (t , clusterUri )
678672 defer deleteKeysFromDBCluster (clusterUri )
679673
680674 chM := make (chan prometheus.Metric , 10000 )
0 commit comments