mirrors/ntex
1
0
Fork 0
mirror of https://github.com/ntex-rs/ntex.git synced 2025-04-04 05:17:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

Improve tests

Browse source
This commit is contained in:
Nikolay Kim 2025-03-28 09:34:52 +01:00
parent 30928d019c
commit d9c14ef9da
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
ntex/tests/http_openssl.rs
View file

@ -438,15 +438,19 @@ impl Drop for SetOnDrop {
async fn test_h2_client_drop() -> io::Result<()> { async fn test_h2_client_drop() -> io::Result<()> {
let count = Arc::new(AtomicUsize::new(0)); let count = Arc::new(AtomicUsize::new(0));
let count2 = count.clone(); let count2 = count.clone();
let (tx, rx) = ::oneshot::channel(); let (tx, _rx) = ::oneshot::channel();
let tx = Arc::new(Mutex::new(Some(tx))); let tx = Arc::new(Mutex::new(Some(tx)));
let (tx2, rx2) = ::oneshot::channel();
let tx2 = Arc::new(Mutex::new(Some(tx2)));
let srv = test_server(move || { let srv = test_server(move || {
let tx = tx.clone(); let tx = tx.clone();
let tx2 = tx2.clone();
let count = count2.clone(); let count = count2.clone();
HttpService::build() HttpService::build()
.h2(move |req: Request| { .h2(move |req: Request| {
let tx = tx.clone(); let tx = tx.clone();
let _ = tx2.lock().unwrap().take().unwrap().send(());
let count = count.clone(); let count = count.clone();
async move { async move {
let _st = SetOnDrop(count, tx); let _st = SetOnDrop(count, tx);
@ -460,9 +464,13 @@ async fn test_h2_client_drop() -> io::Result<()> {
.map_err(|_| ()) .map_err(|_| ())
}); });
let result = timeout(Millis(250), srv.srequest(Method::GET, "/").send()).await; let task = rt::spawn(timeout(Millis(150), srv.srequest(Method::GET, "/").send()));
let _ = rx2.await;
let result = task.await.unwrap();
assert!(result.is_err()); assert!(result.is_err());
let _ = rx.await; //let _ = _rx.await;
sleep(Millis(150)).await;
assert_eq!(count.load(Ordering::Relaxed), 1); assert_eq!(count.load(Ordering::Relaxed), 1);
Ok(()) Ok(())
} }