add more tests

yahoo · Oct 19, 2020 · 2c6c8e0 · 2c6c8e0
1 parent ca49a2d
commit 2c6c8e0
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/config/helper_test.go b/config/helper_test.go
@@ -5,6 +5,7 @@ package config
 
 import (
 	"encoding/json"
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -85,3 +86,16 @@ func TestSetDefault(t *testing.T) {
 	SetDefault(&vv, 6)
 	assert.Equal(t, uint(5), vv)
 }
+
+func TestGetEnvInt(t *testing.T) {
+	v := GetEnvInt("NOT_EXIST", 10)
+	assert.Equal(t, 10, v)
+
+	os.Setenv("PANOPTES_TEST", "11")
+	v = GetEnvInt("TEST", 10)
+	assert.Equal(t, 11, v)
+
+	os.Setenv("PANOPTES_TEST", "CHAR")
+	v = GetEnvInt("TEST", 10)
+	assert.Equal(t, 10, v)
+}
diff --git a/telemetry/helper_test.go b/telemetry/helper_test.go
@@ -195,3 +195,13 @@ func TestGetValue(t *testing.T) {
 		assert.Equal(t, row.v, v)
 	}
 }
+
+func TestMergeLabel(t *testing.T) {
+	kl := map[string]string{"a": "b", "c": "q"}
+	pl := map[string]string{"a": "b"}
+	exp := map[string]string{"/prefix/a": "b", "a": "b", "c": "q"}
+
+	m := MergeLabels(kl, pl, "/prefix")
+
+	assert.Equal(t, exp, m)
+}