tests: remove test_routes.py and fix integration tests

sparckles · Dec 20, 2022 · 79d83de · 79d83de
1 parent 8120f24
commit 79d83de
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 37 deletions.
diff --git a/integration_tests/test_dir/build/index.html b/integration_tests/test_dir/build/index.html
diff --git a/integration_tests/test_routes.py b/integration_tests/test_routes.py
diff --git a/robyn/router.py b/robyn/router.py
@@ -28,18 +28,29 @@ def _format_response(self, res):
         # handle file handlers
         response = {}
         if type(res) == dict:
-            if "status_code" not in res:
-                res["status_code"] = 200
-                response = res
-            else:
-                if type(res["status_code"]) == str:
-                    res["status_code"] = int(res["status_code"])
-
-                response = {"status_code": 200, "body": res["body"], **res}
+            status_code = res.get("status_code", 200)
+            headers = res.get("headers", {})
+            body = res.get("body", "")
+
+            if type(status_code) != int:
+                status_code = int(status_code)  # status_code can potentially be string
+
+            response = {
+                "status_code": status_code,
+                "body": body,
+                "headers": headers,
+                **res,
+            }
         else:
-            response = {"status_code": 200, "body": res, "type": "text", "headers": {"Content-Type": "text/plain"}}
+            response = {
+                "status_code": 200,
+                "body": res,
+                "type": "text",
+                "headers": {"Content-Type": "text/plain"},
+            }
 
         return response
+
     def add_route(
         self, route_type: str, endpoint: str, handler: Callable, is_const: bool
     ) -> Union[Callable, CoroutineType]: