From 7c55126aaaa06da605a50e062181b8f82fced198 Mon Sep 17 00:00:00 2001 From: Lukas Jenicek Date: Mon, 1 Jul 2024 20:28:50 +0200 Subject: [PATCH] fix rollback in tests (#19) --- tests/pgkit_test.go | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tests/pgkit_test.go b/tests/pgkit_test.go index 2d44b57..1b2cb47 100644 --- a/tests/pgkit_test.go +++ b/tests/pgkit_test.go @@ -139,15 +139,14 @@ func TestGetOneInTransaction(t *testing.T) { q2 := DB.SQL.Select("*").From("accounts") var account Account - tx, err := DB.Conn.BeginTx(ctx, pgx.TxOptions{}) - assert.NoError(t, err) - - defer tx.Rollback(ctx) + err := pgx.BeginFunc(ctx, DB.Conn, func(tx pgx.Tx) error { + if err := DB.TxQuery(tx).GetOne(ctx, q2, &account); err != nil { + return fmt.Errorf("get one: %w", err) + } - err = DB.TxQuery(tx).GetOne(ctx, q2, &account) - assert.NoError(t, err) + return nil + }) - err = tx.Commit(ctx) assert.NoError(t, err) }