tests: improve coverage

Author: leandrodamascena

tests/functional/event_handler/_pydantic/merge_handlers/tagged_handler.py

@@ -0,0 +1,25 @@
+"""Handler with tags for testing tag merging."""
+
+from __future__ import annotations
+
+from aws_lambda_powertools.event_handler import APIGatewayRestResolver
+
+app = APIGatewayRestResolver()
+
+
+@app.get("/tagged")
+def tagged_endpoint():
+    """Endpoint in tagged handler."""
+    return {"tagged": True}
+
+
+# Override get_openapi_schema to include tags
+_original_get_openapi_schema = app.get_openapi_schema
+
+
+def get_openapi_schema_with_tags(**kwargs):
+    kwargs.setdefault("tags", ["handler-tag"])
+    return _original_get_openapi_schema(**kwargs)
+
+
+app.get_openapi_schema = get_openapi_schema_with_tags

tests/functional/event_handler/_pydantic/test_openapi_merge.py

@@ -338,3 +338,16 @@ def test_openapi_merge_add_schema():
     # THEN it should be included in the merged schema
     schema = merge.get_openapi_schema()
     assert "/external" in schema["paths"]
+
+
+def test_openapi_merge_tags_from_schema():
+    # GIVEN an OpenAPIMerge without config tags
+    merge = OpenAPIMerge(title="Tags API", version="1.0.0")
+
+    # WHEN discovering a handler that has tags in its schema
+    merge.discover(path=MERGE_HANDLERS_PATH, pattern="**/tagged_handler.py")
+
+    # THEN schema tags should include tags from discovered handler
+    schema = merge.get_openapi_schema()
+    tag_names = [t["name"] for t in schema.get("tags", [])]
+    assert "handler-tag" in tag_names