rama 0.2.0

modular service framework
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

use super::utils;
use rama::{
    Context, Layer,
    http::service::web::response::{IntoResponse, Json},
    http::{BodyExtractExt, Request, headers::Accept, server::HttpServer},
    net::address::ProxyAddress,
    rt::Executor,
    service::service_fn,
};
use serde_json::{Value, json};

#[cfg(feature = "compression")]
use rama::http::layer::compression::CompressionLayer;

#[tokio::test]
#[ignore]
async fn test_https_connect_proxy() {
    utils::init_tracing();

    tokio::spawn(async {
        HttpServer::auto(Executor::default())
            .listen(
                "127.0.0.1:63002",
                (
                    #[cfg(feature = "compression")]
                    CompressionLayer::new(),
                )
                    .into_layer(service_fn(async |req: Request| {
                        tracing::debug!(uri = %req.uri(), "serve request");
                        Ok(Json(json!({
                            "method": req.method().as_str(),
                            "path": req.uri().path(),
                        }))
                        .into_response())
                    })),
            )
            .await
            .unwrap();
    });

    let runner = utils::ExampleRunner::interactive("https_connect_proxy", Some("rustls"));

    let mut ctx = Context::default();
    ctx.insert(ProxyAddress::try_from("https://john:secret@127.0.0.1:62016").unwrap());

    // test regular proxy flow
    let result = runner
        .get("http://127.0.0.1:63002/foo/bar")
        .typed_header(Accept::json())
        .send(ctx.clone())
        .await
        .unwrap()
        .try_into_json::<Value>()
        .await
        .unwrap();
    let expected_value = json!({"method":"GET","path":"/foo/bar"});
    assert_eq!(expected_value, result);
}