class SessionManagerTest(TestCase):
def setUp(self):
self.storage = Middleware.get_middleware_by_uri(
'mongodb-default-testsession://'
)
self.collection = MongoCollection(self.storage._backend)
self.manager = Session(collection=self.collection)
self.user = '******'
def tearDown(self):
self.collection.remove()
def test_keep_alive(self):
self.manager.session_start(self.user)
sleep(1)
got = self.manager.keep_alive(self.user)
session = self.collection.find_one({'_id': self.user})
self.assertTrue(isinstance(session, dict))
self.assertEqual(got, session['last_check'])
def test_session_start(self):
got = self.manager.session_start(self.user)
session = self.collection.find_one({'_id': self.user})
self.assertTrue(isinstance(session, dict))
self.assertTrue(session['active'])
self.assertEqual(got, session['session_start'])
def test_session_start_already_started(self):
self.test_session_start()
got = self.manager.session_start(self.user)
self.assertTrue(got is None)
def test_is_session_active(self):
self.assertFalse(self.manager.is_session_active(self.user))
self.manager.session_start(self.user)
self.assertTrue(self.manager.is_session_active(self.user))
def test_sessions_close(self):
got = self.manager.session_start(self.user)
self.manager.alive_session_duration = 0
self.assertTrue(got is not None)
sessions = self.manager.sessions_close()
self.assertTrue(len(sessions) > 0)
self.assertEqual(got, sessions[0]['last_check'])
class SessionManagerTest(TestCase):
def setUp(self):
self.storage = Middleware.get_middleware_by_uri(
'mongodb-default-testsession://')
self.collection = MongoCollection(self.storage._backend)
self.manager = Session(collection=self.collection)
self.user = '******'
def tearDown(self):
self.collection.remove()
def test_keep_alive(self):
self.manager.session_start(self.user)
sleep(1)
got = self.manager.keep_alive(self.user)
session = self.collection.find_one({'_id': self.user})
self.assertTrue(isinstance(session, dict))
self.assertEqual(got, session['last_check'])
def test_session_start(self):
got = self.manager.session_start(self.user)
session = self.collection.find_one({'_id': self.user})
self.assertTrue(isinstance(session, dict))
self.assertTrue(session['active'])
self.assertEqual(got, session['session_start'])
def test_session_start_already_started(self):
self.test_session_start()
got = self.manager.session_start(self.user)
self.assertTrue(got is None)
def test_is_session_active(self):
self.assertFalse(self.manager.is_session_active(self.user))
self.manager.session_start(self.user)
self.assertTrue(self.manager.is_session_active(self.user))
def test_sessions_close(self):
got = self.manager.session_start(self.user)
self.manager.alive_session_duration = 0
self.assertTrue(got is not None)
sessions = self.manager.sessions_close()
self.assertTrue(len(sessions) > 0)
self.assertEqual(got, sessions[0]['last_check'])
