blob: 890dead4fb3d271abd16bfa16b377e958263c94a (
plain)
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
|
use hyper::{Request, Body, Method};
use std::path::Path;
use std::io::BufRead;
pub struct ExampleParts {
pub method: String,
pub path_and_query: String,
pub body: Option<String>,
}
pub fn load_parts_by_name(name: &str) -> ExampleParts {
let path = format!("tests/files/{}.txt", name);
let path = Path::new(&path);
load_parts(&path)
}
pub fn load_parts(path: &Path) -> ExampleParts {
let file = std::fs::File::open(path).unwrap();
let mut lines = std::io::BufReader::new(file).lines();
let first_line: Vec<String> = lines.next().unwrap().unwrap().split(" ").map(|v| v.into()).collect();
let body: Vec<String> = lines.map(|v| v.into()).collect::<Result<Vec<String>, _>>().unwrap();
let body: Option<String> = if body.len() <= 1 {
None
} else {
Some(body.join("\n"))
};
ExampleParts {
method: first_line[0].clone(),
path_and_query: first_line[1].clone(),
body: body,
}
}
pub fn load_request(path: &Path) -> Request<Body> {
let parts = load_parts(path);
Request::builder()
.uri(parts.path_and_query)
.method(Method::from_bytes(parts.method.as_bytes()).expect("valid method in example text file"))
.body(match parts.body {
Some(data) => Body::from(data),
None => Body::empty(),
})
.expect("constructing upstream request")
}
|
