[-] fix BeginTx() when returning error

pashagolub · Sep 8, 2023 · 887bff8 · 887bff8
1 parent 7ccae6c
commit 887bff8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pgxmock.go b/pgxmock.go
@@ -362,7 +362,10 @@ func (c *pgxmock) BeginTx(ctx context.Context, txOptions pgx.TxOptions) (pgx.Tx,
 	if err != nil {
 		return nil, err
 	}
-	return c, ex.waitForDelay(ctx)
+	if err = ex.waitForDelay(ctx); err != nil {
+		return nil, err
+	}
+	return c, nil
 }
 
 func (c *pgxmock) Prepare(ctx context.Context, name, query string) (*pgconn.StatementDescription, error) {