warpdrive 0.1.0

A compatibility library for running Warp filters within Axum servers.
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

use axum::http::StatusCode as AxumStatusCode;
use serde_json::json;
use warp::{
    Reply,
    http::{response::Response as WarpResponse, status::StatusCode as WarpStatusCode},
    hyper::Body as WarpBody,
    reply::{json, with_header, with_status},
};

use crate::convert_response::into_axum_response;

#[tokio::test]
async fn test_basic_response() {
    let warp_response = WarpResponse::builder()
        .status(WarpStatusCode::OK)
        .body(WarpBody::from("Hello World!"))
        .unwrap();

    let axum_response = into_axum_response(warp_response).await.unwrap();

    assert_eq!(axum_response.status(), AxumStatusCode::OK);
}

#[tokio::test]
async fn test_response_with_status() {
    let warp_response = WarpResponse::builder()
        .status(WarpStatusCode::NOT_FOUND)
        .body(WarpBody::empty())
        .unwrap();

    let axum_response = into_axum_response(warp_response).await.unwrap();

    assert_eq!(axum_response.status(), AxumStatusCode::NOT_FOUND);
}

#[tokio::test]
async fn test_response_with_headers() {
    let warp_response = WarpResponse::builder()
        .header(warp::http::header::CONTENT_TYPE, "text/plain")
        .body(WarpBody::empty())
        .unwrap();

    let axum_response = into_axum_response(warp_response).await.unwrap();

    assert_eq!(
        axum_response
            .headers()
            .get(axum::http::header::CONTENT_TYPE)
            .unwrap(),
        "text/plain"
    );
}

#[tokio::test]
async fn test_json_response() {
    let data = json!({
        "message": "Hello World!"
    });
    let warp_response = json(&data).into_response();

    let axum_response = into_axum_response(warp_response).await.unwrap();

    assert_eq!(
        axum_response
            .headers()
            .get(axum::http::header::CONTENT_TYPE)
            .unwrap(),
        "application/json"
    );
}

#[tokio::test]
async fn test_multiple_headers() {
    let response = with_header(
        with_header("Hello World!", "X-Custom-Header", "custom-value"),
        "X-Another-Header",
        "another-value",
    )
    .into_response();

    let axum_response = into_axum_response(response).await.unwrap();

    assert_eq!(
        axum_response.headers().get("X-Custom-Header").unwrap(),
        "custom-value"
    );
    assert_eq!(
        axum_response.headers().get("X-Another-Header").unwrap(),
        "another-value"
    );
}

#[tokio::test]
async fn test_different_http_versions() {
    use axum::http::Version as AxumVersion;
    use warp::http::Version as WarpVersion;

    let versions = vec![
        WarpVersion::HTTP_09,
        WarpVersion::HTTP_10,
        WarpVersion::HTTP_11,
        WarpVersion::HTTP_2,
        WarpVersion::HTTP_3,
    ];

    for version in versions {
        let response = WarpResponse::builder()
            .version(version)
            .body(WarpBody::from("Hello"))
            .unwrap();

        let axum_response = into_axum_response(response).await.unwrap();

        // Version should be preserved or fallback to HTTP_11
        assert!(matches!(
            axum_response.version(),
            AxumVersion::HTTP_09
                | AxumVersion::HTTP_10
                | AxumVersion::HTTP_11
                | AxumVersion::HTTP_2
                | AxumVersion::HTTP_3
        ));
    }
}

#[tokio::test]
async fn test_complex_response() {
    let data = json!({
        "status": "success",
        "data": {
            "message": "Hello World!",
            "code": 200
        }
    });

    let response = with_header(
        with_status(json(&data), WarpStatusCode::OK),
        "X-Rate-Limit",
        "100",
    )
    .into_response();

    let axum_response = into_axum_response(response).await.unwrap();

    assert_eq!(axum_response.status(), AxumStatusCode::OK);
    assert_eq!(
        axum_response
            .headers()
            .get(axum::http::header::CONTENT_TYPE)
            .unwrap(),
        "application/json"
    );
    assert_eq!(axum_response.headers().get("X-Rate-Limit").unwrap(), "100");
}