typeway-server 0.1.0

Server runtime for the typeway type-level web 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
122

//! Integration tests for multiple API versions served simultaneously.

use std::sync::Arc;
use std::time::Duration;

use typeway_core::*;
use typeway_macros::*;
use typeway_server::typed::*;
use typeway_server::*;

// --- Setup ---

typeway_path!(type UsersPath = "users");

#[derive(serde::Serialize)]
struct UserV1 {
    name: String,
}
#[derive(serde::Serialize)]
struct UserV2 {
    name: String,
    email: String,
}

struct V1;
impl ApiVersion for V1 {
    const PREFIX: &'static str = "v1";
}

struct V2;
impl ApiVersion for V2 {
    const PREFIX: &'static str = "v2";
}

// Two versioned endpoints in the same API.
type API = (
    Versioned<V1, GetEndpoint<UsersPath, Vec<UserV1>>>,
    Versioned<V2, GetEndpoint<UsersPath, Vec<UserV2>>>,
);

async fn get_users_v1() -> Json<Vec<UserV1>> {
    Json(vec![UserV1 {
        name: "Alice".into(),
    }])
}

async fn get_users_v2() -> Json<Vec<UserV2>> {
    Json(vec![UserV2 {
        name: "Alice".into(),
        email: "alice@example.com".into(),
    }])
}

async fn start_server() -> u16 {
    let server = Server::<API>::new((bind!(get_users_v1), bind!(get_users_v2)));
    let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap();
    let port = listener.local_addr().unwrap().port();

    tokio::spawn(async move {
        let router = Arc::new(server.into_router());
        loop {
            let (stream, _) = listener.accept().await.unwrap();
            let io = hyper_util::rt::TokioIo::new(stream);
            let svc = RouterService::new(router.clone());
            let hyper_svc = hyper_util::service::TowerToHyperService::new(svc);
            tokio::spawn(async move {
                let _ = hyper::server::conn::http1::Builder::new()
                    .serve_connection(io, hyper_svc)
                    .await;
            });
        }
    });

    tokio::time::sleep(Duration::from_millis(50)).await;
    port
}

#[tokio::test]
async fn v1_returns_v1_response() {
    let port = start_server().await;
    let resp = reqwest::get(format!("http://127.0.0.1:{port}/v1/users"))
        .await
        .unwrap();
    assert_eq!(resp.status(), 200);
    let body: serde_json::Value = resp.json().await.unwrap();
    let users = body.as_array().unwrap();
    assert_eq!(users.len(), 1);
    assert_eq!(users[0]["name"], "Alice");
    assert!(users[0].get("email").is_none());
}

#[tokio::test]
async fn v2_returns_v2_response() {
    let port = start_server().await;
    let resp = reqwest::get(format!("http://127.0.0.1:{port}/v2/users"))
        .await
        .unwrap();
    assert_eq!(resp.status(), 200);
    let body: serde_json::Value = resp.json().await.unwrap();
    let users = body.as_array().unwrap();
    assert_eq!(users.len(), 1);
    assert_eq!(users[0]["name"], "Alice");
    assert_eq!(users[0]["email"], "alice@example.com");
}

#[tokio::test]
async fn unversioned_path_returns_404() {
    let port = start_server().await;
    let resp = reqwest::get(format!("http://127.0.0.1:{port}/users"))
        .await
        .unwrap();
    assert_eq!(resp.status(), 404);
}

#[tokio::test]
async fn wrong_version_returns_404() {
    let port = start_server().await;
    let resp = reqwest::get(format!("http://127.0.0.1:{port}/v3/users"))
        .await
        .unwrap();
    assert_eq!(resp.status(), 404);
}