metadata_test

Signed-off-by: Angelo De Caro <adc@zurich.ibm.com>

Author: adecaro

token/core/common/meta/metadata_test.go

@@ -0,0 +1,74 @@
+package meta
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTransferActionMetadata(t *testing.T) {
+	tests := []struct {
+		name     string
+		attrs    map[interface{}]interface{}
+		expected map[string][]byte
+	}{
+		{
+			name: "Basic metadata extraction",
+			attrs: map[interface{}]interface{}{
+				"TransferMetadataPrefixKey1": []byte("value1"),
+				"TransferMetadataPrefixKey2": []byte("value2"),
+			},
+			expected: map[string][]byte{
+				"Key1": []byte("value1"),
+				"Key2": []byte("value2"),
+			},
+		},
+		{
+			name:     "Empty attrs",
+			attrs:    map[interface{}]interface{}{},
+			expected: map[string][]byte{},
+		},
+		{
+			name: "Non-string keys",
+			attrs: map[interface{}]interface{}{
+				123: []byte("value"),
+			},
+			expected: map[string][]byte{},
+		},
+		{
+			name: "Invalid value types",
+			attrs: map[interface{}]interface{}{
+				"TransferMetadataPrefixKey": 123,
+			},
+			expected: map[string][]byte{},
+		},
+		{
+			name: "Exact prefix match",
+			attrs: map[interface{}]interface{}{
+				"TransferMetadataPrefix": []byte("value"),
+			},
+			expected: map[string][]byte{
+				"": []byte("value"),
+			},
+		},
+		{
+			name: "No prefix match",
+			attrs: map[interface{}]interface{}{
+				"WrongPrefixKey": []byte("value"),
+			},
+			expected: map[string][]byte{},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := TransferActionMetadata(tt.attrs)
+			assert.Equal(t, len(result), len(tt.expected), "Expected %d items, got %d", len(tt.expected), len(result))
+			for key, value := range tt.expected {
+				gotValue, ok := result[key]
+				assert.True(t, ok, "Expected key %s to exist in result", key)
+				assert.Equal(t, value, gotValue, "Expected value for %s: %v, got: %v", key, value, gotValue)
+			}
+		})
+	}
+}