diff --git a/backend/internal/repository/batch_image_repo_integration_test.go b/backend/internal/repository/batch_image_repo_integration_test.go index 51a436fb71..174e0657e3 100644 --- a/backend/internal/repository/batch_image_repo_integration_test.go +++ b/backend/internal/repository/batch_image_repo_integration_test.go @@ -166,6 +166,8 @@ func TestBatchImageRepository_ItemCustomIDUniqueness(t *testing.T) { }) require.NoError(t, err) + _, err = tx.ExecContext(ctx, `SAVEPOINT batch_image_duplicate_item`) + require.NoError(t, err) _, err = repo.CreateBatchImageItem(ctx, service.CreateBatchImageItemParams{ JobID: firstBatchID, CustomID: "line-1", @@ -173,6 +175,8 @@ func TestBatchImageRepository_ItemCustomIDUniqueness(t *testing.T) { }) require.Error(t, err) require.True(t, errors.Is(err, service.ErrBatchImageItemExists)) + _, rollbackErr := tx.ExecContext(ctx, `ROLLBACK TO SAVEPOINT batch_image_duplicate_item`) + require.NoError(t, rollbackErr) _, err = repo.CreateBatchImageItem(ctx, service.CreateBatchImageItemParams{ JobID: secondBatchID,