foxy-io 0.2.19

A configuration-driven and hyper-extensible HTTP proxy library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.

#[cfg(test)]
mod tests {
    use crate::{
        HttpMethod, ProxyRequest, ProxyResponse, ProxyError,
        SecurityProvider, SecurityChain, SecurityStage
    };
    use crate::core::RequestContext;
    use async_trait::async_trait;
    use reqwest::Body;
    use std::sync::Arc;
    use tokio::sync::RwLock;

    // Helper function to create a test request
    fn create_test_request(method: HttpMethod, path: &str, headers: Vec<(&'static str, &'static str)>, target: &str) -> ProxyRequest {
        let mut header_map = reqwest::header::HeaderMap::new();
        for (name, value) in headers {
            header_map.insert(
                reqwest::header::HeaderName::from_static(name),
                reqwest::header::HeaderValue::from_static(value),
            );
        }

        ProxyRequest {
            method,
            path: path.to_string(),
            query: None,
            headers: header_map,
            body: Body::from(Vec::new()),
            context: Arc::new(RwLock::new(RequestContext::default())),
            custom_target: Some(target.to_string()),
        }
    }

    #[tokio::test]
    async fn test_security_chain_bypass_routes() {
        // Create a security chain with a mock provider
        let mut chain = SecurityChain::new(vec!["/health".to_string(), "/public/".to_string()]);

        // Add a mock provider
        let mock_provider = MockSecurityProvider::new();

        chain.add(Arc::new(mock_provider));

        // Test bypass routes
        let request = create_test_request(HttpMethod::Get, "/health", vec![], "http://test.co.za");
        let result = chain.apply_pre(request).await;
        assert!(result.is_ok());

        let request = create_test_request(HttpMethod::Post, "/health", vec![], "http://test.co.za");
        let result = chain.apply_pre(request).await;
        assert!(result.is_ok());

        let request = create_test_request(HttpMethod::Get, "/public/docs", vec![], "http://test.co.za");
        let result = chain.apply_pre(request).await;
        assert!(result.is_ok());

        // Test non-bypass route
        let request = create_test_request(HttpMethod::Get, "/api/users", vec![], "http://test.co.za");
        let result = chain.apply_pre(request).await;
        assert!(result.is_err());
    }

    // Mock implementations for testing
    #[derive(Debug)]
    struct MockSecurityProvider {}

    impl MockSecurityProvider {
        fn new() -> Self {
            Self {}
        }
    }

    #[async_trait]
    impl SecurityProvider for MockSecurityProvider {
        fn stage(&self) -> SecurityStage {
            SecurityStage::Both
        }

        fn name(&self) -> &str {
            "mock-provider"
        }

        async fn pre(&self, request: ProxyRequest) -> Result<ProxyRequest, ProxyError> {
            // This mock always fails authentication unless bypassed
            Err(ProxyError::SecurityError("Mock authentication failure".to_string()))
        }
    }
}