From fd689be6d85c1176760baabbd9738b47535eceef Mon Sep 17 00:00:00 2001
From: fakeshadow <24548779@qq.com>
Date: Sat, 6 Feb 2021 09:33:24 -0800
Subject: [PATCH] add test for manual drop

---
 actix-http/src/client/connection.rs | 44 +++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/actix-http/src/client/connection.rs b/actix-http/src/client/connection.rs
index 6d891b101..971a76da6 100644
--- a/actix-http/src/client/connection.rs
+++ b/actix-http/src/client/connection.rs
@@ -355,3 +355,47 @@ where
         }
     }
 }
+
+#[cfg(test)]
+mod test {
+    use std::sync::{
+        atomic::{AtomicUsize, Ordering},
+        Arc,
+    };
+
+    use super::*;
+
+    #[actix_rt::test]
+    async fn test_manual_drop() {
+        let var = Arc::new(AtomicUsize::new(0));
+
+        let var_1 = var.clone();
+        let fut = async move {
+            actix_rt::time::sleep(std::time::Duration::from_millis(500)).await;
+            var_1.fetch_add(1, Ordering::SeqCst);
+        };
+        let (fut, handle) = WakeupOnDrop::new(fut);
+
+        actix_rt::spawn(fut);
+
+        actix_rt::time::sleep(std::time::Duration::from_millis(100)).await;
+
+        handle.wake();
+        drop(handle);
+
+        actix_rt::time::sleep(std::time::Duration::from_millis(500)).await;
+        assert_eq!(var.load(Ordering::SeqCst), 0);
+
+        let var_1 = var.clone();
+        let fut = async move {
+            actix_rt::time::sleep(std::time::Duration::from_millis(500)).await;
+            var_1.fetch_add(1, Ordering::SeqCst);
+        };
+
+        let (fut, _handle) = WakeupOnDrop::new(fut);
+        actix_rt::spawn(fut);
+
+        actix_rt::time::sleep(std::time::Duration::from_millis(600)).await;
+        assert_eq!(var.load(Ordering::SeqCst), 1);
+    }
+}