0b1f513826
This PR adds a simplified Linear toolkit focused on core functionality: ## What's Included **Tools:** - `get_issue` - Get detailed information about Linear issues - `get_teams` - Get team information and details **Models:** - `DateRange` enum with comprehensive date range support (TODAY, YESTERDAY, THIS_WEEK, LAST_WEEK, THIS_MONTH, LAST_MONTH, THIS_YEAR, LAST_YEAR, LAST_7_DAYS, LAST_30_DAYS) - Timezone-aware datetime handling following Google toolkit patterns ## What's Simplified This toolkit has been streamlined by removing: - Cycles management tools - Projects management tools - Users management tools - Workflows management tools - Corresponding tests and evaluations for removed features ## Quality Assurance - All linting and formatting checks pass - Comprehensive test coverage for included functionality - Follows established patterns from Google toolkit --------- Co-authored-by: Eric Gustin <34000337+EricGustin@users.noreply.github.com>
145 lines
4.4 KiB
Python
145 lines
4.4 KiB
Python
import pytest
|
|
from arcade_tdk import ToolContext
|
|
|
|
from arcade_linear.tools.teams import get_teams
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_teams_success(
|
|
mock_context: ToolContext,
|
|
mock_httpx_client,
|
|
mock_httpx_response,
|
|
build_team_dict,
|
|
build_graphql_response,
|
|
build_paginated_response,
|
|
):
|
|
"""Test successful team retrieval"""
|
|
# Create sample teams
|
|
team1 = build_team_dict(name="Team Alpha", key="ALPHA")
|
|
team2 = build_team_dict(name="Team Beta", key="BETA")
|
|
|
|
# Build paginated response
|
|
teams_response = build_paginated_response([team1, team2])
|
|
|
|
# Build full GraphQL response
|
|
graphql_data = build_graphql_response({"teams": teams_response})
|
|
|
|
# Mock the HTTP response
|
|
http_response = mock_httpx_response(200, graphql_data)
|
|
mock_httpx_client.post.return_value = http_response
|
|
|
|
result = await get_teams(context=mock_context)
|
|
|
|
assert result["total_count"] == 2
|
|
assert len(result["teams"]) == 2
|
|
assert result["teams"][0]["name"] == "Team Alpha"
|
|
assert result["teams"][0]["key"] == "ALPHA"
|
|
assert result["teams"][1]["name"] == "Team Beta"
|
|
assert result["teams"][1]["key"] == "BETA"
|
|
|
|
# Verify the request was made correctly
|
|
mock_httpx_client.post.assert_called_once()
|
|
call_args = mock_httpx_client.post.call_args
|
|
|
|
# Check URL
|
|
assert call_args[0][0] == "https://api.linear.app/graphql"
|
|
|
|
# Check that query is in the request body
|
|
request_body = call_args[1]["json"]
|
|
assert "query" in request_body
|
|
assert "teams" in request_body["query"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_teams_with_team_filter(
|
|
mock_context: ToolContext,
|
|
mock_httpx_client,
|
|
mock_httpx_response,
|
|
build_team_dict,
|
|
build_graphql_response,
|
|
build_paginated_response,
|
|
):
|
|
"""Test team retrieval with name filter"""
|
|
# Create sample team
|
|
team = build_team_dict(name="Engineering Team", key="ENG")
|
|
|
|
# Build responses
|
|
teams_response = build_paginated_response([team])
|
|
graphql_data = build_graphql_response({"teams": teams_response})
|
|
http_response = mock_httpx_response(200, graphql_data)
|
|
mock_httpx_client.post.return_value = http_response
|
|
|
|
result = await get_teams(context=mock_context, team_name="Engineering")
|
|
|
|
assert result["total_count"] == 1
|
|
assert len(result["teams"]) == 1
|
|
assert result["teams"][0]["name"] == "Engineering Team"
|
|
|
|
# Verify filter was applied in the request
|
|
call_args = mock_httpx_client.post.call_args
|
|
request_body = call_args[1]["json"]
|
|
assert "variables" in request_body
|
|
assert "filter" in request_body["variables"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_teams_empty_result(
|
|
mock_context: ToolContext,
|
|
mock_httpx_client,
|
|
mock_httpx_response,
|
|
build_graphql_response,
|
|
build_paginated_response,
|
|
):
|
|
"""Test handling of empty teams list"""
|
|
# Build empty response
|
|
teams_response = build_paginated_response([])
|
|
graphql_data = build_graphql_response({"teams": teams_response})
|
|
http_response = mock_httpx_response(200, graphql_data)
|
|
mock_httpx_client.post.return_value = http_response
|
|
|
|
result = await get_teams(context=mock_context)
|
|
|
|
assert result["total_count"] == 0
|
|
assert len(result["teams"]) == 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_teams_graphql_error(
|
|
mock_context: ToolContext,
|
|
mock_httpx_client,
|
|
mock_httpx_response,
|
|
):
|
|
"""Test handling of GraphQL errors"""
|
|
# Build error response
|
|
error_data = {
|
|
"data": None,
|
|
"errors": [
|
|
{"message": "Authentication required", "extensions": {"code": "UNAUTHENTICATED"}}
|
|
],
|
|
}
|
|
http_response = mock_httpx_response(200, error_data)
|
|
mock_httpx_client.post.return_value = http_response
|
|
|
|
# The tool should raise an exception for GraphQL errors
|
|
with pytest.raises(Exception) as exc_info:
|
|
await get_teams(context=mock_context)
|
|
|
|
assert "Authentication required" in str(exc_info.value)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_teams_http_error(
|
|
mock_context: ToolContext,
|
|
mock_httpx_client,
|
|
mock_httpx_response,
|
|
):
|
|
"""Test handling of HTTP errors"""
|
|
# Mock HTTP 401 error
|
|
error_response = mock_httpx_response(401, {"error": "Unauthorized"})
|
|
mock_httpx_client.post.return_value = error_response
|
|
|
|
with pytest.raises(Exception) as exc_info:
|
|
await get_teams(context=mock_context)
|
|
|
|
# Should contain HTTP status information
|
|
assert "401" in str(exc_info.value) or "Unauthorized" in str(exc_info.value)
|