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'))
