@@ -191,37 +191,37 @@ def test_get_expired(self):
191191 self .assertFalse (pool ._sessions .full ())
192192
193193 def test_get_empty_default_timeout (self ):
194- from six . moves . queue import Empty
194+ import queue
195195
196196 pool = self ._make_one (size = 1 )
197- queue = pool ._sessions = _Queue ()
197+ session_queue = pool ._sessions = _Queue ()
198198
199- with self .assertRaises (Empty ):
199+ with self .assertRaises (queue . Empty ):
200200 pool .get ()
201201
202- self .assertEqual (queue ._got , {"block" : True , "timeout" : 10 })
202+ self .assertEqual (session_queue ._got , {"block" : True , "timeout" : 10 })
203203
204204 def test_get_empty_explicit_timeout (self ):
205- from six . moves . queue import Empty
205+ import queue
206206
207207 pool = self ._make_one (size = 1 , default_timeout = 0.1 )
208- queue = pool ._sessions = _Queue ()
208+ session_queue = pool ._sessions = _Queue ()
209209
210- with self .assertRaises (Empty ):
210+ with self .assertRaises (queue . Empty ):
211211 pool .get (timeout = 1 )
212212
213- self .assertEqual (queue ._got , {"block" : True , "timeout" : 1 })
213+ self .assertEqual (session_queue ._got , {"block" : True , "timeout" : 1 })
214214
215215 def test_put_full (self ):
216- from six . moves . queue import Full
216+ import queue
217217
218218 pool = self ._make_one (size = 4 )
219219 database = _Database ("name" )
220220 SESSIONS = [_Session (database )] * 4
221221 database ._sessions .extend (SESSIONS )
222222 pool .bind (database )
223223
224- with self .assertRaises (Full ):
224+ with self .assertRaises (queue . Full ):
225225 pool .put (_Session (database ))
226226
227227 self .assertTrue (pool ._sessions .full ())
@@ -481,37 +481,37 @@ def test_get_hit_w_ping_expired(self):
481481 self .assertFalse (pool ._sessions .full ())
482482
483483 def test_get_empty_default_timeout (self ):
484- from six . moves . queue import Empty
484+ import queue
485485
486486 pool = self ._make_one (size = 1 )
487- queue = pool ._sessions = _Queue ()
487+ session_queue = pool ._sessions = _Queue ()
488488
489- with self .assertRaises (Empty ):
489+ with self .assertRaises (queue . Empty ):
490490 pool .get ()
491491
492- self .assertEqual (queue ._got , {"block" : True , "timeout" : 10 })
492+ self .assertEqual (session_queue ._got , {"block" : True , "timeout" : 10 })
493493
494494 def test_get_empty_explicit_timeout (self ):
495- from six . moves . queue import Empty
495+ import queue
496496
497497 pool = self ._make_one (size = 1 , default_timeout = 0.1 )
498- queue = pool ._sessions = _Queue ()
498+ session_queue = pool ._sessions = _Queue ()
499499
500- with self .assertRaises (Empty ):
500+ with self .assertRaises (queue . Empty ):
501501 pool .get (timeout = 1 )
502502
503- self .assertEqual (queue ._got , {"block" : True , "timeout" : 1 })
503+ self .assertEqual (session_queue ._got , {"block" : True , "timeout" : 1 })
504504
505505 def test_put_full (self ):
506- from six . moves . queue import Full
506+ import queue
507507
508508 pool = self ._make_one (size = 4 )
509509 database = _Database ("name" )
510510 SESSIONS = [_Session (database )] * 4
511511 database ._sessions .extend (SESSIONS )
512512 pool .bind (database )
513513
514- with self .assertRaises (Full ):
514+ with self .assertRaises (queue . Full ):
515515 pool .put (_Session (database ))
516516
517517 self .assertTrue (pool ._sessions .full ())
@@ -522,7 +522,7 @@ def test_put_non_full(self):
522522 from google .cloud .spanner_v1 import pool as MUT
523523
524524 pool = self ._make_one (size = 1 )
525- queue = pool ._sessions = _Queue ()
525+ session_queue = pool ._sessions = _Queue ()
526526
527527 now = datetime .datetime .utcnow ()
528528 database = _Database ("name" )
@@ -531,8 +531,8 @@ def test_put_non_full(self):
531531 with _Monkey (MUT , _NOW = lambda : now ):
532532 pool .put (session )
533533
534- self .assertEqual (len (queue ._items ), 1 )
535- ping_after , queued = queue ._items [0 ]
534+ self .assertEqual (len (session_queue ._items ), 1 )
535+ ping_after , queued = session_queue ._items [0 ]
536536 self .assertEqual (ping_after , now + datetime .timedelta (seconds = 3000 ))
537537 self .assertIs (queued , session )
538538
@@ -690,39 +690,39 @@ def test_bind_w_timestamp_race(self):
690690 self .assertTrue (pool ._pending_sessions .empty ())
691691
692692 def test_put_full (self ):
693- from six . moves . queue import Full
693+ import queue
694694
695695 pool = self ._make_one (size = 4 )
696696 database = _Database ("name" )
697697 SESSIONS = [_Session (database ) for _ in range (4 )]
698698 database ._sessions .extend (SESSIONS )
699699 pool .bind (database )
700700
701- with self .assertRaises (Full ):
701+ with self .assertRaises (queue . Full ):
702702 pool .put (_Session (database ))
703703
704704 self .assertTrue (pool ._sessions .full ())
705705
706706 def test_put_non_full_w_active_txn (self ):
707707 pool = self ._make_one (size = 1 )
708- queue = pool ._sessions = _Queue ()
708+ session_queue = pool ._sessions = _Queue ()
709709 pending = pool ._pending_sessions = _Queue ()
710710 database = _Database ("name" )
711711 session = _Session (database )
712712 txn = session .transaction ()
713713
714714 pool .put (session )
715715
716- self .assertEqual (len (queue ._items ), 1 )
717- _ , queued = queue ._items [0 ]
716+ self .assertEqual (len (session_queue ._items ), 1 )
717+ _ , queued = session_queue ._items [0 ]
718718 self .assertIs (queued , session )
719719
720720 self .assertEqual (len (pending ._items ), 0 )
721721 txn .begin .assert_not_called ()
722722
723723 def test_put_non_full_w_committed_txn (self ):
724724 pool = self ._make_one (size = 1 )
725- queue = pool ._sessions = _Queue ()
725+ session_queue = pool ._sessions = _Queue ()
726726 pending = pool ._pending_sessions = _Queue ()
727727 database = _Database ("name" )
728728 session = _Session (database )
@@ -731,7 +731,7 @@ def test_put_non_full_w_committed_txn(self):
731731
732732 pool .put (session )
733733
734- self .assertEqual (len (queue ._items ), 0 )
734+ self .assertEqual (len (session_queue ._items ), 0 )
735735
736736 self .assertEqual (len (pending ._items ), 1 )
737737 self .assertIs (pending ._items [0 ], session )
@@ -740,14 +740,14 @@ def test_put_non_full_w_committed_txn(self):
740740
741741 def test_put_non_full (self ):
742742 pool = self ._make_one (size = 1 )
743- queue = pool ._sessions = _Queue ()
743+ session_queue = pool ._sessions = _Queue ()
744744 pending = pool ._pending_sessions = _Queue ()
745745 database = _Database ("name" )
746746 session = _Session (database )
747747
748748 pool .put (session )
749749
750- self .assertEqual (len (queue ._items ), 0 )
750+ self .assertEqual (len (session_queue ._items ), 0 )
751751 self .assertEqual (len (pending ._items ), 1 )
752752 self .assertIs (pending ._items [0 ], session )
753753
@@ -924,13 +924,13 @@ def full(self):
924924 return len (self ._items ) >= self ._size
925925
926926 def get (self , ** kwargs ):
927- from six . moves . queue import Empty
927+ import queue
928928
929929 self ._got = kwargs
930930 try :
931931 return self ._items .pop ()
932932 except IndexError :
933- raise Empty ()
933+ raise queue . Empty ()
934934
935935 def put (self , item , ** kwargs ):
936936 self ._put = kwargs
0 commit comments