code-cov

Signed-off-by: Manik2708 <[email protected]>

Author: Manik2708

internal/storage/v2/elasticsearch/tracestore/tag_appender_test.go

@@ -146,3 +146,99 @@ func TestAppendTagWhenTagKeysAsField(t *testing.T) {
 	expected := dbmodel.KeyValue{Key: "testing.key.2", Type: "int64", Value: "2"}
 	assert.Equal(t, expected, tags[0])
 }
+
+func TestAppendTag(t *testing.T) {
+	t.Run("allTagsAsFields=true", func(t *testing.T) {
+		testAppendTags(t, true)
+	})
+	t.Run("allTagsAsFields=false", func(t *testing.T) {
+		testAppendTags(t, false)
+	})
+}
+
+func testAppendTags(t *testing.T, allTagsAsFields bool) {
+	tests := []struct {
+		name   string
+		value  pcommon.Value
+		dbType dbmodel.ValueType
+	}{
+		{
+			name:   "int.val",
+			value:  pcommon.NewValueInt(1),
+			dbType: dbmodel.Int64Type,
+		},
+		{
+			name:   "string.val",
+			value:  pcommon.NewValueStr("testing-string"),
+			dbType: dbmodel.StringType,
+		},
+		{
+			name:   "bool.val",
+			value:  pcommon.NewValueBool(true),
+			dbType: dbmodel.BoolType,
+		},
+		{
+			name:   "double.val",
+			value:  pcommon.NewValueDouble(1.2),
+			dbType: dbmodel.Float64Type,
+		},
+		{
+			name:   "bytes.val",
+			value:  pcommon.NewValueBytes(),
+			dbType: dbmodel.BinaryType,
+		},
+		{
+			name:   "map.val",
+			value:  pcommon.NewValueMap(),
+			dbType: dbmodel.StringType,
+		},
+		{
+			name:   "slice.val",
+			value:  pcommon.NewValueSlice(),
+			dbType: dbmodel.StringType,
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			tagAppender := newTagAppender(allTagsAsFields, nil, ".")
+			tagAppender.appendTag(test.name, test.value)
+			tags, tagMap := tagAppender.getTags()
+			if allTagsAsFields && test.dbType != dbmodel.BinaryType {
+				assert.Empty(t, tags)
+				assert.Len(t, tagMap, 1)
+				switch test.dbType {
+				case dbmodel.StringType:
+					assert.Equal(t, test.value.AsString(), tagMap[test.name])
+				case dbmodel.Int64Type:
+					assert.Equal(t, test.value.Int(), tagMap[test.name])
+				case dbmodel.BoolType:
+					assert.Equal(t, test.value.Bool(), tagMap[test.name])
+				case dbmodel.Float64Type:
+					assert.InDelta(t, test.value.Double(), tagMap[test.name], 0.01)
+				default:
+					t.Errorf("unknown db type: %v", test.dbType)
+				}
+			} else {
+				assert.Len(t, tags, 1)
+				assert.Empty(t, tagMap)
+				expected := dbmodel.KeyValue{
+					Key:   test.name,
+					Type:  test.dbType,
+					Value: test.value.AsString(),
+				}
+				assert.Equal(t, expected, tags[0])
+			}
+		})
+	}
+}
+
+func TestAttributeToDbType_UnknownType(t *testing.T) {
+	dbType := attributeToDbType(pcommon.ValueType(13))
+	assert.Equal(t, dbmodel.ValueType(""), dbType)
+}
+
+func TestAttributeToDbValue_UnknownType(t *testing.T) {
+	val := pcommon.NewValueEmpty()
+	dbValue := attributeToDbValue(pcommon.NewValueEmpty())
+	assert.Equal(t, val.AsString(), dbValue)
+}