witmproxy 0.0.1-alpha

A WASM-in-the-middle proxy
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

mod tests {
    use crate::db::Db;
    use crate::plugins::registry::PluginRegistry;
    use crate::test_utils::{create_ca_and_config, test_component_path};
    use crate::wasm::Runtime;
    use crate::web::WebServer;
    use anyhow::Result;
    use std::sync::Arc;
    use tempfile::tempdir;
    use tokio::sync::RwLock;

    #[tokio::test]
    async fn test_plugin_upsert() -> Result<()> {
        let _ = rustls::crypto::ring::default_provider().install_default();

        // Create CA and config
        let (ca, config) = create_ca_and_config().await;

        // Create a temporary database for testing
        let temp_dir = tempdir().unwrap();
        let db_path = temp_dir.path().join("test.db");
        let db = Db::from_path(db_path, "test_password").await.unwrap();
        db.migrate().await.unwrap();

        // Create runtime
        let runtime = Runtime::try_default().unwrap();

        // Create plugin registry
        let plugin_registry = Arc::new(RwLock::new(PluginRegistry::new(db, runtime)?));

        let mut web_server = WebServer::new(ca.clone(), Some(plugin_registry), config);
        web_server.start().await.unwrap();
        let bind_addr = web_server.listen_addr().unwrap();

        let wasm_path = test_component_path()?;
        let component_bytes = std::fs::read(&wasm_path)?;

        // Create a temporary file with the component bytes for upload
        let temp_file = tempfile::NamedTempFile::new().unwrap();
        std::fs::write(temp_file.path(), &component_bytes).unwrap();

        // Create form with file upload
        let form = reqwest::multipart::Form::new()
            .file("file", temp_file.path())
            .await
            .unwrap();

        // Create HTTP client that accepts self-signed certificates
        let client = reqwest::Client::builder()
            .danger_accept_invalid_certs(true)
            .build()
            .unwrap();

        let response = client
            .post(format!("https://{}/api/plugins", bind_addr))
            .multipart(form)
            .send()
            .await
            .unwrap();

        assert!(
            response.status().is_success(),
            "Expected successful response, got: {} - {}",
            response.status(),
            response
                .text()
                .await
                .unwrap_or_else(|_| "Unable to read response body".to_string())
        );
        Ok(())
    }
}