humphrey 0.7.0

A Performance-Focused, Dependency-Free Web Server.
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

#![allow(unused_imports)]
use crate::http::cookie::{SameSite, SetCookie};
use crate::http::headers::{HeaderType, Headers};
use crate::http::response::Response;
use crate::http::status::StatusCode;
use crate::tests::mock_stream::MockStream;

use std::collections::{BTreeMap, VecDeque};
use std::iter::FromIterator;
use std::time::Duration;

#[test]
fn test_response() {
    let response = Response::empty(StatusCode::OK)
        .with_bytes(b"<body>test</body>")
        .with_header(HeaderType::ContentType, "text/html")
        .with_header(HeaderType::ContentLanguage, "en-GB")
        .with_header(HeaderType::Date, "Thu, 1 Jan 1970 00:00:00 GMT"); // this would never be manually set in prod, but is obviously required for testing

    assert_eq!(
        response.get_headers().get(&HeaderType::ContentType),
        Some("text/html")
    );

    let expected_bytes: Vec<u8> = b"HTTP/1.1 200 OK\r\nDate: Thu, 1 Jan 1970 00:00:00 GMT\r\nContent-Language: en-GB\r\nContent-Type: text/html\r\n\r\n<body>test</body>\r\n".to_vec();
    let bytes: Vec<u8> = response.into();

    assert_eq!(bytes, expected_bytes);
}

#[test]
fn test_cookie_response() {
    let response = Response::empty(StatusCode::OK)
        .with_bytes(b"Hello, world!")
        .with_cookie(
            SetCookie::new("X-Example-Cookie", "example-value")
                .with_path("/")
                .with_max_age(Duration::from_secs(3600))
                .with_secure(true),
        )
        .with_cookie(
            SetCookie::new("X-Example-Token", "example-token")
                .with_domain("example.com")
                .with_same_site(SameSite::Strict)
                .with_secure(true),
        );

    assert_eq!(
        response.get_headers().get_all(&HeaderType::SetCookie),
        vec![
            "X-Example-Cookie=example-value; Max-Age=3600; Path=/; Secure",
            "X-Example-Token=example-token; Domain=example.com; SameSite=Strict; Secure"
        ]
    );

    let expected_bytes: Vec<u8> =
        b"HTTP/1.1 200 OK\r\nSet-Cookie: X-Example-Cookie=example-value; Max-Age=3600; Path=/; Secure\r\nSet-Cookie: X-Example-Token=example-token; Domain=example.com; SameSite=Strict; Secure\r\n\r\nHello, world!\r\n"
            .to_vec();
    let bytes: Vec<u8> = response.into();

    assert_eq!(bytes, expected_bytes);
}

#[test]
fn test_response_from_stream() {
    let test_data = b"HTTP/1.1 404 Not Found\r\nContent-Length: 51\r\n\r\nThe requested resource was not found on the server.\r\n";
    let mut stream = MockStream::with_data(VecDeque::from_iter(test_data.iter().cloned()));
    let response = Response::from_stream(&mut stream);

    assert!(response.is_ok());

    let response = response.unwrap();
    let expected_body = b"The requested resource was not found on the server.".to_vec();
    assert_eq!(response.body, expected_body);
    assert_eq!(response.version, "HTTP/1.1".to_string());
    assert_eq!(response.status_code, StatusCode::NotFound);

    let mut expected_headers: Headers = Headers::new();
    expected_headers.add(HeaderType::ContentLength, "51");
    assert_eq!(response.headers, expected_headers);
}