From 20cff6fc0b61f09432b576f8065985ccad1f853b Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Mon, 17 Sep 2018 16:50:35 -0700
Subject: [PATCH] update apply tests

---
 src/service/apply.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/service/apply.rs b/src/service/apply.rs
index 8a9ec1ed..06da34c7 100644
--- a/src/service/apply.rs
+++ b/src/service/apply.rs
@@ -173,7 +173,7 @@ mod tests {
     use futures::future::{ok, FutureResult};
     use futures::{Async, Future, Poll};
 
-    use service::{Service, ServiceExt};
+    use service::{IntoService, Service, ServiceExt};
 
     #[derive(Clone)]
     struct Srv;
@@ -194,8 +194,12 @@ mod tests {
 
     #[test]
     fn test_call() {
-        let mut srv =
-            Srv.apply(|req: &'static str, srv| srv.call(()).map(move |res| (req, res)));
+        let blank = |req| Ok(req);
+
+        let mut srv = blank.into_service().apply(Srv, |req: &'static str, srv| {
+            srv.call(()).map(move |res| (req, res))
+        });
+        assert!(srv.poll_ready().is_ok());
         let res = srv.call("srv").poll();
         assert!(res.is_ok());
         assert_eq!(res.unwrap(), Async::Ready(("srv", ())));