class TestListPullRequestsGithubUserHandlerPAT(TestPullRequest):
# Test no PAT
@patch("jupyterlab_pullrequests.base.PullRequestsAPIHandler.get_manager",
Mock(return_value=PullRequestsGithubManager(None)))
def test_pat_none(self):
response = self.fetch('/pullrequests/prs/user?filter=created')
self.assertEqual(response.code, 400)
self.assertIn("No Github access token specified", response.reason)
# Test empty PAT
@patch("jupyterlab_pullrequests.base.PullRequestsAPIHandler.get_manager",
Mock(return_value=PullRequestsGithubManager("")))
def test_pat_empty(self):
response = self.fetch('/pullrequests/prs/user?filter=created')
self.assertEqual(response.code, 400)
self.assertIn("No Github access token specified", response.reason)
# Test invalid PAT
@patch("jupyterlab_pullrequests.base.PullRequestsAPIHandler.get_manager",
Mock(return_value=PullRequestsGithubManager("invalid")))
def test_pat_invalid(self):
response = self.fetch('/pullrequests/prs/user?filter=created')
self.assertEqual(response.code, 401)
self.assertIn("Invalid response in", response.reason)
def get_manager(self):
c = GitHubConfig(config=self.config)
p = c.platform
if p.lower() == "github":
return PullRequestsGithubManager(c.github_access_token)
elif p.lower() == "codecommit":
raise NotImplementedError()
else:
return PullRequestsGithubManager(c.github_access_token)
async def test_valid(self):
manager = PullRequestsGithubManager("valid-pat")
prev_content = json.dumps(read_sample_response("ipynb_base.json"))
curr_content = json.dumps(read_sample_response("ipynb_remote.json"))
result = await manager.get_file_nbdiff(prev_content, curr_content)
expected_result = read_sample_response("ipynb_nbdiff.json")
assert result == expected_result
async def test_bad_unknown(self, mock_fetch):
manager = PullRequestsGithubManager("valid-pat")
mock_fetch.side_effect = Exception()
with (pytest.raises(HTTPError)) as e:
await manager.call_github("invalid-link")
assert e.value.status_code == HTTPStatus.INTERNAL_SERVER_ERROR
assert "Unknown error in" in e.value.reason
async def test_valid_new(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
mock_call_github.return_value = read_sample_response(
'github_comments_post.json')
PRCommentNew = namedtuple(
"PRCommentNew", ["text", "commit_id", "filename", "position"])
body = PRCommentNew("test text", 123, "test.ipynb", 3)
result = await manager.post_file_comment(
"https://api.github.com/repos/octocat/repo/pulls/1", "test.ipynb",
body)
expected_result = {
'id': 299659626,
'line_number': 9,
'text': 'test',
'updated_at': '2019-07-02T19:58:38Z',
'user_name': 'timnlupo',
'user_pic': 'https://avatars1.githubusercontent.com/u/9003282?v=4',
'in_reply_to_id': 296364299
}
mock_call_github.assert_called_with(
'https://api.github.com/repos/octocat/repo/pulls/1/comments',
body={
'body': 'test text',
'commit_id': 123,
'path': 'test.ipynb',
'position': 3
},
method='POST')
assert result == expected_result
async def test_bad_gitresponse(self, mock_fetch):
manager = PullRequestsGithubManager("valid-pat")
mock_fetch.side_effect = HTTPClientError(code=404)
with (pytest.raises(HTTPError)) as e:
await manager.call_github("invalid-link")
assert e.value.status_code == 404
assert "Invalid response in" in e.value.reason
async def test_get_file_content_calls(self, mock_get_pr_links,
mock_validate_pr_link,
mock_get_single_content):
manager = PullRequestsGithubManager("valid-pat")
mock_get_pr_links.return_value = {
'base_url': 'http://base_url.com',
'head_url': 'http://head_url.com',
'commit_id': '123sha'
}
mock_validate_pr_link.side_effect = [
'http://base_url_download.com', 'http://head_url_download.com'
]
mock_get_single_content.side_effect = ['base content', 'head content']
result = await manager.get_file_content("valid-prid", "valid-filename")
mock_get_pr_links.assert_called_once_with("valid-prid",
"valid-filename")
assert mock_validate_pr_link.call_count == 2
mock_validate_pr_link.assert_has_calls(
[call("http://base_url.com"),
call("http://head_url.com")],
any_order=True)
assert mock_get_single_content.call_count == 2
mock_get_single_content.assert_has_calls([
call("http://base_url_download.com"),
call("http://head_url_download.com")
],
any_order=True)
assert result == {
'base_content': 'base content',
'head_content': 'head content',
'commit_id': '123sha'
}
async def test_call(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
mock_call_github.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_github.assert_called_with(
'https://api.github.com/repos/octocat/repo/pulls/1/files')
assert result == [{
'name': 'README.md',
'status': 'added',
'additions': 1,
'deletions': 0
}]
async def test_get_file_content(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
mock_call_github.side_effect = [
read_sample_response('github_pr_links.json'),
read_sample_response('github_validate_pr_link.json'),
read_sample_response('github_validate_pr_link.json'),
'test code content', 'test new code content'
]
result = await manager.get_file_content("valid-prid", "valid-filename")
assert mock_call_github.call_count == 5
assert result == {
'base_content': 'test code content',
'head_content': 'test new code content',
'commit_id': '02fb374e022fbe7aaa4cd69c0dc3928e6422dfaa'
}
async def test_result(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
mock_call_github.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',
'internal_id':
457075994,
'url':
'https://github.com/timnlupo/juypterlabpr-test/pull/1'
}]
async def test_call(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
mock_call_github.return_value = read_sample_response(
'github_comments_get.json')
result = await manager.get_file_comments(
"https://api.github.com/repos/octocat/repo/pulls/1", "test.ipynb")
mock_call_github.assert_called_with(
'https://api.github.com/repos/octocat/repo/pulls/1/comments')
expected_result = [{
"id":
296364299,
"line_number":
9,
"text":
"too boring",
"updated_at":
"2019-06-21T19:21:20Z",
"user_name":
"timnlupo",
"user_pic":
"https://avatars1.githubusercontent.com/u/9003282?v=4"
}]
assert result == expected_result
async def test_assigned(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
await manager.list_prs("notoctocat", "assigned")
mock_call_github.assert_called_with(
'https://api.github.com/search/issues?q=+state:open+type:pr+assignee:notoctocat'
)
async def test_pat_valid(self, mock_call_github):
manager = PullRequestsGithubManager("valid-pat")
mock_call_github.return_value = read_sample_response(
'github_current_user.json')
result = await manager.get_current_user()
assert result == {'username': '******'}
async def test_valid(self, mock_fetch):
manager = PullRequestsGithubManager("valid-pat")
result = await manager.call_github("valid-link")
assert result["test1"] == "test2"
async def test_bad_parse(self, mock_fetch):
manager = PullRequestsGithubManager("valid-pat")
with (pytest.raises(HTTPError)) as e:
await manager.call_github("invalid-link")
assert e.value.status_code == HTTPStatus.BAD_REQUEST
assert "Invalid response in" in e.value.reason
async def test_pat_empty(self):
manager = PullRequestsGithubManager("")
with (pytest.raises(HTTPError)) as e:
await manager.get_current_user()
assert e.value.status_code == HTTPStatus.BAD_REQUEST
assert "No Github access token specified" in e.value.reason
response = self.fetch('/pullrequests/prs/user?filter=created')
self.assertEqual(response.code, 400)
self.assertIn("No Github access token specified", response.reason)
# Test invalid PAT
@patch("jupyterlab_pullrequests.base.PullRequestsAPIHandler.get_manager",
Mock(return_value=PullRequestsGithubManager("invalid")))
def test_pat_invalid(self):
response = self.fetch('/pullrequests/prs/user?filter=created')
self.assertEqual(response.code, 401)
self.assertIn("Invalid response in", response.reason)
# Test list pull requests params
@patch("jupyterlab_pullrequests.base.PullRequestsAPIHandler.get_manager",
Mock(return_value=PullRequestsGithubManager(valid_access_token)))
class TestListPullRequestsGithubUserHandlerParam(TestPullRequest):
# Test missing parameter
def test_param_missing(self):
response = self.fetch('/pullrequests/prs/user')
self.assertEqual(response.code, 400)
self.assertIn("Missing argument 'filter'", response.reason)
# Test no parameter
def test_param_none(self):
response = self.fetch('/pullrequests/prs/user?filter=')
self.assertEqual(response.code, 400)
self.assertIn("Invalid argument 'filter'", response.reason)
# Test invalid parameter