Skip to content

Commit d0e3c64

Browse files
committed
fix test start server listen both socket address
1 parent 5af3196 commit d0e3c64

File tree

2 files changed

+16
-5
lines changed

2 files changed

+16
-5
lines changed

lib/src/http/stream.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,8 +121,14 @@ impl AsyncWrite for Stream {
121121
async fn test_connect() {
122122
crate::tests::start_server(30000).await;
123123
let stream = Stream::connect(&Url::from("http://127.0.0.1:30000/get"), Duration::from_secs(2)).await;
124+
if let Err(error) = stream.as_ref() {
125+
println!("{:?}", error);
126+
}
124127
assert!(stream.is_ok());
125128
let stream = Stream::connect(&Url::from("http://localhost:30000/get"), Duration::from_secs(2)).await;
129+
if let Err(error) = stream.as_ref() {
130+
println!("{:?}", error);
131+
}
126132
assert!(stream.is_ok());
127133
let stream = Stream::connect(&Url::from("http://localhost:88/get"), Duration::from_secs(2)).await;
128134
assert!(stream.is_err());

lib/src/lib.rs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ pub mod tests {
3737
use serde_json::json;
3838
use serde_json::Value;
3939
use std::collections::HashMap;
40+
use std::net::ToSocketAddrs;
41+
use tokio::net::TcpListener;
4042

4143
pub async fn start_server(port: u16) {
4244
let router = Router::new()
@@ -45,11 +47,14 @@ pub mod tests {
4547
.route("/json", post(handle_json))
4648
.route("/form", post(handle_form))
4749
.route("/multipart", post(handle_multipart));
48-
49-
let listener = tokio::net::TcpListener::bind(("127.0.0.1", port)).await.unwrap();
50-
tokio::spawn(async move {
51-
axum::serve(listener, router).await.unwrap();
52-
});
50+
let addrs = ("localhost", port).to_socket_addrs().unwrap();
51+
for addr in addrs {
52+
let listener = TcpListener::bind(addr).await.unwrap();
53+
let router = router.to_owned();
54+
tokio::spawn(async move {
55+
axum::serve(listener, router).await.unwrap();
56+
});
57+
}
5358
}
5459

5560
async fn handle_get(headers: HeaderMap, Query(params): Query<HashMap<String, String>>) -> Json<Value> {

0 commit comments

Comments
 (0)