def test_get_user_from_session_returns_correct_user(self): session = UserSession(self.db) user = '******' session_id = '23423534' expected_date = datetime.datetime.utcnow() self.db.session.insert_one({'user': user, 'session_id': session_id, 'expiry': expected_date}) self.assertEqual(session.get_user_from_session(session_id), user)
def test_get_user_from_session_user_is_none_returns_none(self): session = UserSession(self.db) self.assertIsNone(session.get_user_from_session(None))
def test_get_user_from_session_user_doesnt_exist_returns_none(self): session = UserSession(self.db) self.assertIsNone(session.get_user_from_session('0000001'))