add more delay

This commit is contained in:
fakeshadow 2021-04-12 06:07:45 +08:00
parent f13d76c67d
commit 18615d95f5
1 changed files with 9 additions and 2 deletions

View File

@ -438,7 +438,7 @@ async fn test_service_restart() {
}
#[actix_rt::test]
async fn test_worker_restart() {
async fn worker_restart() {
use actix_service::{Service, ServiceFactory};
use futures_core::future::LocalBoxFuture;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
@ -540,7 +540,8 @@ async fn test_worker_restart() {
assert_eq!("1", id);
stream.shutdown().await.unwrap();
sleep(Duration::from_secs(5)).await;
// TODO: Remove sleep if it can pass CI.
sleep(Duration::from_secs(3)).await;
// worker 2 restarting and work goes to worker 1.
let mut stream = TcpStream::connect(addr).await.unwrap();
@ -550,6 +551,9 @@ async fn test_worker_restart() {
assert_eq!("1", id);
stream.shutdown().await.unwrap();
// TODO: Remove sleep if it can pass CI.
sleep(Duration::from_secs(3)).await;
// worker 2 restarted but worker 1 was still the next to accept connection.
let mut stream = TcpStream::connect(addr).await.unwrap();
let n = stream.read(&mut buf).await.unwrap();
@ -557,6 +561,9 @@ async fn test_worker_restart() {
assert_eq!("1", id);
stream.shutdown().await.unwrap();
// TODO: Remove sleep if it can pass CI.
sleep(Duration::from_secs(3)).await;
// worker 2 accept connection again but it's id is 3.
let mut stream = TcpStream::connect(addr).await.unwrap();
let n = stream.read(&mut buf).await.unwrap();