[test] add test cases, add function utility base64_encode()

src/utilities/utilities.py

@@ -1,17 +1,20 @@
 """Various utilities (logger, time benchmark, args dump, numerical and stats info)"""
 
 
-def is_base64(sb):
+def prepare_base64_input(sb):
+    if isinstance(sb, str):
+        # If there's any unicode here, an exception will be thrown and the function will return false
+        return bytes(sb, 'ascii')
+    elif isinstance(sb, bytes):
+        return sb
+    raise ValueError("Argument must be string or bytes")
+
+
+def is_base64(sb: str or bytes):
     import base64
 
     try:
-        if isinstance(sb, str):
-            # If there's any unicode here, an exception will be thrown and the function will return false
-            sb_bytes = bytes(sb, 'ascii')
-        elif isinstance(sb, bytes):
-            sb_bytes = sb
-        else:
-            raise ValueError("Argument must be string or bytes")
+        sb_bytes = prepare_base64_input(sb)
         return base64.b64encode(base64.b64decode(sb_bytes, validate=True)) == sb_bytes
     except ValueError:
         return False
@@ -24,3 +27,10 @@ def base64_decode(s):
         return base64.b64decode(s, validate=True).decode("utf-8")
 
     return s
+
+
+def base64_encode(sb: str or bytes):
+    import base64
+
+    sb_bytes = prepare_base64_input(sb)
+    return base64.b64encode(sb_bytes)

tests/io/test_lambda_helpers.py

@@ -1,6 +1,8 @@
 import json
 
 from src.io.lambda_helpers import get_parsed_bbox_points, get_parsed_request_body
+from src.utilities.type_hints import RawRequestInput
+from src.utilities.utilities import base64_encode
 from tests import TEST_EVENTS_FOLDER
 
 
@@ -12,3 +14,34 @@ def test_get_parsed_bbox_points():
             raw_body = get_parsed_request_body(**input_output["input"])
             output = get_parsed_bbox_points(raw_body)
             assert output == input_output["output"]
+
+
+def test_get_parsed_request_body():
+    input_event = {
+        "event": {
+            "bbox": {
+                "ne": {"lat": 38.03932961278458, "lng": 15.36808069832851},
+                "sw": {"lat": 37.455509218936974, "lng": 14.632807441554068}
+            },
+            "prompt": [{"type": "point", "data": {"lat": 37.0, "lng": 15.0}, "label": 0}],
+            "zoom": 10, "source_type": "Satellite", "debug": True
+        }
+    }
+    expected_output_dict = {
+        "bbox": {
+            "ne": {"lat": 38.03932961278458, "lng": 15.36808069832851},
+            "sw": {"lat": 37.455509218936974, "lng": 14.632807441554068}
+        },
+        "prompt": [{"type": "point", "data": {"lat": 37.0, "lng": 15.0}, "label": 0}],
+        "zoom": 10, "source_type": "Satellite", "debug": True
+    }
+    output = get_parsed_request_body(input_event["event"])
+    assert output == RawRequestInput.model_validate(input_event["event"])
+
+    input_event_str = json.dumps(input_event["event"])
+    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")}
+    output = get_parsed_request_body(event)
+    assert output == RawRequestInput.model_validate(expected_output_dict)