7e352fbe91
### Adds the following tools to the Github Toolkit:
1. CreateIssueComment
2. SetStarred
3. CountStargazers
4. ListOrgRepositories
5. GetRepository
6. ListRepositoryActivities
7. ListReviewCommentsInARepository
8. ListPullRequests
9. GetPullRequest
10. UpdatePullRequest
11. ListPullRequestCommits
12. CreateReplyForReviewComment
13. ListReviewCommentsOnPullRequest
14. CreateReviewComment
Adds evals for all of these tools and unit tests.
---------
Co-authored-by: Sam Partee <sam@arcade-ai.com>
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
from arcade_github.tools.models import RepoType
|
|
from arcade_github.tools.repositories import (
|
|
count_stargazers,
|
|
get_repository,
|
|
list_org_repositories,
|
|
list_repository_activities,
|
|
list_review_comments_in_a_repository,
|
|
)
|
|
from httpx import Response
|
|
|
|
from arcade.core.errors import ToolExecutionError
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_context():
|
|
context = AsyncMock()
|
|
context.authorization.token = "mock_token" # noqa: S105
|
|
return context
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_client():
|
|
with patch("arcade_github.tools.repositories.httpx.AsyncClient") as client:
|
|
yield client.return_value.__aenter__.return_value
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"status_code,error_message,expected_error",
|
|
[
|
|
(422, "Validation Failed", "Error accessing.*: Validation failed"),
|
|
(301, "Moved Permanently", "Error accessing.*: Moved permanently"),
|
|
(404, "Not Found", "Error accessing.*: Resource not found"),
|
|
(503, "Service Unavailable", "Error accessing.*: Service unavailable"),
|
|
(410, "Gone", "Error accessing.*: Gone"),
|
|
],
|
|
)
|
|
async def test_error_responses(
|
|
mock_context, mock_client, status_code, error_message, expected_error
|
|
):
|
|
mock_client.get.return_value = Response(status_code, json={"message": error_message})
|
|
mock_client.post.return_value = Response(status_code, json={"message": error_message})
|
|
|
|
with pytest.raises(ToolExecutionError, match=expected_error):
|
|
if status_code == 422:
|
|
await list_org_repositories(mock_context, "org", repo_type=RepoType.ALL)
|
|
elif status_code == 301:
|
|
await count_stargazers("owner", "repo")
|
|
elif status_code == 404:
|
|
await list_org_repositories(mock_context, "non_existent_org")
|
|
elif status_code == 503:
|
|
await get_repository(mock_context, "owner", "repo")
|
|
elif status_code == 410:
|
|
await list_review_comments_in_a_repository(mock_context, "owner", "repo")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_repository_activities_invalid_cursor(mock_context, mock_client):
|
|
mock_client.get.return_value = Response(422, json={"message": "Validation Failed"})
|
|
|
|
with pytest.raises(ToolExecutionError, match="Error accessing.*: Validation failed"):
|
|
await list_repository_activities(mock_context, "owner", "repo", before="invalid_cursor")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_count_stargazers_success(mock_client):
|
|
mock_client.get.return_value = Response(200, json={"stargazers_count": 42})
|
|
|
|
result = await count_stargazers("owner", "repo")
|
|
assert result == "The repository owner/repo has 42 stargazers."
|