def test_get_session_invalidated(self): # If clear the global engine, a new engine is used for get_session(). base = sql.Base() session1 = base.get_session() sql.set_global_engine(None) session2 = base.get_session() self.assertIsNot(session1.bind, session2.bind)
def tearDown(self): super(MultiLDAPandSQLIdentity, self).tearDown() self.opt_in_group("identity", domain_config_dir=self.orig_config_dir) self.opt_in_group("identity", domain_specific_drivers_enabled=self.orig_config_domains_enabled) sql.ModelBase.metadata.drop_all(bind=self.engine) self.engine.dispose() sql.set_global_engine(None)
def test_notify_on_set(self): # If call sql.set_global_engine(), notify callbacks get called. cb_mon = CallbackMonitor() sql.register_global_engine_callback(cb_mon.call_this) fake_engine = object() sql.set_global_engine(fake_engine) cb_mon.check()
def test_notify_register_same(self): # If you register the same callback twice, only gets called once. cb_mon = CallbackMonitor() sql.register_global_engine_callback(cb_mon.call_this) sql.register_global_engine_callback(cb_mon.call_this) fake_engine = object() # Note that cb_mon.call_this would raise if it's called twice. sql.set_global_engine(fake_engine) cb_mon.check()
def test_multi_notify(self): # You can also set multiple notify callbacks and they each get called. cb_mon1 = CallbackMonitor() cb_mon2 = CallbackMonitor() sql.register_global_engine_callback(cb_mon1.call_this) sql.register_global_engine_callback(cb_mon2.call_this) fake_engine = object() sql.set_global_engine(fake_engine) cb_mon1.check() cb_mon2.check()
def test_set_same_engine(self): # If you set the global engine to the same engine, callbacks don't get # called. fake_engine = object() sql.set_global_engine(fake_engine) cb_mon = CallbackMonitor(expect_called=False) sql.register_global_engine_callback(cb_mon.call_this) # Note that cb_mon.call_this would raise if it's called. sql.set_global_engine(fake_engine) cb_mon.check()
def test_notify_once(self): # After a callback is called, it's not called again if set global # engine again. cb_mon = CallbackMonitor() sql.register_global_engine_callback(cb_mon.call_this) fake_engine = object() sql.set_global_engine(fake_engine) fake_engine = object() # Note that cb_mon.call_this would raise if it's called again. sql.set_global_engine(fake_engine) cb_mon.check()
def test_callback_throws(self): # If a callback function raises, # a) the caller doesn't know about it, # b) other callbacks are still called cb_mon1 = CallbackMonitor(raise_=True) cb_mon2 = CallbackMonitor() sql.register_global_engine_callback(cb_mon1.call_this) sql.register_global_engine_callback(cb_mon2.call_this) fake_engine = object() sql.set_global_engine(fake_engine) cb_mon1.check() cb_mon2.check()
def tearDown(self): sql.set_global_engine(None) super(KcMasterSqlTestCase, self).tearDown()
def tearDown(self): sql.set_global_engine(None) super(TestGlobalEngine, self).tearDown()
def tearDown(self): sql.set_global_engine(None) super(SqlTests, self).tearDown()
def tearDown(self): sql.ModelBase.metadata.drop_all(bind=self.engine) self.engine.dispose() sql.set_global_engine(None) super(SqlTests, self).tearDown()
def tearDown(self): sql.ModelBase.metadata.drop_all(bind=self.engine) self.engine.dispose() sql.set_global_engine(None) super(LdapIdentitySqlAssignment, self).tearDown()