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
|
//! Helper functions for the directory client code
use std::fmt::Write;
use crate::body::{EncodedRequest, RequestBody};
/// Encode an HTTP request in a quick and dirty HTTP 1.0 format.
pub(crate) fn encode_request(req: &http::Request<RequestBody>) -> EncodedRequest {
let mut s = format!("{} {} HTTP/1.0\r\n", req.method(), req.uri());
for (key, val) in req.headers().iter() {
write!(
s,
"{}: {}\r\n",
key,
val.to_str()
.expect("Added an HTTP header that wasn't UTF-8!")
)
.expect("Write to string failed");
}
if req.method() == http::Method::POST || !req.body().is_empty() {
write!(s, "Content-Length: {}\r\n", req.body().len())
.expect("Added an HTTP header that wasn't UTF-8!");
}
s.push_str("\r\n");
let mut body = EncodedRequest::from_header(s);
body.set_body(req.body().clone());
body
}
/// Testing helper: convert a request to a String.
///
/// # Panics
///
/// Panics if the request's body is not valid UTF-8
#[cfg(test)]
pub(crate) fn request_to_string(req: &http::Request<RequestBody>) -> String {
let body = encode_request(req);
let bytes: Vec<u8> = body.to_owned();
String::from_utf8(bytes).expect("Body was not UTF-8")
}
#[cfg(test)]
mod test {
// @@ begin test lint list maintained by maint/add_warning @@
#![allow(clippy::bool_assert_comparison)]
#![allow(clippy::clone_on_copy)]
#![allow(clippy::dbg_macro)]
#![allow(clippy::mixed_attributes_style)]
#![allow(clippy::print_stderr)]
#![allow(clippy::print_stdout)]
#![allow(clippy::single_char_pattern)]
#![allow(clippy::unwrap_used)]
#![allow(clippy::unchecked_time_subtraction)]
#![allow(clippy::useless_vec)]
#![allow(clippy::needless_pass_by_value)]
#![allow(clippy::string_slice)] // See arti#2571
//! <!-- @@ end test lint list maintained by maint/add_warning @@ -->
use std::sync::Arc;
use super::*;
fn build_request(body: String, headers: &[(&str, &str)]) -> http::Request<RequestBody> {
let mut builder = http::Request::builder().method("GET").uri("/index.html");
for (name, value) in headers {
builder = builder.header(*name, *value);
}
let body = RequestBody::from(Arc::<str>::from(body));
builder.body(body).unwrap()
}
#[test]
fn format() {
fn chk_format(body: &str, content_length_expected: &str) {
let req = build_request(body.to_string(), &[]);
assert_eq!(
request_to_string(&req),
format!("GET /index.html HTTP/1.0\r\n{content_length_expected}\r\n{body}")
);
let req = build_request(body.to_string(), &[("X-Marsupial", "Opossum")]);
assert_eq!(
request_to_string(&req),
format!(
"GET /index.html HTTP/1.0\r\nx-marsupial: Opossum\r\n{content_length_expected}\r\n{body}",
)
);
}
chk_format("", "");
chk_format("hello", "Content-Length: 5\r\n");
}
}
|
