fix: parse int status code to str (#264)

sparckles · Aug 17, 2022 · 3abe5cb · 3abe5cb
1 parent a75c9b6
commit 3abe5cb
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/integration_tests/base_routes.py b/integration_tests/base_routes.py
@@ -69,6 +69,11 @@ def return_404_post():
     return {"status_code": "404", "body": "hello", "type": "text"}
 
 
+@app.get('/int_status_code')
+def return_int_status_code():
+    return {"status_code": 202, "body": "hello", "type": "text"}
+
+
 @app.before_request("/")
 async def hello_before_request(request):
     global callCount

diff --git a/integration_tests/test_status_code.py b/integration_tests/test_status_code.py
@@ -17,3 +17,8 @@ def test_307_get_request(session):
     r = requests.get(f"{BASE_URL}/redirect")
     assert r.text == "This is the redirected route"
 
+
+def test_int_status_code(session):
+    r = requests.get(f"{BASE_URL}/int_status_code")
+    assert r.status_code == 202
+
diff --git a/robyn/router.py b/robyn/router.py
@@ -22,6 +22,9 @@ def _format_response(self, res):
                 res["status_code"] = "200"
                 response = res
             else:
+                if type(res["status_code"]) == int:
+                    res["status_code"] = str(res["status_code"])
+
                 response = {
                     "status_code": "200",
                     "body": res["body"],