vetis-compio 0.1.1

Very Tiny Server with compio runtime
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

mod host_tests {

    use crate::host::{path::HandlerPath, HostImpl};
    use http::StatusCode;
    use http_body_util::BodyExt;
    use hyper_body_utils::HttpBody;
    use vetis::host::{handler_fn, Host, HostConfig};

    #[compio::test]
    async fn test_add_host() -> Result<(), Box<dyn std::error::Error>> {
        let config = HostConfig::builder()
            .hostname("localhost")
            .root_directory("src/tests".into())
            .build()
            .unwrap();

        let mut host = HostImpl::new(config);
        let handler_path = HandlerPath::builder()
            .uri("/")
            .handler(handler_fn(|_request| async move {
                Ok(vetis::Response::builder()
                    .status(StatusCode::OK)
                    .text("Hello, world!"))
            }))
            .build()
            .unwrap();
        host.add_path(handler_path);

        assert_eq!(
            host.config()
                .hostname(),
            "localhost"
        );

        Ok(())
    }

    #[compio::test]
    async fn test_handle_request() -> Result<(), Box<dyn std::error::Error>> {
        let config = HostConfig::builder()
            .hostname("localhost")
            .root_directory("src/tests".into())
            .build()
            .unwrap();

        let mut host = HostImpl::new(config);
        let handler_path = HandlerPath::builder()
            .uri("/")
            .handler(handler_fn(|_request| async move {
                Ok(vetis::Response::builder()
                    .status(StatusCode::OK)
                    .text("Hello, world!"))
            }))
            .build()
            .unwrap();
        host.add_path(handler_path);

        assert_eq!(
            host.config()
                .hostname(),
            "localhost"
        );

        let body = HttpBody::from_text("Hello, world!");

        let request = http::Request::builder()
            .uri("/")
            .body(body)
            .unwrap();

        let (parts, body) = request.into_parts();

        let request = vetis::Request::from_parts(parts, body);

        let response = host
            .route(request)
            .await?;

        let (parts, body) = response
            .into_inner()
            .into_parts();
        assert_eq!(parts.status, StatusCode::OK);
        assert_eq!(
            body.collect()
                .await
                .unwrap()
                .to_bytes()
                .as_ref(),
            b"Hello, world!"
        );

        Ok(())
    }
}