| 85 | } |
| 86 | |
| 87 | func TestInTx_CapturesRollbackError(t *testing.T) { |
| 88 | t.Parallel() |
| 89 | |
| 90 | sqlDB, mock, err := sqlmock.New() |
| 91 | require.NoError(t, err) |
| 92 | t.Cleanup(func() { _ = sqlDB.Close() }) |
| 93 | |
| 94 | db := database.New(sqlDB) |
| 95 | |
| 96 | callbackErr := xerrors.New("callback failed") |
| 97 | rollbackErr := xerrors.New("rollback failed") |
| 98 | |
| 99 | mock.ExpectBegin() |
| 100 | mock.ExpectRollback().WillReturnError(rollbackErr) |
| 101 | |
| 102 | err = db.InTx(func(_ database.Store) error { |
| 103 | return callbackErr |
| 104 | }, nil) |
| 105 | require.EqualError(t, err, "defer (rollback failed): execute transaction: callback failed") |
| 106 | require.ErrorIs(t, err, callbackErr, |
| 107 | "returned error should still match the callback error when rollback fails") |
| 108 | require.NotErrorIs(t, err, rollbackErr, |
| 109 | "rollback failure should be reported in the message, not wrapped in the error chain") |
| 110 | |
| 111 | require.NoError(t, mock.ExpectationsWereMet()) |
| 112 | } |
| 113 | |
| 114 | func testSQLDB(t testing.TB) *sql.DB { |
| 115 | t.Helper() |