mirror of
https://github.com/psycopg/psycopg2.git
synced 2025-02-25 21:20:32 +03:00
Unregister test adapters to keep a more precise references count
This commit is contained in:
parent
a44441f5e5
commit
131c6a25e9
|
@ -166,7 +166,11 @@ class AdaptSubclassTest(unittest.TestCase):
|
|||
|
||||
register_adapter(A, lambda a: AsIs("a"))
|
||||
register_adapter(B, lambda b: AsIs("b"))
|
||||
self.assertEqual('b', adapt(C()).getquoted())
|
||||
try:
|
||||
self.assertEqual('b', adapt(C()).getquoted())
|
||||
finally:
|
||||
del psycopg2.extensions.adapters[A, psycopg2.extensions.ISQLQuote]
|
||||
del psycopg2.extensions.adapters[B, psycopg2.extensions.ISQLQuote]
|
||||
|
||||
def test_no_mro_no_joy(self):
|
||||
from psycopg2.extensions import adapt, register_adapter, AsIs
|
||||
|
@ -175,7 +179,11 @@ class AdaptSubclassTest(unittest.TestCase):
|
|||
class B(A): pass
|
||||
|
||||
register_adapter(A, lambda a: AsIs("a"))
|
||||
self.assertRaises(psycopg2.ProgrammingError, adapt, B())
|
||||
try:
|
||||
self.assertRaises(psycopg2.ProgrammingError, adapt, B())
|
||||
finally:
|
||||
del psycopg2.extensions.adapters[A, psycopg2.extensions.ISQLQuote]
|
||||
|
||||
|
||||
def test_suite():
|
||||
return unittest.TestLoader().loadTestsFromName(__name__)
|
||||
|
|
Loading…
Reference in New Issue
Block a user