def test_get_profile_raises_error_on_unknown_user(self):
client = GithubClient()
error = github.UnknownObjectException('test error', None)
with mock.patch.object(client.client, 'get_user', side_effect=error):
with self.assertRaises(UnknownProfileError) as cm:
client.get_profile('foobar')
self.assertEqual(str(cm.exception), 'No such GitHub account: foobar')
def test_get_repository_data_retrieves_data(self):
user = mock.MagicMock()
repo_1 = mock.MagicMock(fork=True,
stargazers_count=2,
open_issues_count=4,
watchers_count=6,
language='Python',
topics=['test', 'repositories'])
repo_1.get_commits.return_value.totalCount = 30
repo_2 = mock.MagicMock(
fork=False,
stargazers_count=3,
open_issues_count=0,
watchers_count=0,
language='JavaScript',
topics=None,
)
repo_2.get_commits.return_value.totalCount = 79
user.get_repos.return_value = [repo_1, repo_2]
data = GithubClient._get_repository_data(user)
self.assertIsInstance(data, dict)
self.assertEqual(data['repositories'], {'original': 1, 'forked': 1})
self.assertEqual(data['stars'], 5)
self.assertEqual(data['issues'], 4)
self.assertEqual(data['watchers'], 6)
self.assertEqual(data['commits'], 79)
self.assertCountEqual(data['languages'], ['Python', 'JavaScript'])
self.assertCountEqual(data['topics'], ['test', 'repositories'])
def test_get_user_data_retrieves_data(self):
user = mock.MagicMock()
user.followers = 5
user.following = 2
user.get_starred.return_value.totalCount = 8
data = GithubClient._get_user_data(user)
self.assertEqual(data, {'followers': 5, 'following': 2, 'starred': 8})
def test_get_profile_raises_error_on_rate_limit(self):
client = GithubClient()
error = github.RateLimitExceededException(None, None)
with ExitStack() as stack:
context_managers = (
mock.patch.object(client.client, 'get_user'),
mock.patch.object(client, '_get_user_data', return_value={}),
mock.patch.object(client,
'_get_repository_data',
side_effect=error),
)
for context_manager in context_managers:
stack.enter_context(context_manager)
with self.assertRaises(RateLimitError) as cm:
client.get_profile('foobar')
self.assertEqual(str(cm.exception), 'Exceeded GitHub rate limit')
def test_get_profile_raises_error_on_bad_credentials(self):
client = GithubClient()
error = github.BadCredentialsException(None, None)
with ExitStack() as stack:
context_managers = (
mock.patch.object(client.client, 'get_user',
side_effect=error),
mock.patch.object(client, '_get_user_data', return_value={}),
mock.patch.object(client,
'_get_repository_data',
return_value={}),
)
for context_manager in context_managers:
stack.enter_context(context_manager)
with self.assertRaises(InvalidCredentialsError) as cm:
client.get_profile('foobar')
self.assertEqual(str(cm.exception),
'Cannot authenticate with given credentials')
def test_get_profile_assembles_profile_data(self):
client = GithubClient()
with ExitStack() as stack:
context_managers = (
mock.patch.object(client.client, 'get_user'),
mock.patch.object(client,
'_get_user_data',
return_value={'user': '******'}),
mock.patch.object(client,
'_get_repository_data',
return_value={'repo': 'data'}),
)
for context_manager in context_managers:
stack.enter_context(context_manager)
data = client.get_profile('foobar')
client._get_user_data.assert_called_once_with(
client.client.get_user.return_value)
client._get_repository_data.assert_called_once_with(
client.client.get_user.return_value)
self.assertEqual(data, {'user': '******', 'repo': 'data'})
def test_init_creates_client(self):
client = GithubClient()
self.assertTrue(hasattr(client, 'client'))
self.assertIsInstance(client.client, github.Github)