1515from tgdb .entities .horizon .transaction import (
1616 Commit ,
1717 ConflictError ,
18- Isolation ,
18+ IsolationLevel ,
1919 PreparedCommit ,
2020 SerializableTransaction ,
2121 Transaction ,
@@ -47,7 +47,7 @@ def test_with_only_start(
4747 """
4848
4949 horizon .start_transaction (
50- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
50+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
5151 )
5252
5353 if object == "bool" :
@@ -68,10 +68,10 @@ def test_two_concurrent_transactions(
6868 """
6969
7070 horizon .start_transaction (
71- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
71+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
7272 )
7373 horizon .start_transaction (
74- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
74+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
7575 )
7676
7777 if object == "bool" :
@@ -125,12 +125,12 @@ def test_with_two_start(horizon: Horizon) -> None:
125125 """
126126
127127 horizon .start_transaction (
128- 2 , UUID (int = 1 ), Isolation .serializable_read_and_write
128+ 2 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
129129 )
130130
131131 with raises (InvalidTransactionStateError ):
132132 horizon .start_transaction (
133- 3 , UUID (int = 1 ), Isolation .serializable_read_and_write
133+ 3 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
134134 )
135135
136136
@@ -144,7 +144,7 @@ def test_rollback_with_transaction(
144144 """
145145
146146 horizon .start_transaction (
147- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
147+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
148148 )
149149 horizon .rollback_transaction (2 , UUID (int = 1 ))
150150
@@ -168,7 +168,7 @@ def test_commit_with_transaction_without_transaction_effect(
168168 """
169169
170170 horizon .start_transaction (
171- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
171+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
172172 )
173173 commit = horizon .commit_transaction (2 , UUID (int = 1 ), [])
174174
@@ -195,7 +195,7 @@ def test_commit_completion_with_transaction_without_transaction_effect(
195195 """
196196
197197 horizon .start_transaction (
198- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
198+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
199199 )
200200 horizon .commit_transaction (2 , UUID (int = 1 ), [])
201201 commit = horizon .complete_commit (3 , UUID (int = 1 ))
@@ -244,7 +244,7 @@ def test_commit_with_transaction_with_effect(
244244 """
245245
246246 horizon .start_transaction (
247- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
247+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
248248 )
249249
250250 prepared_commit = horizon .commit_transaction (2 , UUID (int = 1 ), [
@@ -290,11 +290,11 @@ def test_horizon_movement(
290290 """
291291
292292 horizon .start_transaction (
293- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
293+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
294294 )
295295
296296 horizon .start_transaction (
297- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
297+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
298298 )
299299
300300 horizon .commit_transaction (3 , UUID (int = 1 ), [])
@@ -304,7 +304,7 @@ def test_horizon_movement(
304304 assert len (horizon ) == 1
305305
306306 horizon .start_transaction (
307- 5 , UUID (int = 3 ), Isolation .serializable_read_and_write
307+ 5 , UUID (int = 3 ), IsolationLevel .serializable_read_and_write
308308 )
309309
310310 horizon .commit_transaction (6 , UUID (int = 2 ), [])
@@ -333,13 +333,13 @@ def test_with_sequential_transactions(
333333 """
334334
335335 horizon .start_transaction (
336- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
336+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
337337 )
338338 commit = horizon .commit_transaction (2 , UUID (int = 1 ), [MutatedTuple (tuple_ (1 , "a" ))])
339339 commit1 = horizon .complete_commit (3 , commit .xid )
340340
341341 horizon .start_transaction (
342- 4 , UUID (int = 2 ), Isolation .serializable_read_and_write
342+ 4 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
343343 )
344344 commit = horizon .commit_transaction (5 , UUID (int = 2 ), [MutatedTuple (tuple_ (1 , "b" ))])
345345 commit2 = horizon .complete_commit (6 , commit .xid )
@@ -369,11 +369,11 @@ def test_conflict_by_id_with_left_transaction(
369369 """
370370
371371 horizon .start_transaction (
372- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
372+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
373373 )
374374
375375 horizon .start_transaction (
376- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
376+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
377377 )
378378
379379 commit = horizon .commit_transaction (3 , UUID (int = 1 ), [MutatedTuple (tuple_ (1 , "a" ))])
@@ -408,11 +408,11 @@ def test_conflict_by_id_with_subset_transaction(
408408 """
409409
410410 horizon .start_transaction (
411- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
411+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
412412 )
413413
414414 horizon .start_transaction (
415- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
415+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
416416 )
417417
418418 commit2 = horizon .commit_transaction (3 , UUID (int = 2 ), [MutatedTuple (tuple_ (1 , "b" ))])
@@ -459,13 +459,13 @@ def test_conflict_by_id_with_left_long_distance_transaction(
459459 """
460460
461461 horizon .start_transaction (
462- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
462+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
463463 )
464464 horizon .start_transaction (
465- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
465+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
466466 )
467467 horizon .start_transaction (
468- 3 , UUID (int = 3 ), Isolation .serializable_read_and_write
468+ 3 , UUID (int = 3 ), IsolationLevel .serializable_read_and_write
469469 )
470470
471471 commit = horizon .commit_transaction (4 , UUID (int = 2 ), [MutatedTuple (tuple_ ("y" ))])
@@ -504,17 +504,17 @@ def test_max_len() -> None:
504504 horizon = horizon_ (max_len = 2 , time = 0 , max_transaction_age = 1000 )
505505
506506 horizon .start_transaction (
507- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
507+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
508508 )
509509 assert len (horizon ) == 1
510510
511511 horizon .start_transaction (
512- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
512+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
513513 )
514514 assert len (horizon ) == 2
515515
516516 horizon .start_transaction (
517- 3 , UUID (int = 3 ), Isolation .serializable_read_and_write
517+ 3 , UUID (int = 3 ), IsolationLevel .serializable_read_and_write
518518 )
519519 assert len (horizon ) == 2
520520
@@ -531,22 +531,22 @@ def test_max_transaction_age() -> None:
531531 horizon = horizon_ (max_len = 1000 , time = 0 , max_transaction_age = 2 )
532532
533533 horizon .start_transaction (
534- 1 , UUID (int = 1 ), Isolation .serializable_read_and_write
534+ 1 , UUID (int = 1 ), IsolationLevel .serializable_read_and_write
535535 )
536536 assert len (horizon ) == 1
537537
538538 horizon .start_transaction (
539- 2 , UUID (int = 2 ), Isolation .serializable_read_and_write
539+ 2 , UUID (int = 2 ), IsolationLevel .serializable_read_and_write
540540 )
541541 assert len (horizon ) == 2
542542
543543 horizon .start_transaction (
544- 3 , UUID (int = 3 ), Isolation .serializable_read_and_write
544+ 3 , UUID (int = 3 ), IsolationLevel .serializable_read_and_write
545545 )
546546 assert len (horizon ) == 2
547547
548548 horizon .start_transaction (
549- 4 , UUID (int = 4 ), Isolation .serializable_read_and_write
549+ 4 , UUID (int = 4 ), IsolationLevel .serializable_read_and_write
550550 )
551551 assert len (horizon ) == 2
552552
@@ -584,7 +584,7 @@ def transaction_del(self: Transaction, *args: Any, **kwargs: Any) -> None: # no
584584 time += 1
585585
586586 horizon .start_transaction (
587- time , UUID (int = xid_int ), Isolation .serializable_read_and_write
587+ time , UUID (int = xid_int ), IsolationLevel .serializable_read_and_write
588588 )
589589
590590 assert transaction_counter == 100
@@ -598,7 +598,7 @@ def transaction_del(self: Transaction, *args: Any, **kwargs: Any) -> None: # no
598598
599599 time += 1
600600 horizon .start_transaction (
601- time , UUID (int = 200 ), Isolation .serializable_read_and_write
601+ time , UUID (int = 200 ), IsolationLevel .serializable_read_and_write
602602 )
603603
604604 SerializableTransaction .__init__ = old_transaction_init # type: ignore[method-assign]
0 commit comments