Jeremiah Lowin commited on
Commit
194b8dd
·
1 Parent(s): 96d7545

Update tests

Browse files
Files changed (1) hide show
  1. tests/server/openapi/test_openapi.py +16 -24
tests/server/openapi/test_openapi.py CHANGED
@@ -2709,25 +2709,22 @@ class TestGlobalTagsParameter:
2709
  )
2710
 
2711
  # Check tool has both original and global tags
2712
- tools = server._tool_manager.list_tools()
2713
- create_item_tool = next((t for t in tools if "create_item" in t.name), None)
2714
- assert create_item_tool is not None
2715
  assert "items" in create_item_tool.tags # Original OpenAPI tag
2716
  assert "global" in create_item_tool.tags # Global tag
2717
  assert "api-v1" in create_item_tool.tags # Global tag
2718
 
2719
  # Check resource has both original and global tags
2720
- resources = list(server._resource_manager.get_resources().values())
2721
- get_items_resource = next((r for r in resources if "get_items" in r.name), None)
2722
- assert get_items_resource is not None
2723
  assert "items" in get_items_resource.tags # Original OpenAPI tag
2724
  assert "global" in get_items_resource.tags # Global tag
2725
  assert "api-v1" in get_items_resource.tags # Global tag
2726
 
2727
  # Check resource template has both original and global tags
2728
- templates = list(server._resource_manager.get_templates().values())
2729
- get_item_template = next((t for t in templates if "get_item" in t.name), None)
2730
- assert get_item_template is not None
2731
  assert "items" in get_item_template.tags # Original OpenAPI tag
2732
  assert "global" in get_item_template.tags # Global tag
2733
  assert "api-v1" in get_item_template.tags # Global tag
@@ -2754,25 +2751,22 @@ class TestGlobalTagsParameter:
2754
  )
2755
 
2756
  # Check tool has both original and global tags
2757
- tools = server._tool_manager.list_tools()
2758
- create_item_tool = next((t for t in tools if "create_item" in t.name), None)
2759
- assert create_item_tool is not None
2760
  assert "items" in create_item_tool.tags # Original OpenAPI tag
2761
  assert "openapi-global" in create_item_tool.tags # Global tag
2762
  assert "service" in create_item_tool.tags # Global tag
2763
 
2764
  # Check resource has both original and global tags
2765
- resources = list(server._resource_manager.get_resources().values())
2766
- get_items_resource = next((r for r in resources if "get_items" in r.name), None)
2767
- assert get_items_resource is not None
2768
  assert "items" in get_items_resource.tags # Original OpenAPI tag
2769
  assert "openapi-global" in get_items_resource.tags # Global tag
2770
  assert "service" in get_items_resource.tags # Global tag
2771
 
2772
  # Check resource template has both original and global tags
2773
- templates = list(server._resource_manager.get_templates().values())
2774
- get_item_template = next((t for t in templates if "get_item" in t.name), None)
2775
- assert get_item_template is not None
2776
  assert "items" in get_item_template.tags # Original OpenAPI tag
2777
  assert "openapi-global" in get_item_template.tags # Global tag
2778
  assert "service" in get_item_template.tags # Global tag
@@ -2800,17 +2794,15 @@ class TestGlobalTagsParameter:
2800
  )
2801
 
2802
  # Check that all three types of tags are present on the tool
2803
- tools = server._tool_manager.list_tools()
2804
- create_item_tool = next((t for t in tools if "create_item" in t.name), None)
2805
- assert create_item_tool is not None
2806
  assert "items" in create_item_tool.tags # Original OpenAPI tag
2807
  assert "global" in create_item_tool.tags # Global tag
2808
  assert "route-specific" in create_item_tool.tags # RouteMap mcp_tag
2809
 
2810
  # Check that resource only has OpenAPI and global tags (no route-specific since different RouteMap)
2811
- resources = list(server._resource_manager.get_resources().values())
2812
- get_items_resource = next((r for r in resources if "get_items" in r.name), None)
2813
- assert get_items_resource is not None
2814
  assert "items" in get_items_resource.tags # Original OpenAPI tag
2815
  assert "global" in get_items_resource.tags # Global tag
2816
  assert "route-specific" not in get_items_resource.tags # Not from this RouteMap
 
2709
  )
2710
 
2711
  # Check tool has both original and global tags
2712
+ tools = await server.get_tools()
2713
+ create_item_tool = tools["create_item_items_post"]
 
2714
  assert "items" in create_item_tool.tags # Original OpenAPI tag
2715
  assert "global" in create_item_tool.tags # Global tag
2716
  assert "api-v1" in create_item_tool.tags # Global tag
2717
 
2718
  # Check resource has both original and global tags
2719
+ resources = await server.get_resources()
2720
+ get_items_resource = resources["resource://get_items_items_get"]
 
2721
  assert "items" in get_items_resource.tags # Original OpenAPI tag
2722
  assert "global" in get_items_resource.tags # Global tag
2723
  assert "api-v1" in get_items_resource.tags # Global tag
2724
 
2725
  # Check resource template has both original and global tags
2726
+ templates = await server.get_resource_templates()
2727
+ get_item_template = templates["resource://get_item_items/{item_id}"]
 
2728
  assert "items" in get_item_template.tags # Original OpenAPI tag
2729
  assert "global" in get_item_template.tags # Global tag
2730
  assert "api-v1" in get_item_template.tags # Global tag
 
2751
  )
2752
 
2753
  # Check tool has both original and global tags
2754
+ tools = await server.get_tools()
2755
+ create_item_tool = tools["create_item_items_post"]
 
2756
  assert "items" in create_item_tool.tags # Original OpenAPI tag
2757
  assert "openapi-global" in create_item_tool.tags # Global tag
2758
  assert "service" in create_item_tool.tags # Global tag
2759
 
2760
  # Check resource has both original and global tags
2761
+ resources = await server.get_resources()
2762
+ get_items_resource = resources["resource://get_items_items_get"]
 
2763
  assert "items" in get_items_resource.tags # Original OpenAPI tag
2764
  assert "openapi-global" in get_items_resource.tags # Global tag
2765
  assert "service" in get_items_resource.tags # Global tag
2766
 
2767
  # Check resource template has both original and global tags
2768
+ templates = await server.get_resource_templates()
2769
+ get_item_template = templates["resource://get_item_items/{item_id}"]
 
2770
  assert "items" in get_item_template.tags # Original OpenAPI tag
2771
  assert "openapi-global" in get_item_template.tags # Global tag
2772
  assert "service" in get_item_template.tags # Global tag
 
2794
  )
2795
 
2796
  # Check that all three types of tags are present on the tool
2797
+ tools = await server.get_tools()
2798
+ create_item_tool = tools["create_item_items_post"]
 
2799
  assert "items" in create_item_tool.tags # Original OpenAPI tag
2800
  assert "global" in create_item_tool.tags # Global tag
2801
  assert "route-specific" in create_item_tool.tags # RouteMap mcp_tag
2802
 
2803
  # Check that resource only has OpenAPI and global tags (no route-specific since different RouteMap)
2804
+ resources = await server.get_resources()
2805
+ get_items_resource = resources["resource://get_items_items_get"]
 
2806
  assert "items" in get_items_resource.tags # Original OpenAPI tag
2807
  assert "global" in get_items_resource.tags # Global tag
2808
  assert "route-specific" not in get_items_resource.tags # Not from this RouteMap