def test_get_token(self): """Ensure token is present""" token = self.token_dao.insert_user_token(self.user) token_gen = token.generate_auth_token() valid_token = Token.get_auth_token(token_gen) self.assertEqual(valid_token.to_json(), token.to_json())
def test_expire_token(self): """Ensure token is expired""" invalid_token = Token.get_auth_token("some token") self.assertIsNone(invalid_token)