def test_registry(self): from zope.interface import Interface, implementer from aiorm import registry class IAdaptable(Interface): pass class IAdapter(Interface): pass class IAdapter2(Interface): pass @implementer(IAdaptable) class Adaptable: pass @implementer(IAdapter) class Adapter: pass @implementer(IAdapter2) class Adapter2: pass self.assertRaises(NotImplementedError, registry.get, IAdapter) registry.register(Adapter, adapt=IAdaptable) # introspect iface self.assertRaises(ValueError, registry.register, Adapter, adapt=IAdaptable) registry.register(Adaptable, IAdaptable, adapt=IAdaptable) # explicit registry.register(Adapter2, adapt=Adaptable) # 3 "types" accepted for the adapt parameter self.assertEqual(registry.get(IAdapter, adapt=IAdaptable), Adapter) self.assertEqual(registry.get(IAdapter, adapt=Adaptable), Adapter) self.assertEqual(registry.get(IAdapter, adapt=Adaptable()), Adapter) self.assertEqual(registry.get(IAdaptable, adapt=IAdaptable), Adaptable) self.assertEqual(registry.get(IAdaptable, adapt=Adaptable), Adaptable) registry.unregister(Adapter, IAdaptable) self.assertRaises(NotImplementedError, registry.get, IAdapter, adapt=IAdaptable) registry.unregister(Adapter) registry.unregister(Adapter2)
def test_registry(self): from zope.interface import Interface, implementer from aiorm import registry class IAdaptable(Interface): pass class IAdapter(Interface): pass class IAdapter2(Interface): pass @implementer(IAdaptable) class Adaptable: pass @implementer(IAdapter) class Adapter: pass @implementer(IAdapter2) class Adapter2: pass self.assertRaises(NotImplementedError, registry.get, IAdapter) registry.register(Adapter, adapt=IAdaptable) # introspect iface self.assertRaises(ValueError, registry.register, Adapter, adapt=IAdaptable) registry.register(Adaptable, IAdaptable, adapt=IAdaptable) # explicit registry.register(Adapter2, adapt=Adaptable) # 3 "types" accepted for the adapt parameter self.assertEqual(registry.get(IAdapter, adapt=IAdaptable), Adapter) self.assertEqual(registry.get(IAdapter, adapt=Adaptable), Adapter) self.assertEqual(registry.get(IAdapter, adapt=Adaptable()), Adapter) self.assertEqual(registry.get(IAdaptable, adapt=IAdaptable), Adaptable) self.assertEqual(registry.get(IAdaptable, adapt=Adaptable), Adaptable) registry.unregister(Adapter, IAdaptable) self.assertRaises(NotImplementedError, registry.get, IAdapter, adapt=IAdaptable) registry.unregister(Adapter) registry.unregister(Adapter2)
def tearDown(self): registry.unregister(DummyDialect) DummyDialect._instance = None
def tearDownClass(cls): registry.unregister(DummyDriver)
def aioTearDown(self): from aiorm import registry from aiorm.orm.dialect.postgresql import Dialect, CreateTableDialect registry.unregister(Dialect) registry.unregister(CreateTableDialect) yield from registry.disconnect('sample')
def aioTearDown(self): from aiorm import registry from aiorm.orm.dialect.postgresql import Dialect, CreateTableDialect registry.unregister(Dialect) registry.unregister(CreateTableDialect) yield from registry.disconnect('sample')
def tearDownClass(cls): registry.unregister(DummyDriver)