From cf2c73a0e959db49c6672ea6fa07f53a7b7bdd41 Mon Sep 17 00:00:00 2001 From: Aris Tzoumas Date: Mon, 22 Jul 2024 19:45:27 +0300 Subject: [PATCH] chore(redshift-data): assert that rows affected are available after commit --- sqlconnect/internal/redshift/driver/driver_test.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sqlconnect/internal/redshift/driver/driver_test.go b/sqlconnect/internal/redshift/driver/driver_test.go index 1b4ac22..a393d11 100644 --- 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")