def test_get_engine_forked(self, create_engine):
s = SessionManager()
s._after_fork()
engine = s.get_engine('dburi', foo=1)
create_engine.assert_called_with('dburi', foo=1)
assert engine is create_engine()
engine2 = s.get_engine('dburi', foo=1)
assert engine2 is engine
def test_get_engine_forked(self, create_engine):
s = SessionManager()
s._after_fork()
engine = s.get_engine('dburi', foo=1)
create_engine.assert_called_with('dburi', foo=1)
assert engine is create_engine()
engine2 = s.get_engine('dburi', foo=1)
assert engine2 is engine
def test_get_engine_forked(self, create_engine):
s = SessionManager()
s._after_fork()
engine = s.get_engine("dburi", foo=1)
create_engine.assert_called_with("dburi", foo=1)
self.assertIs(engine, create_engine())
engine2 = s.get_engine("dburi", foo=1)
self.assertIs(engine2, engine)
def test_create_session_forked(self, sessionmaker):
s = SessionManager()
s.get_engine = Mock(name="get_engine")
s._after_fork()
engine, session = s.create_session("dburi", short_lived_sessions=True)
sessionmaker.assert_called_with(bind=s.get_engine())
self.assertIs(session, sessionmaker())
sessionmaker.return_value = Mock(name="new")
engine, session2 = s.create_session("dburi", short_lived_sessions=True)
sessionmaker.assert_called_with(bind=s.get_engine())
self.assertIsNot(session2, session)
sessionmaker.return_value = Mock(name="new2")
engine, session3 = s.create_session("dburi", short_lived_sessions=False)
sessionmaker.assert_called_with(bind=s.get_engine())
self.assertIs(session3, session2)
def test_create_session_forked(self, sessionmaker):
s = SessionManager()
s.get_engine = Mock(name='get_engine')
s._after_fork()
engine, session = s.create_session('dburi', short_lived_sessions=True)
sessionmaker.assert_called_with(bind=s.get_engine())
assert session is sessionmaker()
sessionmaker.return_value = Mock(name='new')
engine, session2 = s.create_session('dburi', short_lived_sessions=True)
sessionmaker.assert_called_with(bind=s.get_engine())
assert session2 is not session
sessionmaker.return_value = Mock(name='new2')
engine, session3 = s.create_session('dburi',
short_lived_sessions=False)
sessionmaker.assert_called_with(bind=s.get_engine())
assert session3 is session2
def test_create_session_forked(self, sessionmaker):
s = SessionManager()
s.get_engine = Mock(name='get_engine')
s._after_fork()
engine, session = s.create_session('dburi', short_lived_sessions=True)
sessionmaker.assert_called_with(bind=s.get_engine())
assert session is sessionmaker()
sessionmaker.return_value = Mock(name='new')
engine, session2 = s.create_session('dburi', short_lived_sessions=True)
sessionmaker.assert_called_with(bind=s.get_engine())
assert session2 is not session
sessionmaker.return_value = Mock(name='new2')
engine, session3 = s.create_session(
'dburi', short_lived_sessions=False)
sessionmaker.assert_called_with(bind=s.get_engine())
assert session3 is session2
def test_after_fork(self):
s = SessionManager()
assert not s.forked
s._after_fork()
assert s.forked
def test_after_fork(self):
s = SessionManager()
assert not s.forked
s._after_fork()
assert s.forked
def test_after_fork(self):
s = SessionManager()
self.assertFalse(s.forked)
s._after_fork()
self.assertTrue(s.forked)
def test_after_fork(self):
s = SessionManager()
self.assertFalse(s.forked)
s._after_fork()
self.assertTrue(s.forked)
