fix: correct tests

Author: Michael Brewer

tests/functional/event_handler/test_api_gateway.py

@@ -12,10 +12,10 @@
 
 from aws_lambda_powertools.event_handler import content_types
 from aws_lambda_powertools.event_handler.api_gateway import (
-    ALBEventResolver,
-    APIGatewayProxyEventResolver,
-    APIGatewayProxyEventV2Resolver,
+    ALBResolver,
+    APIGatewayHttpResolver,
     ApiGatewayResolver,
+    APIGatewayRestResolver,
     CORSConfig,
     ProxyEventType,
     Response,
@@ -51,7 +51,7 @@ def read_media(file_name: str) -> bytes:
 
 def test_alb_event():
     # GIVEN an Application Load Balancer proxy type event
-    app = ALBEventResolver()
+    app = ALBResolver()
 
     @app.get("/lambda")
     def foo():
@@ -72,7 +72,7 @@ def foo():
 
 def test_api_gateway_v1():
     # GIVEN a Http API V1 proxy type event
-    app = APIGatewayProxyEventResolver()
+    app = APIGatewayRestResolver()
 
     @app.get("/my/path")
     def get_lambda() -> Response:
@@ -111,7 +111,7 @@ def get_lambda() -> Response:
 
 def test_api_gateway_v2():
     # GIVEN a Http API V2 proxy type event
-    app = APIGatewayProxyEventV2Resolver()
+    app = APIGatewayHttpResolver()
 
     @app.post("/my/path")
     def my_path() -> Response:

tests/functional/test_metrics.py

@@ -496,7 +496,8 @@ def lambda_handler(evt, context):
     # THEN ColdStart metric and function_name and service dimension should be logged
     assert output["ColdStart"] == [1.0]
     assert output["function_name"] == "example_fn"
-    assert output['service'] == service
+    assert output["service"] == service
+
 
 def test_log_metrics_capture_cold_start_metric_no_service(capsys, namespace):
     # GIVEN Metrics is initialized without service
@@ -515,7 +516,7 @@ def lambda_handler(evt, context):
     # THEN ColdStart metric and function_name dimension should be logged
     assert output["ColdStart"] == [1.0]
     assert output["function_name"] == "example_fn"
-    assert output.get('service') is None
+    assert output.get("service") is None
 
 
 def test_emit_cold_start_metric_only_once(capsys, namespace, service, metric):