Include connect info with app

Author: banool

src/lib.rs

@@ -35,7 +35,7 @@
 //!     runtime.block_on(async_block);
 //! }
 
-use axum::{body::HttpBody, BoxError};
+use axum::{body::HttpBody, BoxError, ServiceExt};
 use bytes::Bytes;
 use http::{
     header::{HeaderName, HeaderValue},
@@ -44,7 +44,6 @@ use http::{
 use hyper::{Body, Server};
 use std::convert::TryFrom;
 use std::net::{SocketAddr, TcpListener};
-use tower::make::Shared;
 use tower_service::Service;
 
 pub struct TestClient {
@@ -68,7 +67,7 @@ impl TestClient {
         println!("Listening on {}", addr);
 
         tokio::spawn(async move {
-            let server = Server::from_tcp(listener).unwrap().serve(Shared::new(svc));
+            let server = Server::from_tcp(listener).unwrap().serve(svc.into_make_service_with_connect_info::<SocketAddr>());
             server.await.expect("server error");
         });