use header::{Header, HeaderFormat};
use std::fmt;
use header::parsing::from_one_raw_str;
#[derive(Clone, PartialEq, Debug)]
pub struct UserAgent(pub String);
deref!(UserAgent => String);
impl Header for UserAgent {
fn header_name() -> &'static str {
"User-Agent"
}
fn parse_header(raw: &[Vec<u8>]) -> Option<UserAgent> {
from_one_raw_str(raw).map(|s| UserAgent(s))
}
}
impl HeaderFormat for UserAgent {
fn fmt_header(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
fmt.write_str(&*self.0)
}
}
bench_header!(bench, UserAgent, { vec![b"cargo bench".to_vec()] });
#[test] fn test_format() {
use std::borrow::ToOwned;
use header::Headers;
let mut head = Headers::new();
head.set(UserAgent("Bunnies".to_owned()));
assert!(head.to_string() == "User-Agent: Bunnies\r\n".to_owned());
}