From 13208daa4ddd138d3131153c0ed30f19763d4fb3 Mon Sep 17 00:00:00 2001
From: Robin Tang <rtang.cs@gmail.com>
Date: Fri, 4 Oct 2024 18:09:33 -0700
Subject: [PATCH] Adding tests.

---
 lib/debezium/converters/bit_varying_test.go | 97 +++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 lib/debezium/converters/bit_varying_test.go

diff --git a/lib/debezium/converters/bit_varying_test.go b/lib/debezium/converters/bit_varying_test.go
new file mode 100644
index 00000000..f79d4a97
--- /dev/null
+++ b/lib/debezium/converters/bit_varying_test.go
@@ -0,0 +1,97 @@
+package converters
+
+import (
+	"github.com/artie-labs/transfer/lib/debezium"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestBitVaryingConverter_ToField(t *testing.T) {
+	{
+		// char size not specified
+		field := NewBitVaryingConverter(0).ToField("foo")
+		assert.Equal(t, "foo", field.FieldName)
+		assert.Equal(t, "bytes", string(field.Type))
+		assert.Equal(t, debezium.Bits, field.DebeziumType)
+		assert.Nil(t, field.Parameters)
+	}
+	{
+		// char max size 1
+		field := NewBitVaryingConverter(1).ToField("foo")
+		assert.Equal(t, "foo", field.FieldName)
+		assert.Equal(t, "bytes", string(field.Type))
+		assert.Nil(t, field.Parameters)
+	}
+	{
+		// char max size 5
+		field := NewBitVaryingConverter(5).ToField("foo")
+		assert.Equal(t, "foo", field.FieldName)
+		assert.Equal(t, "bytes", string(field.Type))
+		assert.Equal(t, debezium.Bits, field.DebeziumType)
+		assert.Nil(t, field.Parameters)
+	}
+}
+
+func TestBitVaryingConverter_Convert(t *testing.T) {
+	{
+		// char size not specified
+		_, err := BitVaryingConverter{}.Convert("foo")
+		assert.ErrorContains(t, err, `invalid binary string "foo": contains non-binary characters`)
+	}
+	{
+		// char max size 1
+		converter := NewBitVaryingConverter(1)
+		{
+			// Invalid value - wrong type
+			_, err := converter.Convert(1234)
+			assert.ErrorContains(t, err, "expected type string, got int")
+		}
+		{
+			// Valid value - 0
+			value, err := converter.Convert("0")
+			assert.NoError(t, err)
+			assert.Equal(t, []byte{}, value)
+		}
+		{
+			// Valid value - 1
+			value, err := converter.Convert("1")
+			assert.NoError(t, err)
+			assert.Equal(t, []byte{1}, value)
+		}
+		{
+			// Invalid value - 2
+			_, err := converter.Convert("2")
+			assert.ErrorContains(t, err, `invalid binary string "2": contains non-binary characters`)
+		}
+	}
+	{
+		// char max size - 5
+		{
+			// Length not matching, but it's fine.
+			converter := NewBitVaryingConverter(8)
+			value, err := converter.Convert("101111")
+			assert.NoError(t, err)
+			assert.Equal(t, []byte{47}, value)
+		}
+		{
+			// Invalid, value contains non 0s and 1s
+			converter := NewBitVaryingConverter(5)
+			_, err := converter.Convert("1011a")
+			assert.ErrorContains(t, err, "invalid binary string")
+		}
+		{
+			// Valid
+			converter := NewBitVaryingConverter(5)
+			value, err := converter.Convert("10101")
+			assert.NoError(t, err)
+			assert.Equal(t, []byte{21}, value)
+		}
+		{
+			// Valid #2
+			converter := NewBitVaryingConverter(5)
+			value, err := converter.Convert("10011")
+			assert.NoError(t, err)
+			assert.Equal(t, []byte{19}, value)
+		}
+	}
+}