Update BQ Test.

artie-labs · Nov 17, 2024 · fc2620f · fc2620f
1 parent 2865543
commit fc2620f
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/clients/bigquery/dialect/dialect_test.go b/clients/bigquery/dialect/dialect_test.go
@@ -66,13 +66,23 @@ func TestBigQueryDialect_BuildCreateTableQuery(t *testing.T) {
 	)
 }
 
-func TestBigQueryDialect_BuildAlterColumnQuery(t *testing.T) {
+func TestBigQueryDialect_BuildDropColumnQuery(t *testing.T) {
 	fakeTableID := &mocks.FakeTableIdentifier{}
 	fakeTableID.FullyQualifiedNameReturns("{TABLE}")
 
 	assert.Equal(t,
 		"ALTER TABLE {TABLE} drop COLUMN {SQL_PART}",
-		BigQueryDialect{}.buildAlterColumnQuery(fakeTableID, constants.Delete, "{SQL_PART}"),
+		BigQueryDialect{}.BuildDropColumnQuery(fakeTableID, "{SQL_PART}"),
+	)
+}
+
+func TestBigQueryDialect_BuildAddColumnQuery(t *testing.T) {
+	fakeTableID := &mocks.FakeTableIdentifier{}
+	fakeTableID.FullyQualifiedNameReturns("{TABLE}")
+
+	assert.Equal(t,
+		"ALTER TABLE {TABLE} add COLUMN {SQL_PART}",
+		BigQueryDialect{}.BuildAddColumnQuery(fakeTableID, "{SQL_PART}"),
 	)
 }