fixing non-existing table test, errors now

PeerDB-io · Sep 28, 2023 · 0ddfc5c · 0ddfc5c
1 parent f47ff9a
commit 0ddfc5c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/flow/connectors/postgres/postgres_cdc_test.go b/flow/connectors/postgres/postgres_cdc_test.go
@@ -448,9 +448,11 @@ func (suite *PostgresCDCTestSuite) TestErrorForTableNotExist() {
 		TableNameSchemaMapping: tableNameSchemaMapping,
 		RelationMessageMapping: relationMessageMapping,
 	})
-	suite.Equal(0, len(recordsWithSchemaDelta.RecordBatch.Records))
-	suite.Nil(recordsWithSchemaDelta.TableSchemaDelta)
-	suite.Nil(err)
+	suite.Nil(recordsWithSchemaDelta)
+	suite.Errorf(
+		err,
+		"error while closing statement batch: ERROR: relation \"%s\" does not exist (SQLSTATE 42P01)",
+		nonExistentFlowSrcTableName)
 
 	err = suite.connector.PullFlowCleanup(nonExistentFlowName)
 	suite.failTestError(err)