12
12
import pytest
13
13
14
14
from dbutils .steady_db import SteadyDBConnection , SteadyDBCursor
15
- from dbutils .steady_db import connect as SteadyDBconnect
15
+ from dbutils .steady_db import connect as steady_db_connect
16
16
17
17
from . import mock_db as dbapi
18
18
@@ -107,13 +107,13 @@ def test_broken_connection():
107
107
SteadyDBConnection (None )
108
108
with pytest .raises (TypeError ):
109
109
SteadyDBCursor (None )
110
- db = SteadyDBconnect (dbapi , database = 'ok' )
110
+ db = steady_db_connect (dbapi , database = 'ok' )
111
111
for i in range (3 ):
112
112
db .close ()
113
113
del db
114
114
with pytest .raises (dbapi .OperationalError ):
115
- SteadyDBconnect (dbapi , database = 'error' )
116
- db = SteadyDBconnect (dbapi , database = 'ok' )
115
+ steady_db_connect (dbapi , database = 'error' )
116
+ db = steady_db_connect (dbapi , database = 'ok' )
117
117
cursor = db .cursor ()
118
118
for i in range (3 ):
119
119
cursor .close ()
@@ -126,7 +126,7 @@ def test_broken_connection():
126
126
127
127
@pytest .mark .parametrize ("closeable" , [False , True ])
128
128
def test_close (closeable ):
129
- db = SteadyDBconnect (dbapi , closeable = closeable )
129
+ db = steady_db_connect (dbapi , closeable = closeable )
130
130
assert db ._con .valid
131
131
db .close ()
132
132
assert closeable ^ db ._con .valid
@@ -138,8 +138,8 @@ def test_close(closeable):
138
138
assert not db ._con .valid
139
139
140
140
141
- def test_connection ():
142
- db = SteadyDBconnect (
141
+ def test_connection (): # noqa: PLR0915
142
+ db = steady_db_connect (
143
143
dbapi , 0 , None , None , None , True ,
144
144
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
145
145
assert isinstance (db , SteadyDBConnection )
@@ -259,7 +259,7 @@ def test_connection():
259
259
260
260
261
261
def test_connection_context_handler ():
262
- db = SteadyDBconnect (
262
+ db = steady_db_connect (
263
263
dbapi , 0 , None , None , None , True ,
264
264
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
265
265
assert db ._con .session == []
@@ -278,7 +278,7 @@ def test_connection_context_handler():
278
278
279
279
280
280
def test_cursor_context_handler ():
281
- db = SteadyDBconnect (
281
+ db = steady_db_connect (
282
282
dbapi , 0 , None , None , None , True ,
283
283
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
284
284
assert db ._con .open_cursors == 0
@@ -290,7 +290,7 @@ def test_cursor_context_handler():
290
290
291
291
292
292
def test_cursor_as_iterator_provided ():
293
- db = SteadyDBconnect (
293
+ db = steady_db_connect (
294
294
dbapi , 0 , None , None , None , True ,
295
295
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
296
296
assert db ._con .open_cursors == 0
@@ -309,7 +309,7 @@ def test_cursor_as_iterator_provided():
309
309
310
310
311
311
def test_cursor_as_iterator_created ():
312
- db = SteadyDBconnect (
312
+ db = steady_db_connect (
313
313
dbapi , 0 , None , None , None , True ,
314
314
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
315
315
assert db ._con .open_cursors == 0
@@ -322,10 +322,10 @@ def test_cursor_as_iterator_created():
322
322
323
323
324
324
def test_connection_creator_function ():
325
- db1 = SteadyDBconnect (
325
+ db1 = steady_db_connect (
326
326
dbapi , 0 , None , None , None , True ,
327
327
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
328
- db2 = SteadyDBconnect (
328
+ db2 = steady_db_connect (
329
329
dbapi .connect , 0 , None , None , None , True ,
330
330
'SteadyDBTestDB' , user = 'SteadyDBTestUser' )
331
331
assert db1 .dbapi () == db2 .dbapi ()
@@ -338,7 +338,7 @@ def test_connection_creator_function():
338
338
339
339
340
340
def test_connection_maxusage ():
341
- db = SteadyDBconnect (dbapi , 10 )
341
+ db = steady_db_connect (dbapi , 10 )
342
342
cursor = db .cursor ()
343
343
for i in range (100 ):
344
344
cursor .execute (f'select test{ i } ' )
@@ -389,7 +389,7 @@ def test_connection_maxusage():
389
389
390
390
391
391
def test_connection_setsession ():
392
- db = SteadyDBconnect (dbapi , 3 , ('set time zone' , 'set datestyle' ))
392
+ db = steady_db_connect (dbapi , 3 , ('set time zone' , 'set datestyle' ))
393
393
assert hasattr (db , '_usage' )
394
394
assert db ._usage == 0
395
395
assert hasattr (db ._con , 'open_cursors' )
@@ -448,29 +448,29 @@ def test_connection_setsession():
448
448
449
449
450
450
def test_connection_failures ():
451
- db = SteadyDBconnect (dbapi )
451
+ db = steady_db_connect (dbapi )
452
452
db .close ()
453
453
db .cursor ()
454
- db = SteadyDBconnect (dbapi , failures = dbapi .InternalError )
454
+ db = steady_db_connect (dbapi , failures = dbapi .InternalError )
455
455
db .close ()
456
456
db .cursor ()
457
- db = SteadyDBconnect (dbapi , failures = dbapi .OperationalError )
457
+ db = steady_db_connect (dbapi , failures = dbapi .OperationalError )
458
458
db .close ()
459
459
with pytest .raises (dbapi .InternalError ):
460
460
db .cursor ()
461
- db = SteadyDBconnect (dbapi , failures = (
461
+ db = steady_db_connect (dbapi , failures = (
462
462
dbapi .OperationalError , dbapi .InterfaceError ))
463
463
db .close ()
464
464
with pytest .raises (dbapi .InternalError ):
465
465
db .cursor ()
466
- db = SteadyDBconnect (dbapi , failures = (
466
+ db = steady_db_connect (dbapi , failures = (
467
467
dbapi .OperationalError , dbapi .InterfaceError , dbapi .InternalError ))
468
468
db .close ()
469
469
db .cursor ()
470
470
471
471
472
472
def test_connection_failure_error ():
473
- db = SteadyDBconnect (dbapi )
473
+ db = steady_db_connect (dbapi )
474
474
cursor = db .cursor ()
475
475
db .close ()
476
476
cursor .execute ('select test' )
@@ -481,7 +481,7 @@ def test_connection_failure_error():
481
481
482
482
483
483
def test_connection_set_sizes ():
484
- db = SteadyDBconnect (dbapi )
484
+ db = steady_db_connect (dbapi )
485
485
cursor = db .cursor ()
486
486
cursor .execute ('get sizes' )
487
487
result = cursor .fetchone ()
@@ -510,15 +510,15 @@ def test_connection_ping_check():
510
510
con_cls = dbapi .Connection
511
511
con_cls .has_ping = False
512
512
con_cls .num_pings = 0
513
- db = SteadyDBconnect (dbapi )
513
+ db = steady_db_connect (dbapi )
514
514
db .cursor ().execute ('select test' )
515
515
assert con_cls .num_pings == 0
516
516
db .close ()
517
517
db .cursor ().execute ('select test' )
518
518
assert con_cls .num_pings == 0
519
519
assert db ._ping_check () is None
520
520
assert con_cls .num_pings == 1
521
- db = SteadyDBconnect (dbapi , ping = 7 )
521
+ db = steady_db_connect (dbapi , ping = 7 )
522
522
db .cursor ().execute ('select test' )
523
523
assert con_cls .num_pings == 2
524
524
db .close ()
@@ -527,15 +527,15 @@ def test_connection_ping_check():
527
527
assert db ._ping_check () is None
528
528
assert con_cls .num_pings == 2
529
529
con_cls .has_ping = True
530
- db = SteadyDBconnect (dbapi )
530
+ db = steady_db_connect (dbapi )
531
531
db .cursor ().execute ('select test' )
532
532
assert con_cls .num_pings == 2
533
533
db .close ()
534
534
db .cursor ().execute ('select test' )
535
535
assert con_cls .num_pings == 2
536
536
assert db ._ping_check ()
537
537
assert con_cls .num_pings == 3
538
- db = SteadyDBconnect (dbapi , ping = 1 )
538
+ db = steady_db_connect (dbapi , ping = 1 )
539
539
db .cursor ().execute ('select test' )
540
540
assert con_cls .num_pings == 3
541
541
db .close ()
@@ -546,13 +546,13 @@ def test_connection_ping_check():
546
546
db .close ()
547
547
assert db ._ping_check ()
548
548
assert con_cls .num_pings == 5
549
- db = SteadyDBconnect (dbapi , ping = 7 )
549
+ db = steady_db_connect (dbapi , ping = 7 )
550
550
db .cursor ().execute ('select test' )
551
551
assert con_cls .num_pings == 7
552
552
db .close ()
553
553
db .cursor ().execute ('select test' )
554
554
assert con_cls .num_pings == 9
555
- db = SteadyDBconnect (dbapi , ping = 3 )
555
+ db = steady_db_connect (dbapi , ping = 3 )
556
556
assert con_cls .num_pings == 9
557
557
db .cursor ()
558
558
assert con_cls .num_pings == 10
@@ -561,7 +561,7 @@ def test_connection_ping_check():
561
561
assert con_cls .num_pings == 11
562
562
cursor .execute ('select test' )
563
563
assert con_cls .num_pings == 11
564
- db = SteadyDBconnect (dbapi , ping = 5 )
564
+ db = steady_db_connect (dbapi , ping = 5 )
565
565
assert con_cls .num_pings == 11
566
566
db .cursor ()
567
567
assert con_cls .num_pings == 11
@@ -575,7 +575,7 @@ def test_connection_ping_check():
575
575
assert con_cls .num_pings == 12
576
576
cursor .execute ('select test' )
577
577
assert con_cls .num_pings == 13
578
- db = SteadyDBconnect (dbapi , ping = 7 )
578
+ db = steady_db_connect (dbapi , ping = 7 )
579
579
assert con_cls .num_pings == 13
580
580
db .cursor ()
581
581
assert con_cls .num_pings == 14
@@ -597,7 +597,7 @@ def test_connection_ping_check():
597
597
598
598
599
599
def test_begin_transaction ():
600
- db = SteadyDBconnect (dbapi , database = 'ok' )
600
+ db = steady_db_connect (dbapi , database = 'ok' )
601
601
cursor = db .cursor ()
602
602
cursor .close ()
603
603
cursor .execute ('select test12' )
@@ -624,7 +624,7 @@ def test_begin_transaction():
624
624
625
625
626
626
def test_with_begin_extension ():
627
- db = SteadyDBconnect (dbapi , database = 'ok' )
627
+ db = steady_db_connect (dbapi , database = 'ok' )
628
628
db ._con ._begin_called_with = None
629
629
630
630
def begin (a , b = None , c = 7 ):
@@ -639,7 +639,7 @@ def begin(a, b=None, c=7):
639
639
640
640
641
641
def test_cancel_transaction ():
642
- db = SteadyDBconnect (dbapi , database = 'ok' )
642
+ db = steady_db_connect (dbapi , database = 'ok' )
643
643
cursor = db .cursor ()
644
644
db .begin ()
645
645
cursor .execute ('select test14' )
@@ -650,7 +650,7 @@ def test_cancel_transaction():
650
650
651
651
652
652
def test_with_cancel_extension ():
653
- db = SteadyDBconnect (dbapi , database = 'ok' )
653
+ db = steady_db_connect (dbapi , database = 'ok' )
654
654
db ._con ._cancel_called = None
655
655
656
656
def cancel ():
@@ -666,20 +666,20 @@ def cancel():
666
666
667
667
668
668
def test_reset_transaction ():
669
- db = SteadyDBconnect (dbapi , database = 'ok' )
669
+ db = steady_db_connect (dbapi , database = 'ok' )
670
670
db .begin ()
671
671
assert not db ._con .session
672
672
db .close ()
673
673
assert not db ._con .session
674
- db = SteadyDBconnect (dbapi , database = 'ok' , closeable = False )
674
+ db = steady_db_connect (dbapi , database = 'ok' , closeable = False )
675
675
db .begin ()
676
676
assert not db ._con .session
677
677
db .close ()
678
678
assert db ._con .session == ['rollback' ]
679
679
680
680
681
681
def test_commit_error ():
682
- db = SteadyDBconnect (dbapi , database = 'ok' )
682
+ db = steady_db_connect (dbapi , database = 'ok' )
683
683
db .begin ()
684
684
assert not db ._con .session
685
685
assert db ._con .valid
@@ -703,7 +703,7 @@ def test_commit_error():
703
703
704
704
705
705
def test_rollback_error ():
706
- db = SteadyDBconnect (dbapi , database = 'ok' )
706
+ db = steady_db_connect (dbapi , database = 'ok' )
707
707
db .begin ()
708
708
assert not db ._con .session
709
709
assert db ._con .valid
0 commit comments