def test_get_user_name_and_id(self):
# Given
# mock the post to get the access token
self.mock_requests.post.return_value = mock.MagicMock(
status_code=200, text="access_token=123456")
# mock the get to get the user info
mock_response = mock.MagicMock(status_code=200)
self.mock_requests.get.return_value = mock_response
mock_response.json.return_value = {
"name": "tommy tester",
"id": 123456
}
# When
github_user = GithubUser(
"test-code",
client_id=self.test_client_id,
client_secret=self.test_client_secret,
)
user_name_and_id = github_user._get_user_name_and_id()
# Then
assert user_name_and_id == {
"first_name": "tommy",
"last_name": "tester",
"github_user_id": 123456,
}
def test_get_primary_email(self):
# Given
# mock the post to get the access token
self.mock_requests.post.return_value = mock.MagicMock(
status_code=200, text="access_token=123456")
# mock the request to get the user info
mock_response = mock.MagicMock(status_code=200)
self.mock_requests.get.return_value = mock_response
verified_emails = [{
"email": f"tommy_tester@example_{i}.com",
"verified": True,
"visibility": None,
"primary": False,
} for i in range(5)]
# set one of the verified emails to be the primary
verified_emails[3]["primary"] = True
mock_response.json.return_value = verified_emails
# When
github_user = GithubUser(
"test-code",
client_id=self.test_client_id,
client_secret=self.test_client_secret,
)
primary_email = github_user._get_primary_email()
# Then
assert primary_email == verified_emails[3]["email"]
def test_get_access_token_fail_non_200(self):
# Given
invalid_code = "invalid"
status_code = 400
self.mock_requests.post.return_value = mock.MagicMock(
status_code=status_code)
# When
with pytest.raises(GithubError) as e:
GithubUser(invalid_code,
client_id=self.test_client_id,
client_secret=self.test_client_secret)
# Then - exception raised
assert NON_200_ERROR_MESSAGE.format(status_code) in str(e)
def test_get_access_token_fail_token_expired(self):
# Given
invalid_code = "invalid"
error_description = "there+was+an+error"
self.mock_requests.post.return_value = mock.MagicMock(
text=
f"error=bad_verification_code&error_description={error_description}",
status_code=200)
# When
with pytest.raises(GithubError) as e:
GithubUser(invalid_code,
client_id=self.test_client_id,
client_secret=self.test_client_secret)
# Then
assert error_description.replace("+", " ") in str(e)
def test_get_access_token_success(self):
# Given
test_code = "abc123"
expected_access_token = "access-token"
self.mock_requests.post.return_value = mock.MagicMock(
text=
f"access_token={expected_access_token}&scope=user&token_type=bearer",
status_code=200)
# When
github_user = GithubUser(test_code,
client_id=self.test_client_id,
client_secret=self.test_client_secret)
# Then
assert github_user.access_token == expected_access_token
assert self.mock_requests.post.call_count == 1
request_calls = self.mock_requests.post.call_args
assert request_calls[1]["data"]["code"] == test_code
def get_user_info(self):
github_user = GithubUser(code=self.validated_data["access_token"])
return github_user.get_user_info()