action custom body fix tests

Author: sfavot

tests/redirectionio_router_test.rs

@@ -830,6 +830,108 @@ fn test_06_emojis_1() {
     assert_eq!(action.should_log_request(true, response_status_code, None), true);
 }
 
+fn setup_action_custom_body() -> Router<Rule> {
+    let config: RouterConfig = serde_json::from_str(r#"{"always_match_any_host":false,"ignore_header_case":false,"ignore_host_case":false,"ignore_marketing_query_params":true,"ignore_path_and_query_case":false,"marketing_query_params":["utm_source","utm_medium","utm_campaign","utm_term","utm_content"],"pass_marketing_query_params_to_target":true}"#).expect("cannot deserialize");
+    let mut router = Router::<Rule>::from_config(config);
+
+    let route_1: Rule = serde_json::from_str(r#"{"body_filters":[{"action":"replace_text","content":"{\"pets\":[{\"name\":\"Jo\",\"species\":\"Parrot\",\"birthYear\":2016},{\"name\":\"Charlotte\",\"species\":\"Cat\",\"birthYear\":2008}]}"}],"header_filters":[{"action":"override","header":"Content-Type","value":"application/json"}],"id":"action-custom-body","rank":0,"source":{"path":"/json-body"}}"#).expect("cannot deserialize");
+    router.insert(route_1.into_route(&router.config));
+
+    let route_2: Rule = serde_json::from_str(r#"{"body_filters":[{"action":"replace_text","content":"{\"pets\":[{\"name\":\"Jo\",\"species\":\"Parrot\",\"birthYear\":2016},{\"name\":\"Charlotte\",\"species\":\"Cat\",\"birthYear\":2008}]}"}],"header_filters":[{"action":"override","header":"Content-Type","value":"text/javascript"}],"id":"action-custom-body-with-header","rank":0,"source":{"path":"/json-body-and-header"}}"#).expect("cannot deserialize");
+    router.insert(route_2.into_route(&router.config));
+
+    router
+}
+
+
+#[test]
+fn test_action_custom_body_1() {
+    let router = setup_action_custom_body();
+    let default_config = RouterConfig::default();
+    let request = Request::new(PathAndQueryWithSkipped::from_config(&default_config, r#"/json-body"#), r#"/json-body"#.to_string(),None,None,None,None,None);
+    
+    let request_configured = Request::rebuild_with_config(&router.config, &request);
+    let matched = router.match_request(&request_configured);
+    let traces = router.trace_request(&request_configured);
+    let routes_traces = Trace::<Rule>::get_routes_from_traces(&traces);
+
+    assert_eq!(!matched.is_empty(), true);
+    assert_eq!(!routes_traces.is_empty(), true);
+
+    let mut action = Action::from_routes_rule(matched, &request_configured, None);
+    let response_status_code = 0;
+
+    let action_status_code = action.get_status_code(response_status_code, None);
+    assert_eq!(action_status_code, 0);
+    let body_filter_opt = action.create_filter_body(response_status_code, &[]);
+    assert_eq!(body_filter_opt.is_some(), true);
+
+    let mut body_filter = body_filter_opt.unwrap();
+    let mut new_body = body_filter.filter(r#""#.as_bytes().to_vec(), None);
+    new_body.extend(body_filter.end(None));
+    assert_eq!(new_body, r#"{"pets":[{"name":"Jo","species":"Parrot","birthYear":2016},{"name":"Charlotte","species":"Cat","birthYear":2008}]}"#.as_bytes().to_vec());
+    assert_eq!(action.should_log_request(true, response_status_code, None), true);
+    let mut response_headers = Vec::new();
+
+    response_headers.push(Header {
+        name: r#"Content-Type"#.to_string(),
+        value: r#"text/plain"#.to_string(),
+    });
+
+    let filtered_headers = action.filter_headers(response_headers, response_status_code, false, None);
+    let header_map = Header::create_header_map(filtered_headers);
+
+    let value = header_map.get(r#"Content-Type"#);
+
+    assert!(value.is_some());
+    assert_eq!(value.unwrap(), r#"application/json"#);
+
+}
+
+#[test]
+fn test_action_custom_body_2() {
+    let router = setup_action_custom_body();
+    let default_config = RouterConfig::default();
+    let request = Request::new(PathAndQueryWithSkipped::from_config(&default_config, r#"/json-body-and-header"#), r#"/json-body-and-header"#.to_string(),None,None,None,None,None);
+    
+    let request_configured = Request::rebuild_with_config(&router.config, &request);
+    let matched = router.match_request(&request_configured);
+    let traces = router.trace_request(&request_configured);
+    let routes_traces = Trace::<Rule>::get_routes_from_traces(&traces);
+
+    assert_eq!(!matched.is_empty(), true);
+    assert_eq!(!routes_traces.is_empty(), true);
+
+    let mut action = Action::from_routes_rule(matched, &request_configured, None);
+    let response_status_code = 0;
+
+    let action_status_code = action.get_status_code(response_status_code, None);
+    assert_eq!(action_status_code, 0);
+    let body_filter_opt = action.create_filter_body(response_status_code, &[]);
+    assert_eq!(body_filter_opt.is_some(), true);
+
+    let mut body_filter = body_filter_opt.unwrap();
+    let mut new_body = body_filter.filter(r#""#.as_bytes().to_vec(), None);
+    new_body.extend(body_filter.end(None));
+    assert_eq!(new_body, r#"{"pets":[{"name":"Jo","species":"Parrot","birthYear":2016},{"name":"Charlotte","species":"Cat","birthYear":2008}]}"#.as_bytes().to_vec());
+    assert_eq!(action.should_log_request(true, response_status_code, None), true);
+    let mut response_headers = Vec::new();
+
+    response_headers.push(Header {
+        name: r#"Content-Type"#.to_string(),
+        value: r#"text/plain"#.to_string(),
+    });
+
+    let filtered_headers = action.filter_headers(response_headers, response_status_code, false, None);
+    let header_map = Header::create_header_map(filtered_headers);
+
+    let value = header_map.get(r#"Content-Type"#);
+
+    assert!(value.is_some());
+    assert_eq!(value.unwrap(), r#"text/javascript"#);
+
+}
+
 fn setup_action_disable_log() -> Router<Rule> {
     let config: RouterConfig = serde_json::from_str(r#"{"always_match_any_host":false,"ignore_header_case":false,"ignore_host_case":false,"ignore_marketing_query_params":true,"ignore_path_and_query_case":false,"marketing_query_params":["utm_source","utm_medium","utm_campaign","utm_term","utm_content"],"pass_marketing_query_params_to_target":true}"#).expect("cannot deserialize");
     let mut router = Router::<Rule>::from_config(config);