56 lines
1.5 KiB
Rust
56 lines
1.5 KiB
Rust
use agent::Request;
|
|
use agent::Stream;
|
|
use error::Error;
|
|
use header::Header;
|
|
use std::collections::HashMap;
|
|
use std::io::Write;
|
|
use std::sync::{Arc, Mutex};
|
|
use url::Url;
|
|
use util::vecread::VecRead;
|
|
|
|
mod agent_test;
|
|
mod auth;
|
|
mod body_read;
|
|
mod simple;
|
|
|
|
type RequestHandler = Fn(&Request, &Url) -> Result<Stream, Error> + Send + 'static;
|
|
|
|
lazy_static! {
|
|
pub static ref TEST_HANDLERS: Arc<Mutex<HashMap<String, Box<RequestHandler>>>> =
|
|
{ Arc::new(Mutex::new(HashMap::new())) };
|
|
}
|
|
|
|
pub fn set_handler<H>(path: &str, handler: H)
|
|
where
|
|
H: Fn(&Request, &Url) -> Result<Stream, Error> + Send + 'static,
|
|
{
|
|
let mut handlers = TEST_HANDLERS.lock().unwrap();
|
|
handlers.insert(path.to_string(), Box::new(handler));
|
|
}
|
|
|
|
pub fn make_response(
|
|
status: u16,
|
|
status_text: &str,
|
|
headers: Vec<&str>,
|
|
mut body: Vec<u8>,
|
|
) -> Result<Stream, Error> {
|
|
let mut buf: Vec<u8> = vec![];
|
|
write!(&mut buf, "HTTP/1.1 {} {}\r\n", status, status_text).ok();
|
|
for hstr in headers.iter() {
|
|
let header = hstr.parse::<Header>().unwrap();
|
|
write!(&mut buf, "{}: {}\r\n", header.name(), header.value()).ok();
|
|
}
|
|
write!(&mut buf, "\r\n").ok();
|
|
buf.append(&mut body);
|
|
let read = VecRead::from_vec(buf);
|
|
let write: Vec<u8> = vec![];
|
|
Ok(Stream::Test(Box::new(read), write))
|
|
}
|
|
|
|
pub fn resolve_handler(req: &Request, url: &Url) -> Result<Stream, Error> {
|
|
let mut handlers = TEST_HANDLERS.lock().unwrap();
|
|
let path = url.path();
|
|
let handler = handlers.remove(path).unwrap();
|
|
handler(req, url)
|
|
}
|