Skip to content

Commit ba6452a

Browse files
pietroalbinijplatte
authored andcommitted
link: update tests to the new headers API
1 parent 4fa2b1a commit ba6452a

File tree

1 file changed

+18
-14
lines changed

1 file changed

+18
-14
lines changed

src/common/link.rs

Lines changed: 18 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -60,14 +60,14 @@ use {Header, Raw};
6060
/// # Examples
6161
///
6262
/// ```
63-
/// use headers::{Headers, link::{Link, LinkValue, RelationType}};
63+
/// use headers::{HeaderMap, HeaderMapExt, link::{Link, LinkValue, RelationType}};
6464
///
6565
/// let link_value = LinkValue::new("http://example.com/TheBook/chapter2")
6666
/// .push_rel(RelationType::Previous)
6767
/// .set_title("previous chapter");
6868
///
69-
/// let mut headers = Headers::new();
70-
/// headers.set(
69+
/// let mut headers = HeaderMap::new();
70+
/// headers.typed_insert(
7171
/// Link::new(vec![link_value])
7272
/// );
7373
/// ```
@@ -901,11 +901,15 @@ mod tests {
901901

902902
use Header;
903903

904-
// use proto::ServerTransaction;
905-
use bytes::BytesMut;
906-
907904
use mime;
908905

906+
fn parse_header(values: &[&[u8]]) -> Result<Link, ::Error> {
907+
let values = values.iter()
908+
.map(|val| ::HeaderValue::from_bytes(val).expect("invalid header value"))
909+
.collect::<Vec<_>>();
910+
Link::decode(&mut values.iter())
911+
}
912+
909913
#[test]
910914
fn test_link() {
911915
let link_value = LinkValue::new("http://example.com/TheBook/chapter2")
@@ -918,7 +922,7 @@ mod tests {
918922

919923
let expected_link = Link::new(vec![link_value]);
920924

921-
let link = Header::parse_header(&vec![link_header.to_vec()].into());
925+
let link = parse_header(&[link_header]);
922926
assert_eq!(link.ok(), Some(expected_link));
923927
}
924928

@@ -939,7 +943,7 @@ mod tests {
939943

940944
let expected_link = Link::new(vec![first_link, second_link]);
941945

942-
let link = Header::parse_header(&vec![link_header.to_vec()].into());
946+
let link = parse_header(&[link_header]);
943947
assert_eq!(link.ok(), Some(expected_link));
944948
}
945949

@@ -963,7 +967,7 @@ mod tests {
963967

964968
let expected_link = Link::new(vec![link_value]);
965969

966-
let link = Header::parse_header(&vec![link_header.to_vec()].into());
970+
let link = parse_header(&[link_header]);
967971
assert_eq!(link.ok(), Some(expected_link));
968972
}
969973

@@ -1034,31 +1038,31 @@ mod tests {
10341038
let link_a = b"http://example.com/TheBook/chapter2; \
10351039
rel=\"previous\"; rev=next; title=\"previous chapter\"";
10361040

1037-
let mut err: Result<Link, _> = Header::parse_header(&vec![link_a.to_vec()].into());
1041+
let mut err: Result<Link, _> = parse_header(&[link_a]);
10381042
assert_eq!(err.is_err(), true);
10391043

10401044
let link_b = b"<http://example.com/TheBook/chapter2>; \
10411045
=\"previous\"; rev=next; title=\"previous chapter\"";
10421046

1043-
err = Header::parse_header(&vec![link_b.to_vec()].into());
1047+
err = parse_header(&[link_b]);
10441048
assert_eq!(err.is_err(), true);
10451049

10461050
let link_c = b"<http://example.com/TheBook/chapter2>; \
10471051
rel=; rev=next; title=\"previous chapter\"";
10481052

1049-
err = Header::parse_header(&vec![link_c.to_vec()].into());
1053+
err = parse_header(&[link_c]);
10501054
assert_eq!(err.is_err(), true);
10511055

10521056
let link_d = b"<http://example.com/TheBook/chapter2>; \
10531057
rel=\"previous\"; rev=next; title=";
10541058

1055-
err = Header::parse_header(&vec![link_d.to_vec()].into());
1059+
err = parse_header(&[link_d]);
10561060
assert_eq!(err.is_err(), true);
10571061

10581062
let link_e = b"<http://example.com/TheBook/chapter2>; \
10591063
rel=\"previous\"; rev=next; attr=unknown";
10601064

1061-
err = Header::parse_header(&vec![link_e.to_vec()].into());
1065+
err = parse_header(&[link_e]);
10621066
assert_eq!(err.is_err(), true);
10631067
}
10641068

0 commit comments

Comments
 (0)