Fix flaky test

Author: taiki-e

Cargo.toml

@@ -9,8 +9,6 @@ edition = "2018"
 rust-version = "1.39"
 license = "Apache-2.0 OR MIT"
 repository = "https://github.com/smol-rs/async-task"
-homepage = "https://github.com/smol-rs/async-task"
-documentation = "https://docs.rs/async-task"
 description = "Task abstraction for building executors"
 keywords = ["futures", "task", "executor", "spawn"]
 categories = ["asynchronous", "concurrency", "no-std"]
@@ -21,8 +19,9 @@ default = ["std"]
 std = []
 
 [dev-dependencies]
-atomic-waker = "1.0.0"
-easy-parallel = "3.1.0"
+atomic-waker = "1"
+easy-parallel = "3"
+flaky_test = "0.1"
 flume = { version = "0.10", default-features = false }
-once_cell = "1.4.1"
-smol = "1.0.1"
+once_cell = "1"
+smol = "1"

tests/waker_panic.rs

@@ -238,10 +238,15 @@ fn wake_and_cancel_during_run() {
         .run();
 }
 
-#[test]
+#[flaky_test::flaky_test]
 fn cancel_and_wake_during_run() {
     future!(f, get_waker, POLL, DROP_F);
     schedule!(s, chan, SCHEDULE, DROP_S);
+    POLL.store(0, Ordering::SeqCst);
+    DROP_F.store(0, Ordering::SeqCst);
+    SCHEDULE.store(0, Ordering::SeqCst);
+    DROP_S.store(0, Ordering::SeqCst);
+
     let (runnable, task) = async_task::spawn(f, s);
 
     runnable.run();