Adding tests.

artie-labs · Jul 23, 2024 · b92d914 · b92d914
1 parent 10c2d7c
commit b92d914
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/lib/dynamo/message_test.go b/lib/dynamo/message_test.go
@@ -27,6 +27,26 @@ func TestTransformAttributeValue(t *testing.T) {
 		assert.Equal(t, float64(123), actualValue)
 		assert.Equal(t, debezium.Float, fieldType)
 	}
+	{
+		// Bytes
+		actualValue, fieldType, err := transformAttributeValue(&dynamodb.AttributeValue{
+			B: []byte("hello"),
+		})
+		assert.NoError(t, err, err)
+		assert.Equal(t, []byte("hello"), actualValue)
+		assert.Equal(t, debezium.Bytes, fieldType)
+	}
+	{
+		// Bytes set
+		actualValue, fieldType, err := transformAttributeValue(&dynamodb.AttributeValue{
+			BS: [][]byte{
+				[]byte("hello"),
+			},
+		})
+		assert.NoError(t, err, err)
+		assert.Equal(t, [][]byte{[]byte("hello")}, actualValue)
+		assert.Equal(t, debezium.Array, fieldType)
+	}
 	{
 		// Boolean
 		actualValue, fieldType, err := transformAttributeValue(&dynamodb.AttributeValue{