You can now return response headers

Author: sansyrox

.gitignore

@@ -11,3 +11,6 @@ dist/
 # DS_Store
 *.DS_Store
 .vscode/settings.json
+
+# python venv
+venv

integration_tests/base_routes.py

@@ -166,26 +166,14 @@ def shutdown_handler():
 
 @app.get("/redirect")
 async def redirect(request):
-    return {"status_code": "307", "body": "", "type": "text"}
+    return {"status_code": "307", "body": "", "type": "text", "headers": jsonify({"Location": "redirect_route"})}
 
 
 @app.get("/redirect_route")
 async def redirect_route(request):
     return "This is the redirected route"
 
 
-@app.before_request("/redirect")
-async def redirect_before_request(request):
-    request["headers"]["Location"] = "redirect_route"
-    return ""
-
-
-@app.after_request("/redirect")
-async def redirect_after_request(request):
-    request["headers"]["Location"] = "redirect_route"
-    return ""
-
-
 if __name__ == "__main__":
     ROBYN_URL = os.getenv("ROBYN_URL", "0.0.0.0")
     app.add_header("server", "robyn")

integration_tests/test_status_code.py

@@ -12,6 +12,7 @@ def test_404_post_request_status_code(session):
     r = requests.post(f"{BASE_URL}/404")
     assert r.status_code == 404
 
+
 def test_307_get_request(session):
     r = requests.get(f"{BASE_URL}/redirect")
     assert r.text == "This is the redirected route"

src/processor.rs

@@ -4,6 +4,7 @@ use std::sync::Arc;
 
 use actix_web::{http::Method, web, HttpRequest, HttpResponse, HttpResponseBuilder};
 use anyhow::{bail, Result};
+use serde_json::Value;
 // pyO3 module
 use crate::types::{Headers, PyFunction};
 use futures_util::stream::StreamExt;
@@ -66,12 +67,16 @@ pub async fn handle_request(
     let body = contents.get("body").unwrap().to_owned();
     let status_code =
         actix_http::StatusCode::from_str(contents.get("status_code").unwrap()).unwrap();
-    let headers = match contents.get("headers") {
-        Some(headers) => headers,
-        None => "",
+
+    let headers: HashMap<String, String> = match contents.get("headers") {
+        Some(headers) => {
+            let h: HashMap<String, String> = serde_json::from_str(headers).unwrap();
+            h
+        }
+        None => HashMap::new(),
     };
 
-    let headers: HashMap<String, String> = serde_json::from_str(headers).unwrap();
+    println!("These are the headers from serde {:?}", headers);
 
     let mut response = HttpResponse::build(status_code);
     apply_headers(&mut response, headers.clone());