diff --git a/lib/destination/ddl/ddl_alter_delete_test.go b/lib/destination/ddl/ddl_alter_delete_test.go index f4246210a..2cedc366a 100644 --- a/lib/destination/ddl/ddl_alter_delete_test.go +++ b/lib/destination/ddl/ddl_alter_delete_test.go @@ -258,13 +258,13 @@ func (d *DDLTestSuite) TestAlterDelete_Complete() { } // Nothing has been deleted, but it is all added to the permissions table. - assert.Equal(d.T(), originalColumnLength, len(bqTc.Columns().GetColumns()), bqTc.Columns().GetColumns()) - assert.Equal(d.T(), originalColumnLength, len(redshiftTc.Columns().GetColumns()), redshiftTc.Columns().GetColumns()) - assert.Equal(d.T(), originalColumnLength, len(snowflakeTc.Columns().GetColumns()), snowflakeTc.Columns().GetColumns()) + assert.Len(d.T(), bqTc.GetColumns(), originalColumnLength) + assert.Len(d.T(), redshiftTc.GetColumns(), originalColumnLength) + assert.Len(d.T(), snowflakeTc.GetColumns(), originalColumnLength) - assert.Equal(d.T(), originalColumnLength, len(bqTc.ReadOnlyColumnsToDelete()), bqTc.ReadOnlyColumnsToDelete()) - assert.Equal(d.T(), originalColumnLength, len(redshiftTc.ReadOnlyColumnsToDelete()), redshiftTc.ReadOnlyColumnsToDelete()) - assert.Equal(d.T(), originalColumnLength, len(snowflakeTc.ReadOnlyColumnsToDelete()), snowflakeTc.ReadOnlyColumnsToDelete()) + assert.Len(d.T(), bqTc.ReadOnlyColumnsToDelete(), originalColumnLength) + assert.Len(d.T(), redshiftTc.ReadOnlyColumnsToDelete(), originalColumnLength) + assert.Len(d.T(), snowflakeTc.ReadOnlyColumnsToDelete(), originalColumnLength) for _, column := range cols.GetColumns() { alterTableArgs := ddl.AlterTableArgs{ diff --git a/lib/destination/ddl/ddl_bq_test.go b/lib/destination/ddl/ddl_bq_test.go index 4c6734078..bb718f467 100644 --- a/lib/destination/ddl/ddl_bq_test.go +++ b/lib/destination/ddl/ddl_bq_test.go @@ -149,7 +149,7 @@ func (d *DDLTestSuite) TestAlterTableAddColumns() { // Check all the columns, make sure it's correct. (length) assert.Equal(d.T(), newColsLen+existingColsLen, len(d.bigQueryStore.GetConfigMap().TableConfigCache(tableID).Columns().GetColumns()), d.bigQueryStore.GetConfigMap().TableConfigCache(tableID).Columns()) // Check by iterating over the columns - for _, column := range d.bigQueryStore.GetConfigMap().TableConfigCache(tableID).Columns().GetColumns() { + for _, column := range d.bigQueryStore.GetConfigMap().TableConfigCache(tableID).GetColumns() { existingCol, isOk := existingCols.GetColumn(column.Name()) if !isOk { // Check new cols?