chore: fix tests

newrelic · Oct 24, 2023 · 12913b6 · 12913b6
1 parent e47eda9
commit 12913b6
Showing 1 changed file with 73 additions and 3 deletions.
diff --git a/internal/entities/command_deployment_test.go b/internal/entities/command_deployment_test.go
@@ -28,7 +28,7 @@ func TestParseAttributesSingleKeyValue(t *testing.T) {
 		"key:value",
 	}
 
-	var want = map[string]string{
+	var want = map[string]interface{}{
 		"key": "value",
 	}
 	var errWant error
@@ -39,13 +39,83 @@ func TestParseAttributesSingleKeyValue(t *testing.T) {
 	assert.Equal(t, want, *got)
 }
 
+func TestParseAttributesSingleIntegerKeyValue(t *testing.T) {
+	a := []string{
+		"a:1",
+	}
+
+	var want = map[string]interface{}{
+		"a": "1",
+	}
+	var errWant error
+
+	got, errGot := parseCustomAttributes(&a)
+
+	assert.Equal(t, errWant, errGot)
+	assert.Equal(t, want, *got)
+}
+
+func TestParseAttributesSingleFloatingKeyValue(t *testing.T) {
+	a := []string{
+		"a:1.5",
+	}
+
+	var want = map[string]interface{}{
+		"a": "1.5",
+	}
+	var errWant error
+
+	got, errGot := parseCustomAttributes(&a)
+
+	assert.Equal(t, errWant, errGot)
+	assert.Equal(t, want, *got)
+}
+
+func TestParseAttributesSingleBooleanKeyValue(t *testing.T) {
+	a := []string{
+		"a:true",
+	}
+
+	var want = map[string]interface{}{
+		"a": "true",
+	}
+	var errWant error
+
+	got, errGot := parseCustomAttributes(&a)
+
+	assert.Equal(t, errWant, errGot)
+	assert.Equal(t, want, *got)
+}
+
+func TestParseAttributesMultipleTypesKeyValues(t *testing.T) {
+	a := []string{
+		"a:true",
+		"b:1",
+		"c:1.5",
+		`d:"value"`,
+	}
+
+	var want = map[string]interface{}{
+		"a": "true",
+		"b": "1",
+		"c": "1.5",
+		"d": `"value"`,
+	}
+	var errWant error
+
+	got, errGot := parseCustomAttributes(&a)
+
+	assert.Equal(t, errWant, errGot)
+	assert.Equal(t, want, *got)
+}
+
 func TestParseAttributesTwoKeyValues(t *testing.T) {
 	a := []string{
 		"key:value",
 		"key2:value2",
 	}
 
-	var want = map[string]string{
+	var want = map[string]interface{}{
 		"key":  "value",
 		"key2": "value2",
 	}
@@ -97,6 +167,6 @@ func TestParseAttributesEmptyStringSlice(t *testing.T) {
 	assert.Equal(t, want, got)
 }
 
-func nilPointerMapStringString() *map[string]string {
+func nilPointerMapStringString() *map[string]interface{} {
 	return nil
 }