berkus
/
actix-net
mirror of https://github.com/fafhrd91/actix-net
1
0
Fork 0
Code Issues Releases Wiki Activity

use async tcp stream for connecting in test

This commit is contained in:
fakeshadow 2021-04-06 02:00:03 +08:00
parent 0522f50f64
commit 3faa8717e1
1 changed files with 31 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
actix-server/tests/test_server.rs
View File

@ -265,14 +265,15 @@ async fn test_max_concurrent_connections() {
let _ = h.join().unwrap(); let _ = h.join().unwrap();
} }
#[test] #[actix_rt::test]
fn test_service_restart() { async fn test_service_restart() {
use std::net::Shutdown;
use std::task::{Context, Poll}; use std::task::{Context, Poll};
use std::time::Duration;
use actix_rt::net::TcpStream; use actix_rt::{net::TcpStream, time::sleep};
use actix_service::{fn_factory, Service}; use actix_service::{fn_factory, Service};
use futures_core::future::LocalBoxFuture; use futures_core::future::LocalBoxFuture;
use tokio::io::AsyncWriteExt;
struct TestService(Arc<AtomicUsize>); struct TestService(Arc<AtomicUsize>);
@ -347,16 +348,23 @@ fn test_service_restart() {
}); });
let (server, sys) = rx.recv().unwrap(); let (server, sys) = rx.recv().unwrap();
thread::sleep(time::Duration::from_millis(500));
for _ in 0..5 { for _ in 0..5 {
let conn = net::TcpStream::connect(addr1).unwrap(); TcpStream::connect(addr1)
conn.shutdown(Shutdown::Both).unwrap(); .await
let conn = net::TcpStream::connect(addr2).unwrap(); .unwrap()
conn.shutdown(Shutdown::Both).unwrap(); .shutdown()
.await
.unwrap();
TcpStream::connect(addr2)
.await
.unwrap()
.shutdown()
.await
.unwrap();
} }
thread::sleep(time::Duration::from_secs(1)); sleep(Duration::from_secs(3)).await;
assert!(num_clone.load(Ordering::SeqCst) > 5); assert!(num_clone.load(Ordering::SeqCst) > 5);
assert!(num2_clone.load(Ordering::SeqCst) > 5); assert!(num2_clone.load(Ordering::SeqCst) > 5);
@ -405,16 +413,23 @@ fn test_service_restart() {
}); });
let (server, sys) = rx.recv().unwrap(); let (server, sys) = rx.recv().unwrap();
thread::sleep(time::Duration::from_millis(500));
for _ in 0..5 { for _ in 0..5 {
let conn = net::TcpStream::connect(addr1).unwrap(); TcpStream::connect(addr1)
conn.shutdown(Shutdown::Both).unwrap(); .await
let conn = net::TcpStream::connect(addr2).unwrap(); .unwrap()
conn.shutdown(Shutdown::Both).unwrap(); .shutdown()
.await
.unwrap();
TcpStream::connect(addr2)
.await
.unwrap()
.shutdown()
.await
.unwrap();
} }
thread::sleep(time::Duration::from_secs(1)); sleep(Duration::from_secs(3)).await;
assert!(num_clone.load(Ordering::SeqCst) > 5); assert!(num_clone.load(Ordering::SeqCst) > 5);
assert!(num2_clone.load(Ordering::SeqCst) > 5); assert!(num2_clone.load(Ordering::SeqCst) > 5);