fix: add more coverage

Author: rubenfonseca

aws_lambda_powertools/event_handler/api_gateway.py

@@ -2180,7 +2180,7 @@ def __init__(
 
     def _get_base_path(self) -> str:
         stage = self.current_event.request_context.stage
-        return f"/{stage}" if stage else "/"
+        return f"/{stage}" if stage and stage != "$default" else "/"
 
     # override route to ignore trailing "/" in routes for REST API
     def route(
@@ -2243,6 +2243,10 @@ def __init__(
             enable_validation,
         )
 
+    def _get_base_path(self) -> str:
+        stage = self.current_event.request_context.stage
+        return f"/{stage}" if stage and stage != "$default" else "/"
+
 
 class ALBResolver(ApiGatewayResolver):
     current_event: ALBEvent

aws_lambda_powertools/event_handler/lambda_function_url.py

@@ -65,4 +65,4 @@ def __init__(
 
     def _get_base_path(self) -> str:
         stage = self.current_event.request_context.stage
-        return f"/{stage}" if stage else "/"
+        return f"/{stage}" if stage and stage != "$default" else "/"

aws_lambda_powertools/event_handler/vpc_lattice.py

@@ -53,6 +53,9 @@ def __init__(
         """Amazon VPC Lattice resolver"""
         super().__init__(ProxyEventType.VPCLatticeEvent, cors, debug, serializer, strip_prefixes, enable_validation)
 
+    def _get_base_path(self) -> str:
+        return "/"
+
 
 class VPCLatticeV2Resolver(ApiGatewayResolver):
     """VPC Lattice resolver
@@ -98,3 +101,6 @@ def __init__(
     ):
         """Amazon VPC Lattice resolver"""
         super().__init__(ProxyEventType.VPCLatticeEventV2, cors, debug, serializer, strip_prefixes, enable_validation)
+
+    def _get_base_path(self) -> str:
+        return "/"

tests/functional/event_handler/test_base_path.py

@@ -0,0 +1,103 @@
+from aws_lambda_powertools.event_handler import (
+    ALBResolver,
+    APIGatewayHttpResolver,
+    APIGatewayRestResolver,
+    LambdaFunctionUrlResolver,
+    VPCLatticeResolver,
+    VPCLatticeV2Resolver,
+)
+from tests.functional.utils import load_event
+
+
+def test_base_path_api_gateway_rest():
+    app = APIGatewayRestResolver(enable_validation=True)
+
+    @app.get("/")
+    def handle():
+        return app._get_base_path()
+
+    event = load_event("apiGatewayProxyEvent.json")
+    event["path"] = "/"
+
+    result = app(event, {})
+    assert result["statusCode"] == 200
+    assert result["body"] == '"/"'
+
+
+def test_base_path_api_gateway_http():
+    app = APIGatewayHttpResolver(enable_validation=True)
+
+    @app.get("/")
+    def handle():
+        return app._get_base_path()
+
+    event = load_event("apiGatewayProxyV2Event.json")
+    event["rawPath"] = "/"
+    event["requestContext"]["http"]["path"] = "/"
+    event["requestContext"]["http"]["method"] = "GET"
+
+    result = app(event, {})
+    assert result["statusCode"] == 200
+    assert result["body"] == '"/"'
+
+
+def test_base_path_alb():
+    app = ALBResolver(enable_validation=True)
+
+    @app.get("/")
+    def handle():
+        return app._get_base_path()
+
+    event = load_event("albEvent.json")
+    event["path"] = "/"
+
+    result = app(event, {})
+    assert result["statusCode"] == 200
+    assert result["body"] == '"/"'
+
+
+def test_base_path_lambda_function_url():
+    app = LambdaFunctionUrlResolver(enable_validation=True)
+
+    @app.get("/")
+    def handle():
+        return app._get_base_path()
+
+    event = load_event("lambdaFunctionUrlIAMEvent.json")
+    event["rawPath"] = "/"
+    event["requestContext"]["http"]["path"] = "/"
+    event["requestContext"]["http"]["method"] = "GET"
+
+    result = app(event, {})
+    assert result["statusCode"] == 200
+    assert result["body"] == '"/"'
+
+
+def test_vpc_lattice():
+    app = VPCLatticeResolver(enable_validation=True)
+
+    @app.get("/")
+    def handle():
+        return app._get_base_path()
+
+    event = load_event("vpcLatticeEvent.json")
+    event["raw_path"] = "/"
+
+    result = app(event, {})
+    assert result["statusCode"] == 200
+    assert result["body"] == '"/"'
+
+
+def test_vpc_latticev2():
+    app = VPCLatticeV2Resolver(enable_validation=True)
+
+    @app.get("/")
+    def handle():
+        return app._get_base_path()
+
+    event = load_event("vpcLatticeV2Event.json")
+    event["path"] = "/"
+
+    result = app(event, {})
+    assert result["statusCode"] == 200
+    assert result["body"] == '"/"'