[feat] add test case for get_response

src/utilities/constants.py

@@ -4,6 +4,7 @@ OUTPUT_CRS_STRING = "EPSG:3857"
 ROOT = "/tmp"
 CUSTOM_RESPONSE_MESSAGES = {
     200: "ok",
+    400: "Bad Request",
     422: "Missing required parameter",
     500: "Internal server error"
 }

tests/events/get_response.json

@@ -0,0 +1,29 @@
+{
+  "OK": {
+    "input": {
+      "n_predictions": 1,
+      "geojson": "{\"type\": \"FeatureCollection\", \"features\": [{\"id\": \"0\", \"type\": \"Feature\", \"properties\": {\"raster_val\": 255.0}, \"geometry\": {\"type\": \"Polygon\", \"coordinates\": [[[148.359375, -40.4469470596005], [148.447265625, -40.4469470596005], [148.447265625, -40.51379915504414], [148.359375, -40.4469470596005]]]}}]}",
+      "n_shapes_geojson": 2
+    },
+    "output": {
+      "statusCode": 200,
+      "header": {
+        "Content-Type": "application/json"
+      },
+      "body": "{\"n_predictions\": 1, \"geojson\": \"{\\\"type\\\": \\\"FeatureCollection\\\", \\\"features\\\": [{\\\"id\\\": \\\"0\\\", \\\"type\\\": \\\"Feature\\\", \\\"properties\\\": {\\\"raster_val\\\": 255.0}, \\\"geometry\\\": {\\\"type\\\": \\\"Polygon\\\", \\\"coordinates\\\": [[[148.359375, -40.4469470596005], [148.447265625, -40.4469470596005], [148.447265625, -40.51379915504414], [148.359375, -40.4469470596005]]]}}]}\", \"n_shapes_geojson\": 2, \"duration_run\": 108, \"message\": \"ok\", \"request_id\": \"test_invoke_id\"}",
+      "isBase64Encoded": false
+    }
+  },
+  "400": {
+    "input": {},
+    "output": "{\"statusCode\": 400, \"header\": {\"Content-Type\": \"application/json\"}, \"body\": \"{\\\"duration_run\\\": 108, \\\"message\\\": \\\"Bad Request\\\", \\\"request_id\\\": \\\"test_invoke_id\\\"}\", \"isBase64Encoded\": false}"
+  },
+  "422": {
+    "input": {},
+    "output": "{\"statusCode\": 422, \"header\": {\"Content-Type\": \"application/json\"}, \"body\": \"{\\\"duration_run\\\": 108, \\\"message\\\": \\\"Missing required parameter\\\", \\\"request_id\\\": \\\"test_invoke_id\\\"}\", \"isBase64Encoded\": false}"
+  },
+  "500": {
+    "input": {},
+    "output": "{\"statusCode\": 500, \"header\": {\"Content-Type\": \"application/json\"}, \"body\": \"{\\\"duration_run\\\": 108, \\\"message\\\": \\\"Internal server error\\\", \\\"request_id\\\": \\\"test_invoke_id\\\"}\", \"isBase64Encoded\": false}"
+  }
+}

tests/io/test_lambda_helpers.py

@@ -1,11 +1,37 @@
 import json
-
-from src.io.lambda_helpers import get_parsed_bbox_points, get_parsed_request_body
+import time
+from http import HTTPStatus
+from unittest.mock import patch
+from src.io.lambda_helpers import get_parsed_bbox_points, get_parsed_request_body, get_response
 from src.utilities.type_hints import RawRequestInput
-from src.utilities.utilities import base64_encode
+from src.utilities import utilities
 from tests import TEST_EVENTS_FOLDER
 
 
+@patch.object(time, "time")
+def test_get_response(time_mocked):
+    time_diff = 108
+    end_run = 1000
+    time_mocked.return_value = end_run
+    start_time = end_run-time_diff
+    aws_request_id = "test_invoke_id"
+
+    with open(TEST_EVENTS_FOLDER / "get_response.json") as tst_json:
+        inputs_outputs = json.load(tst_json)
+        body_response = inputs_outputs["OK"]["input"]
+        output = get_response(HTTPStatus.OK.value, start_time, aws_request_id, body_response)
+        assert json.loads(output) == inputs_outputs["OK"]["output"]
+        
+    response_400 = get_response(HTTPStatus.BAD_REQUEST.value, start_time, aws_request_id, {})
+    assert response_400 == inputs_outputs["400"]["output"]
+    
+    response_422 = get_response(HTTPStatus.UNPROCESSABLE_ENTITY.value, start_time, aws_request_id, {})
+    assert response_422 == inputs_outputs["422"]["output"]
+    
+    response_500 = get_response(HTTPStatus.INTERNAL_SERVER_ERROR.value, start_time, aws_request_id, {})
+    assert response_500 == inputs_outputs["500"]["output"]
+
+
 def test_get_parsed_bbox_points():
     with open(TEST_EVENTS_FOLDER / "get_parsed_bbox_points.json") as tst_json:
         inputs_outputs = json.load(tst_json)
@@ -42,6 +68,6 @@ def test_get_parsed_request_body():
     output = get_parsed_request_body(input_event_str)
     assert output == RawRequestInput.model_validate(expected_output_dict)
 
-    event = {"body": base64_encode(input_event_str).decode("utf-8")}
+    event = {"body": utilities.base64_encode(input_event_str).decode("utf-8")}
     output = get_parsed_request_body(event)
     assert output == RawRequestInput.model_validate(expected_output_dict)