Add support for changes to conduit::Body

jtgeibel · jtgeibel · commit 7f2cf53af042 · 2020-03-01T15:05:31.000-05:00
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,12 +1,15 @@
 [package]
 
 name = "conduit-middleware"
-version = "0.9.0-alpha.1"
+version = "0.9.0-alpha.2"
 authors = ["wycats@gmail.com",
            "Alex Crichton <alex@alexcrichton.com>"]
 description = "HTTP Middleware interface using the conduit interface"
 repository = "https://github.com/conduit-rust/conduit-middleware"
 license = "MIT"
 
 [dependencies]
-conduit = "0.9.0-alpha.1"
+conduit = { git = "https://github.com/jtgeibel/conduit", rev = "0deff5b" } # "0.9.0-alpha.2"
+
+[dev-dependencies]
+conduit-test = { git = "https://github.com/jtgeibel/conduit-test", rev = "b71595f" } # "0.9.0-alpha.2"
diff --git a/src/lib.rs b/src/lib.rs
@@ -87,16 +87,20 @@ fn run_afters(
 
 #[cfg(test)]
 mod tests {
+    extern crate conduit_test;
+
     use {AfterResult, AroundMiddleware, BeforeResult, Middleware, MiddlewareBuilder};
 
     use std::any::Any;
     use std::io;
     use std::io::prelude::*;
     use std::net::SocketAddr;
 
+    use self::conduit_test::ResponseExt;
+
     use conduit::{
-        box_error, static_to_body, vec_to_body, Body, Extensions, Handler, HeaderMap, Host, Method,
-        RequestExt, Response, Scheme, StatusCode, TypeMap, Version,
+        box_error, Body, Extensions, Handler, HeaderMap, Host, Method, RequestExt, Response,
+        Scheme, StatusCode, TypeMap, Version,
     };
 
     struct RequestSentinel {
@@ -174,7 +178,7 @@ mod tests {
                 let e = e.to_string().into_bytes();
                 Response::builder()
                     .status(StatusCode::INTERNAL_SERVER_ERROR)
-                    .body(vec_to_body(e))
+                    .body(Body::from_vec(e))
                     .map_err(box_error)
             })
         }
@@ -192,9 +196,7 @@ mod tests {
 
     impl Middleware for NotReached {
         fn after(&self, _: &mut dyn RequestExt, _: AfterResult) -> AfterResult {
-            Response::builder()
-                .body(static_to_body(b""))
-                .map_err(box_error)
+            Response::builder().body(Body::empty()).map_err(box_error)
         }
     }
 
@@ -229,7 +231,7 @@ mod tests {
 
     fn response(string: String) -> Response<Body> {
         Response::builder()
-            .body(vec_to_body(string.into_bytes()))
+            .body(Body::from_vec(string.into_bytes()))
             .unwrap()
     }
 
@@ -255,11 +257,9 @@ mod tests {
         builder.add(MyMiddleware);
 
         let mut req = RequestSentinel::new(Method::GET, "/");
-        let mut res = builder.call(&mut req).ok().expect("No response");
+        let res = builder.call(&mut req).ok().expect("No response");
 
-        let mut s = Vec::new();
-        res.body_mut().write_body(&mut s).unwrap();
-        assert_eq!(s, b"hello");
+        assert_eq!(res.into_cow(), "hello".as_bytes());
     }
 
     #[test]
@@ -282,12 +282,9 @@ mod tests {
         builder.add(ErrorRecovery);
 
         let mut req = RequestSentinel::new(Method::GET, "/");
-        let mut res = builder.call(&mut req).ok().expect("Error not handled");
+        let res = builder.call(&mut req).ok().expect("Error not handled");
 
-        assert_eq!(res.status(), StatusCode::INTERNAL_SERVER_ERROR);
-        let mut s = Vec::new();
-        res.body_mut().write_body(&mut s).unwrap();
-        assert_eq!(s, b"Error in handler");
+        assert_eq!(res.into_cow(), "Error in handler".as_bytes());
     }
 
     #[test]
@@ -297,10 +294,8 @@ mod tests {
         builder.around(MyAroundMiddleware::new());
 
         let mut req = RequestSentinel::new(Method::GET, "/");
-        let mut res = builder.call(&mut req).ok().expect("No response");
+        let res = builder.call(&mut req).ok().expect("No response");
 
-        let mut s = Vec::new();
-        res.body_mut().write_body(&mut s).unwrap();
-        assert_eq!(s, b"hello hello");
+        assert_eq!(res.into_cow(), "hello hello".as_bytes());
     }
 }
