viz-test 0.4.0

The core traits and types in for Viz
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

use std::collections::HashMap;

use viz::{
    middleware::limits, types, Error, Request, RequestLimitsExt, Response, ResponseExt, Result,
    Router, StatusCode,
};
use viz_test::TestServer;

#[tokio::test]
async fn payload() -> Result<()> {
    let router = Router::new()
        .post("/form", |mut req: Request| async move {
            let data = req.form_with_limit::<HashMap<String, String>>().await?;
            Ok(Response::json(data))
        })
        .post("/json", |mut req: Request| async move {
            let data = req.json_with_limit::<HashMap<String, String>>().await?;
            Ok(Response::json(data))
        })
        .post("/multipart", |mut req: Request| async move {
            let _ = req.multipart_with_limit().await?;
            Ok(())
        })
        .with(
            limits::Config::new().limits(
                types::Limits::new()
                    .set("json", 1)
                    .set("form", 1)
                    .set("multipart", 1),
            ),
        );

    let client = TestServer::new(router).await?;

    let mut form = HashMap::new();
    form.insert("username", "viz");
    form.insert("password", "rs");

    // form
    let resp = client
        .post("/form")
        .json(&form)
        .send()
        .await
        .map_err(Error::boxed)?;
    assert_eq!(resp.status(), StatusCode::UNSUPPORTED_MEDIA_TYPE);
    assert_eq!(
        resp.text().await.map_err(Error::boxed)?,
        "unsupported media type, `application/x-www-form-urlencoded` is required"
    );
    let resp = client
        .post("/form")
        .form(&form)
        .send()
        .await
        .map_err(Error::boxed)?;
    assert_eq!(resp.status(), StatusCode::PAYLOAD_TOO_LARGE);
    assert_eq!(
        resp.text().await.map_err(Error::boxed)?,
        "payload is too large"
    );

    // json
    let resp = client
        .post("/json")
        .form(&form)
        .send()
        .await
        .map_err(Error::boxed)?;
    assert_eq!(resp.status(), StatusCode::UNSUPPORTED_MEDIA_TYPE);
    assert_eq!(
        resp.text().await.map_err(Error::boxed)?,
        "unsupported media type, `application/javascript; charset=utf-8` is required"
    );
    let resp = client
        .post("/json")
        .json(&form)
        .send()
        .await
        .map_err(Error::boxed)?;
    assert_eq!(resp.status(), StatusCode::PAYLOAD_TOO_LARGE);
    assert_eq!(
        resp.text().await.map_err(Error::boxed)?,
        "payload is too large"
    );

    // multipart
    let resp = client
        .post("/multipart")
        .json(&form)
        .send()
        .await
        .map_err(Error::boxed)?;
    assert_eq!(resp.status(), StatusCode::UNSUPPORTED_MEDIA_TYPE);
    assert_eq!(
        resp.text().await.map_err(Error::boxed)?,
        "unsupported media type, `multipart/form-data` is required"
    );
    let form = viz_test::multipart::Form::new()
        .text("key3", "value3")
        .text("key4", "value4");
    let resp = client
        .post("/multipart")
        .multipart(form)
        .send()
        .await
        .map_err(Error::boxed)?;
    assert_eq!(resp.status(), StatusCode::PAYLOAD_TOO_LARGE);
    assert_eq!(
        resp.text().await.map_err(Error::boxed)?,
        "payload is too large"
    );

    Ok(())
}