actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-23 15:51:06 +01:00
Code Issues Releases Wiki Activity

use insert for protobuf content type header

Browse Source
This commit is contained in:
Rob Ede 2021-03-22 05:30:16 +00:00
parent c8f1d9671c
commit 15d72b1694
No known key found for this signature in database
GPG Key ID: 97C636207D3EF933
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
actix-protobuf/src/lib.rs
View File

@ -250,7 +250,7 @@ pub trait ProtoBufResponseBuilder {
impl ProtoBufResponseBuilder for HttpResponseBuilder {
fn protobuf<T: Message>(&mut self, value: T) -> Result<HttpResponse, Error> {
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::<MyObject>::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::<MyObject>::new(&req, &mut pl)
.limit(100)