|
9 | 9 | IsolationLevel, |
10 | 10 | ReadVariant, |
11 | 11 | ) |
| 12 | +from psqlpy._internal.exceptions import TransactionClosedError |
12 | 13 | from psqlpy.exceptions import ( |
13 | 14 | InterfaceError, |
14 | | - TransactionBeginError, |
15 | 15 | TransactionExecuteError, |
16 | | - TransactionSavepointError, |
17 | 16 | ) |
18 | 17 |
|
19 | 18 | from tests.helpers import count_rows_in_test_table |
@@ -64,10 +63,10 @@ async def test_transaction_begin( |
64 | 63 | connection = await psql_pool.connection() |
65 | 64 | transaction = connection.transaction() |
66 | 65 |
|
67 | | - with pytest.raises(expected_exception=TransactionBeginError): |
68 | | - await transaction.execute( |
69 | | - f"SELECT * FROM {table_name}", |
70 | | - ) |
| 66 | + # with pytest.raises(expected_exception=TransactionBeginError): |
| 67 | + await transaction.execute( |
| 68 | + f"SELECT * FROM {table_name}", |
| 69 | + ) |
71 | 70 |
|
72 | 71 | await transaction.begin() |
73 | 72 |
|
@@ -170,7 +169,7 @@ async def test_transaction_rollback( |
170 | 169 |
|
171 | 170 | await transaction.rollback() |
172 | 171 |
|
173 | | - with pytest.raises(expected_exception=TransactionBeginError): |
| 172 | + with pytest.raises(expected_exception=TransactionClosedError): |
174 | 173 | await transaction.execute( |
175 | 174 | f"SELECT * FROM {table_name} WHERE name = $1", |
176 | 175 | parameters=[test_name], |
@@ -198,9 +197,8 @@ async def test_transaction_release_savepoint( |
198 | 197 | sp_name_2 = "sp2" |
199 | 198 |
|
200 | 199 | await transaction.create_savepoint(sp_name_1) |
201 | | - |
202 | | - with pytest.raises(expected_exception=TransactionSavepointError): |
203 | | - await transaction.create_savepoint(sp_name_1) |
| 200 | + # There is no problem in creating the same sp_name |
| 201 | + await transaction.create_savepoint(sp_name_1) |
204 | 202 |
|
205 | 203 | await transaction.create_savepoint(sp_name_2) |
206 | 204 |
|
|
0 commit comments