async def test_GitHubManager_user_pat_valid(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.return_value = read_sample_response("github_current_user.json")
result = await manager.get_current_user()
assert result == {"username": "******"}
async def test_GitHubManager_post_comment_valid_reply(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.return_value = read_sample_response("github_comments_post.json")
body = CommentReply("test text", "test.ipynb", 123)
result = await manager.post_comment(
"https://api.github.com/repos/octocat/repo/pulls/1", body
)
mock_call_provider.assert_called_once()
request = mock_call_provider.call_args[0][0]
assert (
request.url == "https://api.github.com/repos/octocat/repo/pulls/1/comments",
)
assert json.loads(request.body.decode("utf-8")) == {
"body": "test text",
"in_reply_to": 123,
}
assert request.method == "POST"
expected_result = {
"id": 299659626,
"text": "test",
"updatedAt": "2019-07-02T19:58:38Z",
"userName": "******",
"userPicture": "https://avatars1.githubusercontent.com/u/9003282?v=4",
"inReplyToId": 296364299,
}
assert result == expected_result
async def test_GitHubManager_get_threads(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.return_value = read_sample_response("github_comments_get.json")
result = await manager.get_threads(
"https://api.github.com/repos/octocat/repo/pulls/1", "test.ipynb"
)
mock_call_provider.assert_called_once()
assert (
mock_call_provider.call_args[0][0].url
== "https://api.github.com/repos/octocat/repo/pulls/1/comments?per_page=100"
)
expected_result = [
{
"comments": [
{
"id": 296364299,
"inReplyToId": None,
"text": "too boring",
"updatedAt": "2019-06-21T19:21:20Z",
"userName": "******",
"userPicture": "https://avatars1.githubusercontent.com/u/9003282?v=4",
}
],
"filename": "test.ipynb",
"id": 296364299,
"line": 9,
"originalLine": None,
"pullRequestId": "https://api.github.com/repos/octocat/repo/pulls/1",
}
]
assert result == expected_result
async def test_GitHubManager_list_prs_assigned(mock_call_provider):
manager = GitHubManager(access_token="valid")
await manager.list_prs("notoctocat", "assigned")
mock_call_provider.assert_called_once_with(
"https://api.github.com/search/issues?q=+state:open+type:pr+assignee:notoctocat",
)
async def test_GitHubManager_call_provider_bad_unknown(mock_fetch):
manager = GitHubManager(access_token="valid")
mock_fetch.side_effect = Exception()
with (pytest.raises(HTTPError)) as e:
await manager._call_provider("invalid-link")
assert e.value.status_code == HTTPStatus.INTERNAL_SERVER_ERROR
assert "Unknown error in" in e.value.reason
async def test_GitHubManager_call_provider_bad_parse(mock_fetch):
mock_fetch.return_value = MagicMock(headers={})
manager = GitHubManager(access_token="valid")
with (pytest.raises(HTTPError)) as e:
await manager._call_provider("invalid-link")
assert e.value.status_code == HTTPStatus.BAD_REQUEST
assert "Invalid response in" in e.value.reason
async def test_GitHubManager_call_provider_bad_gitresponse(mock_fetch):
manager = GitHubManager(access_token="valid")
mock_fetch.side_effect = HTTPClientError(code=404)
with pytest.raises(HTTPError) as e:
await manager._call_provider("invalid-link")
assert e.value.status_code == 404
assert "Invalid response in" in e.value.reason
async def test_GitHubManager_call_provider_pagination_dict(mock_fetch):
"""Check that paginated endpoint returning dictionary got aggregated in a list"""
expected_data = [{"name": "first"}, {"name": "second"}]
manager = GitHubManager(access_token="valid")
mock_fetch.side_effect = [
MagicMock(body=b'{"name":"first"}', headers={"Link": '<next-url>; rel="next"'}),
MagicMock(body=b'{"name":"second"}', headers={}),
]
result = await manager._call_provider("valid-link")
assert result == expected_data
async def test_GitHubManager_get_file_nbdiff():
manager = GitHubManager(access_token="valid")
prev_content = (HERE / "sample_responses" / "github" /
"ipynb_base.json").read_text()
curr_content = (HERE / "sample_responses" / "github" /
"ipynb_remote.json").read_text()
result = await manager.get_file_nbdiff(prev_content, curr_content)
expected_result = json.loads((HERE / "sample_responses" / "github" /
"ipynb_nbdiff.json").read_text())
assert result == expected_result
async def test_GitHubManager_list_files_call(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.return_value = read_sample_response("github_list_files.json")
result = await manager.list_files(
"https://api.github.com/repos/octocat/repo/pulls/1"
)
mock_call_provider.assert_called_once()
assert (
mock_call_provider.call_args[0][0].url
== "https://api.github.com/repos/octocat/repo/pulls/1/files?per_page=100"
)
assert result == [{"name": "README.md", "status": "added"}]
async def test_GitHubManager_list_prs_result(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.return_value = read_sample_response("github_list_prs.json")
result = await manager.list_prs("octocat", "assigned")
assert result == [
{
"id": "https://api.github.com/repos/timnlupo/juypterlabpr-test/pulls/1",
"title": "Interesting PR for feature",
"body": "This is a feature that tests a bunch of different types",
"internalId": 457075994,
"link": "https://github.com/timnlupo/juypterlabpr-test/pull/1",
}
]
async def test_GitHubManager_call_provider_pagination(mock_fetch, link,
expected):
expected_data = [{"name": "first"}, {"name": "second"}, {"name": "third"}]
manager = GitHubManager(access_token="valid")
mock_fetch.side_effect = [
MagicMock(body=b'[{"name":"first"},{"name":"second"}]', headers=link),
MagicMock(body=b'[{"name":"third"}]', headers={}),
]
result = await manager._call_provider("valid-link")
assert mock_fetch.await_count == expected
assert mock_fetch.await_args_list[0][0][0].url.endswith("?per_page=100")
if expected == 2:
assert (mock_fetch.await_args_list[1][0][0].url ==
f"{manager.base_api_url}/next-url")
assert result == expected_data[:(expected + 1)]
async def test_GitHubManager_get_file_diff(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.side_effect = [
read_sample_response("github_pr_links.json"),
MagicMock(body=b"test code content"),
MagicMock(body=b"test new code content"),
]
result = await manager.get_file_diff("valid-prid", "valid-filename")
assert mock_call_provider.call_count == 3
assert result == {
"base": {
"label": "timnlupo:master",
"sha": "a221b6d04be7fff0737c24e1e335a3091eca81e7",
"content": "test code content",
},
"head": {
"label": "timnlupo:dev",
"sha": "02fb374e022fbe7aaa4cd69c0dc3928e6422dfaa",
"content": "test new code content",
},
}
async def test_GitHubManager_post_comment_valid_new(mock_call_provider):
manager = GitHubManager(access_token="valid")
mock_call_provider.side_effect = [
read_sample_response("github_pr_links.json"),
read_sample_response("github_comments_post.json"),
]
body = NewComment(
text="test text", filename="test.ipynb", line=3, originalLine=None
)
result = await manager.post_comment(
"https://api.github.com/repos/octocat/repo/pulls/1", body
)
expected_result = {
"id": 299659626,
"text": "test",
"updatedAt": "2019-07-02T19:58:38Z",
"userName": "******",
"userPicture": "https://avatars1.githubusercontent.com/u/9003282?v=4",
# FIXME use a different sample set without in_reply_to_id
"inReplyToId": 296364299,
}
assert mock_call_provider.call_count == 2
# 1 = 2nd call; 0 = args of call; 0 = first argument
request = mock_call_provider.call_args_list[1][0][0]
assert request.url == "https://api.github.com/repos/octocat/repo/pulls/1/comments"
assert json.loads(request.body.decode("utf-8")) == {
"body": "test text",
"commit_id": "02fb374e022fbe7aaa4cd69c0dc3928e6422dfaa",
"path": "test.ipynb",
"line": 3,
"side": "RIGHT",
}
assert request.method == "POST"
assert result == expected_result
async def test_GitHubManager_call_provider_valid(mock_fetch):
mock_fetch.return_value = MagicMock(body=b'{"test1": "test2"}')
manager = GitHubManager(access_token="valid")
result = await manager._call_provider("valid-link")
assert result[0]["test1"] == "test2"
async def test_GitHubManager_user_pat_empty(mock_call_provider):
manager = GitHubManager(access_token="")
with (pytest.raises(HTTPError)) as e:
await manager.get_current_user()
assert e.value.status_code == HTTPStatus.BAD_REQUEST
assert "No access token specified" in e.value.reason