[-] fix BeginTx() when returning error

pashagolub · pashagolub · commit 887bff8ec54b · 2023-09-08T14:52:31.000+02:00
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) {

Original file line number	Diff line number	Diff line change
`@@ -362,7 +362,10 @@ func (c *pgxmock) BeginTx(ctx context.Context, txOptions pgx.TxOptions) (pgx.Tx,`
`362`	`362`	`if err != nil {`
`363`	`363`	`return nil, err`
`364`	`364`	`}`
`365`		`- return c, ex.waitForDelay(ctx)`
	`365`	`+ if err = ex.waitForDelay(ctx); err != nil {`
	`366`	`+ return nil, err`
	`367`	`+ }`
	`368`	`+ return c, nil`
`366`	`369`	`}`
`367`	`370`
`368`	`371`	`func (c pgxmock) Prepare(ctx context.Context, name, query string) (pgconn.StatementDescription, error) {`