rama 0.4.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
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use super::utils;
use rama::{extensions::Extensions, tcp::client::default_tcp_connect, telemetry::tracing};
use rama_net::address::HostWithPort;
use tokio::io::AsyncReadExt as _;

#[cfg(feature = "boring")]
use rama::{
    net::client::{ConnectorService, EstablishedClientConnection},
    tcp::client::service::TcpConnector,
    tls::boring::client::TlsConnector,
    tls::client::{ServerVerifyMode, TlsClientConfig},
};

#[cfg(feature = "boring")]
use rama_net::client::ConnectRequest;

#[tokio::test]
#[ignore]
async fn test_http_ip() {
    utils::init_tracing();
    let _guard = utils::RamaService::serve_ip(63100, false, false);
    test_http_ip_inner("http://127.0.0.1:63100");
}

#[ignore]
#[tokio::test]
#[cfg(feature = "boring")]
async fn test_https_ip() {
    utils::init_tracing();
    let _guard = utils::RamaService::serve_ip(63118, false, true);
    test_http_ip_inner("https://127.0.0.1:63118");
}

fn test_http_ip_inner(addr: &'static str) {
    // default: txt
    let lines = utils::RamaService::http(vec!["--http1.1", addr]).unwrap();
    assert!(lines.contains("HTTP/1.1 200 OK"));
    assert!(lines.contains("content-type: text/plain; charset=utf-8"));
    assert!(
        lines.split("\r\n").any(|line| line.contains("127.0.0.1")),
        "txt; lines: {lines}"
    );

    // json
    let lines = utils::RamaService::http(vec!["--http1.1", addr, "-H", "accept: application/json"])
        .unwrap();
    assert!(lines.contains("HTTP/1.1 200 OK"));
    assert!(lines.contains("content-type: application/json"));
    assert!(lines.contains(r##""127.0.0.1""##), "json; lines: {lines}");

    // html
    let lines =
        utils::RamaService::http(vec!["--http1.1", addr, "-H", "accept: text/html"]).unwrap();
    assert!(lines.contains("HTTP/1.1 200 OK"));
    assert!(lines.contains("content-type: text/html; charset=utf-8"));
    assert!(
        lines.contains("<code>127.0.0.1</code>"),
        "html; lines: {lines}"
    );
}

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

    let _guard = utils::RamaService::serve_ip(63119, true, false);

    let mut stream = None;
    for i in 0..5 {
        let extensions = Extensions::new();
        match default_tcp_connect(&extensions, HostWithPort::local_ipv4(63119)).await {
            Ok((s, _)) => {
                stream = Some(s);
                break;
            }
            Err(e) => {
                tracing::error!("connect_tcp error: {e}");
                tokio::time::sleep(std::time::Duration::from_millis(500 + 250 * i)).await;
            }
        }
    }
    let mut stream = stream.expect("connect to tls-tcp listener");

    let mut buf = [0; 4];
    stream.read_exact(&mut buf).await.unwrap();
    assert_eq!(&buf, &[127, 0, 0, 1]);
}

#[ignore]
#[tokio::test]
#[cfg(feature = "boring")]
async fn test_tls_tcp_ip() {
    utils::init_tracing();

    let _guard = utils::RamaService::serve_ip(63120, true, true);

    let mut stream = None;
    for i in 0..5 {
        let connector = TlsConnector::secure(TcpConnector::new())
            .with_base_config(TlsClientConfig::new().with_server_verify(ServerVerifyMode::Disable));
        match connector
            .connect(ConnectRequest::new(HostWithPort::local_ipv4(63120)))
            .await
        {
            Ok(EstablishedClientConnection { conn, .. }) => {
                stream = Some(conn);
                break;
            }
            Err(e) => {
                tracing::error!("tls(tcp) connect error: {e}");
                tokio::time::sleep(std::time::Duration::from_millis(500 + 250 * i)).await;
            }
        }
    }
    let mut stream = stream.expect("connect to tls-tcp listener");

    let mut buf = [0; 4];
    stream.read_exact(&mut buf).await.unwrap();
    assert_eq!(&buf, &[127, 0, 0, 1]);
}