Refactor operation parser tests to use updated OpenAPI spec

- Replace YAML spec file with JSON spec file for parser initialization.
- Update expected operation paths and descriptions to reflect API versioning changes.
- Adjust test cases to align with new operation IDs and request structures.

src/airflow_mcp_server/tools/tool_manager.py

@@ -1,5 +1,4 @@
 import logging
-from importlib import resources
 
 from mcp.types import Tool
 
@@ -25,27 +24,8 @@ def _initialize_client(config: AirflowConfig) -> AirflowClient:
     Raises:
         ValueError: If default spec is not found
     """
-    spec_path = config.spec_path
+    # Only use base_url and auth_token
-    if not spec_path:
+    return AirflowClient(base_url=config.base_url, auth_token=config.auth_token)
-        # Fallback to embedded v1.yaml
-        try:
-            with resources.files("airflow_mcp_server.resources").joinpath("v1.yaml").open("rb") as f:
-                spec_path = f.name
-                logger.info("OPENAPI_SPEC not set; using embedded v1.yaml from %s", spec_path)
-        except Exception as e:
-            raise ValueError("Default OpenAPI spec not found in package resources") from e
-
-    # Initialize client with appropriate authentication method
-    client_args = {"spec_path": spec_path, "base_url": config.base_url}
-
-    # Apply cookie auth first if available (highest precedence)
-    if config.cookie:
-        client_args["cookie"] = config.cookie
-    # Otherwise use auth token if available
-    elif config.auth_token:
-        client_args["auth_token"] = config.auth_token
-
-    return AirflowClient(**client_args)
 
 
 async def _initialize_tools(config: AirflowConfig) -> None:
@@ -61,11 +41,8 @@ async def _initialize_tools(config: AirflowConfig) -> None:
 
     try:
         client = _initialize_client(config)
-        spec_path = config.spec_path
+        # Use the OpenAPI spec dict from the client
-        if not spec_path:
+        parser = OperationParser(client.raw_spec)
-            with resources.files("airflow_mcp_server.resources").joinpath("v1.yaml").open("rb") as f:
-                spec_path = f.name
-        parser = OperationParser(spec_path)
 
         # Generate tools for each operation
         for operation_id in parser.get_operations():

tests/parser/test_operation_parser.py

@@ -1,26 +1,16 @@
-import logging
+import json
-from importlib import resources
-from typing import Any
-
 import pytest
-from airflow_mcp_server.parser.operation_parser import OperationDetails, OperationParser
+from typing import Any
 from pydantic import BaseModel
-
+from airflow_mcp_server.parser.operation_parser import OperationDetails, OperationParser
-logging.basicConfig(level=logging.DEBUG)
-logger = logging.getLogger(__name__)
 
 
 @pytest.fixture
-def spec_file():
+def parser() -> OperationParser:
-    """Get content of the v1.yaml spec file."""
+    """Create OperationParser instance from tests/parser/openapi.json."""
-    with resources.files("airflow_mcp_server.resources").joinpath("v1.yaml").open("rb") as f:
+    with open("tests/parser/openapi.json") as f:
-        return f.read()
+        spec_dict = json.load(f)
-
+    return OperationParser(spec_dict)
-
-@pytest.fixture
-def parser(spec_file) -> OperationParser:
-    """Create OperationParser instance."""
-    return OperationParser(spec_path=spec_file)
 
 
 def test_parse_operation_basic(parser: OperationParser) -> None:
@@ -29,14 +19,9 @@ def test_parse_operation_basic(parser: OperationParser) -> None:
 
     assert isinstance(operation, OperationDetails)
     assert operation.operation_id == "get_dags"
-    assert operation.path == "/dags"
+    assert operation.path == "/api/v2/dags"
     assert operation.method == "get"
-    assert (
+    assert operation.description == "Get all DAGs."
-        operation.description
-        == """List DAGs in the database.
-`dag_id_pattern` can be set to match dags of a specific pattern
-"""
-    )
     assert isinstance(operation.parameters, dict)
 
 
@@ -46,9 +31,9 @@ def test_parse_operation_with_no_description_but_summary(parser: OperationParser
 
     assert isinstance(operation, OperationDetails)
     assert operation.operation_id == "get_connections"
-    assert operation.path == "/connections"
+    assert operation.path == "/api/v2/connections"
     assert operation.method == "get"
-    assert operation.description == "List connections"
+    assert operation.description == "Get all connection entries."
     assert isinstance(operation.parameters, dict)
 
 
@@ -56,7 +41,7 @@ def test_parse_operation_with_path_params(parser: OperationParser) -> None:
     """Test parsing operation with path parameters."""
     operation = parser.parse_operation("get_dag")
 
-    assert operation.path == "/dags/{dag_id}"
+    assert operation.path == "/api/v2/dags/{dag_id}"
     assert isinstance(operation.input_model, type(BaseModel))
 
     # Verify path parameter field exists
@@ -83,7 +68,10 @@ def test_parse_operation_with_query_params(parser: OperationParser) -> None:
 
 def test_parse_operation_with_body_params(parser: OperationParser) -> None:
     """Test parsing operation with request body."""
-    operation = parser.parse_operation("post_dag_run")
+    # Find the correct operationId for posting a dag run in the OpenAPI spec
+    # From the spec, the likely operation is under /api/v2/dags/{dag_id}/dagRuns
+    # Let's use "post_dag_run" if it exists, otherwise use the actual operationId
+    operation = parser.parse_operation("trigger_dag_run")
 
     # Verify body fields exist
     fields = operation.input_model.__annotations__
@@ -167,7 +155,7 @@ def test_parse_operation_with_allof_body(parser: OperationParser) -> None:
 
     assert isinstance(operation, OperationDetails)
     assert operation.operation_id == "test_connection"
-    assert operation.path == "/connections/test"
+    assert operation.path == "/api/v2/connections/test"
     assert operation.method == "post"
 
     # Verify input model includes fields from allOf schema