From 15d72b16949bb9f8bdb97a7521d212ac9b77c5c2 Mon Sep 17 00:00:00 2001 From: Rob Ede Date: Mon, 22 Mar 2021 05:30:16 +0000 Subject: [PATCH] use insert for protobuf content type header --- actix-protobuf/src/lib.rs | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/actix-protobuf/src/lib.rs b/actix-protobuf/src/lib.rs index 0649d9e6c..3100c2858 100644 --- a/actix-protobuf/src/lib.rs +++ b/actix-protobuf/src/lib.rs @@ -250,7 +250,7 @@ pub trait ProtoBufResponseBuilder { impl ProtoBufResponseBuilder for HttpResponseBuilder { fn protobuf(&mut self, value: T) -> Result { - self.append_header((CONTENT_TYPE, "application/protobuf")); + self.insert_header((CONTENT_TYPE, "application/protobuf")); let mut body = Vec::new(); value @@ -296,10 +296,8 @@ mod tests { }); let req = TestRequest::default().to_http_request(); let resp = protobuf.respond_to(&req).await.unwrap(); - assert_eq!( - resp.headers().get(header::CONTENT_TYPE).unwrap(), - "application/protobuf" - ); + let ct = resp.headers().get(header::CONTENT_TYPE).unwrap(); + assert_eq!(ct, "application/protobuf"); } #[actix_rt::test] @@ -309,14 +307,14 @@ mod tests { assert_eq!(protobuf.err().unwrap(), ProtoBufPayloadError::ContentType); let (req, mut pl) = TestRequest::get() - .append_header((header::CONTENT_TYPE, "application/text")) + .insert_header((header::CONTENT_TYPE, "application/text")) .to_http_parts(); let protobuf = ProtoBufMessage::::new(&req, &mut pl).await; assert_eq!(protobuf.err().unwrap(), ProtoBufPayloadError::ContentType); let (req, mut pl) = TestRequest::get() - .append_header((header::CONTENT_TYPE, "application/protobuf")) - .append_header((header::CONTENT_LENGTH, "10000")) + .insert_header((header::CONTENT_TYPE, "application/protobuf")) + .insert_header((header::CONTENT_LENGTH, "10000")) .to_http_parts(); let protobuf = ProtoBufMessage::::new(&req, &mut pl) .limit(100)