105 lines
2.2 KiB
Rust
105 lines
2.2 KiB
Rust
#![deny(warnings)]
|
|
#![feature(collections, test)]
|
|
extern crate hyper;
|
|
|
|
extern crate test;
|
|
|
|
use std::fmt;
|
|
use std::io::{self, Read, Write, Cursor};
|
|
use std::net::SocketAddr;
|
|
|
|
use hyper::net;
|
|
|
|
static README: &'static [u8] = include_bytes!("../README.md");
|
|
|
|
struct MockStream {
|
|
read: Cursor<Vec<u8>>
|
|
}
|
|
|
|
impl MockStream {
|
|
fn new() -> MockStream {
|
|
let head = b"HTTP/1.1 200 OK\r\nServer: Mock\r\n\r\n";
|
|
let mut res = head.to_vec();
|
|
res.push_all(README);
|
|
MockStream {
|
|
read: Cursor::new(res)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Clone for MockStream {
|
|
fn clone(&self) -> MockStream {
|
|
MockStream {
|
|
read: Cursor::new(self.read.get_ref().clone())
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Read for MockStream {
|
|
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
|
self.read.read(buf)
|
|
}
|
|
}
|
|
|
|
impl Write for MockStream {
|
|
fn write(&mut self, msg: &[u8]) -> io::Result<usize> {
|
|
// we're mocking, what do we care.
|
|
Ok(msg.len())
|
|
}
|
|
fn flush(&mut self) -> io::Result<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
struct Foo;
|
|
|
|
impl hyper::header::Header for Foo {
|
|
fn header_name() -> &'static str {
|
|
"x-foo"
|
|
}
|
|
fn parse_header(_: &[Vec<u8>]) -> Option<Foo> {
|
|
None
|
|
}
|
|
}
|
|
|
|
impl hyper::header::HeaderFormat for Foo {
|
|
fn fmt_header(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
|
fmt.write_str("Bar")
|
|
}
|
|
}
|
|
|
|
impl net::NetworkStream for MockStream {
|
|
fn peer_addr(&mut self) -> io::Result<SocketAddr> {
|
|
Ok("127.0.0.1:1337".parse().unwrap())
|
|
}
|
|
}
|
|
|
|
struct MockConnector;
|
|
|
|
impl net::NetworkConnector for MockConnector {
|
|
type Stream = MockStream;
|
|
fn connect(&mut self, _: &str, _: u16, _: &str) -> io::Result<MockStream> {
|
|
Ok(MockStream::new())
|
|
}
|
|
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_mock_hyper(b: &mut test::Bencher) {
|
|
let url = "http://127.0.0.1:1337/";
|
|
b.iter(|| {
|
|
let mut req = hyper::client::Request::with_connector(
|
|
hyper::Get, hyper::Url::parse(url).unwrap(), &mut MockConnector
|
|
).unwrap();
|
|
req.headers_mut().set(Foo);
|
|
|
|
let mut s = String::new();
|
|
req
|
|
.start().unwrap()
|
|
.send().unwrap()
|
|
.read_to_string(&mut s).unwrap()
|
|
});
|
|
}
|
|
|