def test_profile_with_expired_session(self, fetch_mock): db_user = User( id=1, name='Joao', email='*****@*****.**', password='******', token=hashlib.sha1('token').hexdigest(), last_login=datetime(2016, 2, 15, 19, 20, 21), ) fetch_mock.return_value = db_user with self.assertRaises(InvalidSession): get_profile('1', 'token')
def profile(user_id): if 'X-Token' not in request.headers: return api_error('Obrigatório uso do header X-Token para usar este endpoint', 400) token = request.headers['X-Token'] try: user_data = get_profile(user_id, token) except AuthorizationError as exc: return api_error(exc.message, 401) return user_data, 200
def test_successful_profile(self, fetch_mock): db_user = User( id=1, name='Joao', email='*****@*****.**', password='******', token=hashlib.sha1('token').hexdigest(), created=datetime(2016, 2, 15, 19, 20, 21), modified=datetime(2016, 2, 15, 19, 20, 21), last_login=datetime(2016, 2, 15, 19, 20, 21), ) fetch_mock.return_value = db_user user_data = get_profile('1', 'token') self.assertEqual(user_data['id'], db_user.id)
def test_profile_with_inexistent_token(self, fetch_mock): fetch_mock.side_effect = NoResultFound with self.assertRaises(InvalidToken): get_profile('1', 'token')