chore(redshift-data): assert that rows affected are available after c…

…ommit
rudderlabs · Jul 22, 2024 · cf2c73a · cf2c73a
1 parent e299e2f
commit cf2c73a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sqlconnect/internal/redshift/driver/driver_test.go b/sqlconnect/internal/redshift/driver/driver_test.go
@@ -265,6 +265,10 @@ func TestRedshiftDriver(t *testing.T) {
 					err = tx.Commit()
 					require.NoError(t, err, "it should be able to commit the transaction")
 
+					rowsAffected, err := res.RowsAffected()
+					require.NoError(t, err, "it should be able to get rows affected after commit")
+					require.EqualValues(t, 1, rowsAffected, "rows affected should be 1")
+
 					var countAfter int
 					err = db.QueryRowContext(ctx, fmt.Sprintf(`SELECT COUNT(*) FROM "%s"."test_table"`, schema)).Scan(&countAfter)
 					require.NoError(t, err, "it should be able to execute a prepared statement")