async-httplib 0.5.0

Low-level HTTP helpers.
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

use std::collections::HashMap;
use async_httplib::{read_first_line, read_header_line, read_exact, parse_method,
    parse_status, parse_version, Method, Status, Version};

#[async_std::test]
async fn reads_request() {
    let mut stream = String::new();
    stream.push_str("POST /path HTTP/1.1\r\n");
    stream.push_str("Host: google.com\r\n");
    stream.push_str("Content-Length: 5\r\n");
    stream.push_str("\r\n");
    stream.push_str("hello");
    let mut stream = stream.as_bytes();

    let (mut method, mut uri, mut version) = (vec![], vec![], vec![]);
    let size = read_first_line(&mut stream, (&mut method, &mut uri, &mut version), None).await.unwrap();
    assert_eq!(size, 21);

    let method = parse_method(method).unwrap();
    let uri = String::from_utf8(uri).unwrap();
    let version = parse_version(version).unwrap();
    assert_eq!(method, Method::Post);
    assert_eq!(uri, "/path");
    assert_eq!(version, Version::Http1_1);

    let mut headers: HashMap<String, String> = HashMap::new();
    loop {
        let (mut name, mut value) = (vec![], vec![]);
        read_header_line(&mut stream, (&mut name, &mut value), None).await.unwrap();
        if name.len() == 0 {
            break;
        } else {
            headers.insert(
                String::from_utf8(name).unwrap(),
                String::from_utf8(value).unwrap(),
            );
        }
    }
    assert_eq!(headers.len(), 2);
    assert_eq!(headers.get("Host").unwrap(), "google.com");
    assert_eq!(headers.get("Content-Length").unwrap(), "5");

    let length = headers.get("Content-Length").unwrap().parse::<usize>().unwrap();
    let mut body = Vec::new();
    let size = read_exact(&mut stream, &mut body, length).await.unwrap();
    assert_eq!(size, 5);
    assert_eq!(String::from_utf8(body).unwrap(), "hello");
}

#[async_std::test]
async fn reads_response() {
    let mut stream = String::new();
    stream.push_str("HTTP/1.1 200 OK\r\n");
    stream.push_str("Content-Length: 5\r\n");
    stream.push_str("\r\n");
    stream.push_str("hello");
    let mut stream = stream.as_bytes();

    let (mut version, mut status, mut message) = (vec![], vec![], vec![]);
    let size = read_first_line(&mut stream, (&mut version, &mut status, &mut message), None).await.unwrap();
    assert_eq!(size, 17);

    let version = parse_version(version).unwrap();
    let status = parse_status(status).unwrap();
    assert_eq!(version, Version::Http1_1);
    assert_eq!(status, Status::Ok);

    let mut headers: HashMap<String, String> = HashMap::new();
    loop {
        let (mut name, mut value) = (vec![], vec![]);
        read_header_line(&mut stream, (&mut name, &mut value), None).await.unwrap();
        if name.len() == 0 {
            break;
        } else {
            headers.insert(
                String::from_utf8(name).unwrap(),
                String::from_utf8(value).unwrap(),
            );
        }
    }
    assert_eq!(headers.len(), 1);
    assert_eq!(headers.get("Content-Length").unwrap(), "5");

    let length = headers.get("Content-Length").unwrap().parse::<usize>().unwrap();
    let mut body = Vec::new();
    let size = read_exact(&mut stream, &mut body, length).await.unwrap();
    assert_eq!(size, 5);
    assert_eq!(String::from_utf8(body).unwrap(), "hello");
}