Move test

artie-labs · May 13, 2024 · 654c1a0 · 654c1a0
1 parent 31a6859
commit 654c1a0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/clients/bigquery/dialect/dialect_test.go b/clients/bigquery/dialect/dialect_test.go
@@ -207,6 +207,12 @@ func TestBuildProcessToastStructColExpression(t *testing.T) {
 	assert.Equal(t, `CASE WHEN COALESCE(TO_JSON_STRING(cc.foo) != '{"key":"__debezium_unavailable_value"}', true) THEN cc.foo ELSE c.foo END`, BigQueryDialect{}.BuildProcessToastStructColExpression("foo"))
 }
 
+func TestQuoteColumns(t *testing.T) {
+	assert.Equal(t, []string{}, columns.QuoteColumns(nil, BigQueryDialect{}))
+	cols := []columns.Column{columns.NewColumn("a", typing.Invalid), columns.NewColumn("b", typing.Invalid)}
+	assert.Equal(t, []string{"`a`", "`b`"}, columns.QuoteColumns(cols, BigQueryDialect{}))
+}
+
 func TestBuildColumnsUpdateFragment(t *testing.T) {
 	var lastCaseColTypes []columns.Column
 	lastCaseCols := []string{"a1", "b2", "c3"}

diff --git a/clients/snowflake/dialect/dialect_test.go b/clients/snowflake/dialect/dialect_test.go
@@ -252,6 +252,12 @@ func TestBuildProcessToastStructColExpression(t *testing.T) {
 	assert.Equal(t, `CASE WHEN COALESCE(cc.foo != {'key': '__debezium_unavailable_value'}, true) THEN cc.foo ELSE c.foo END`, SnowflakeDialect{}.BuildProcessToastStructColExpression("foo"))
 }
 
+func TestQuoteColumns(t *testing.T) {
+	assert.Equal(t, []string{}, columns.QuoteColumns(nil, SnowflakeDialect{}))
+	cols := []columns.Column{columns.NewColumn("a", typing.Invalid), columns.NewColumn("b", typing.Invalid)}
+	assert.Equal(t, []string{`"A"`, `"B"`}, columns.QuoteColumns(cols, SnowflakeDialect{}))
+}
+
 func TestBuildColumnsUpdateFragment(t *testing.T) {
 	var stringAndToastCols []columns.Column
 	for _, col := range []string{"foo", "bar"} {