def testGetTravisTokenError(self): self.mock_request.return_value = (403, 'Unauthorized.') with self.assertRaisesRegex( travis.TravisApiError, (r'Travis API Exception \(HTTP 403\): ' r'Travis Auth API request failed with response: Unauthorized\.')): travis.TravisApi()
def testFetchBuildsError(self, unused_mock_get_travis_token): travis_api = travis.TravisApi() self.mock_request.return_value = (500, 'Server error.') with self.assertRaisesRegex( travis.TravisApiError, (r'Travis API Exception \(HTTP 500\): ' r'Travis Builds API request failed with response: Server error\.' )): travis_api.fetch_builds()
def testGetTravisToken(self): self.mock_request.return_value = (200, { 'access_token': '__travis_token__' }) travis_api = travis.TravisApi() self.assertEqual(travis_api.api_token, '__travis_token__') self.mock_request.assert_called_once_with( travis_api.client, 'POST', '/auth/github?github_token=__github_token__', mock.ANY, mock.ANY)
def testFetchBuildsPage(self, unused_mock_get_travis_token): self.mock_request.return_value = (200, {'builds': [self.build_json]}) travis_api = travis.TravisApi() fetched_builds = travis_api.fetch_builds(page_num=1) # Comparing the Build directly fails due to internal state properties used # by SQLAlchemy. self.assertEqual(fetched_builds[0].started_at, self.build.started_at) self.assertEqual(fetched_builds[0].id, self.build.id) self.assertEqual(fetched_builds[0].number, self.build.number) self.assertEqual(fetched_builds[0].duration, self.build.duration) self.assertEqual(fetched_builds[0].commit_hash, self.build.commit_hash) self.assertEqual(fetched_builds[0].state, self.build.state)
def __init__(self): self.travis = travis.TravisApi() self.session = db.Session() self.cursor = None