2 files changed, 11 insertions, 28 deletions
diff --git a/tokio-test/tests/block_on.rs b/tokio-test/tests/block_on.rs
index 9b4959e9..6d5f481a 100644
--- a/tokio-test/tests/block_on.rs
+++ b/tokio-test/tests/block_on.rs
@@ -1,8 +1,9 @@
 #![warn(rust_2018_idioms)]
 
-use std::time::{Duration, Instant};
+use tokio::timer::delay;
 use tokio_test::block_on;
-use tokio_timer::delay;
+
+use std::time::{Duration, Instant};
 
 #[test]
 fn async_block() {
diff --git a/tokio-test/tests/clock.rs b/tokio-test/tests/clock.rs
index 344c599b..abb61e23 100644
--- a/tokio-test/tests/clock.rs
+++ b/tokio-test/tests/clock.rs
@@ -1,12 +1,11 @@
-#![cfg(feature = "broken")]
 #![warn(rust_2018_idioms)]
 
-use futures::Future;
-use std::time::{Duration, Instant};
+use tokio::timer::delay;
 use tokio_test::clock::MockClock;
-use tokio_test::task::MockTask;
-use tokio_test::{assert_not_ready, assert_ready};
-use tokio_timer::delay;
+use tokio_test::task;
+use tokio_test::{assert_pending, assert_ready};
+
+use std::time::{Duration, Instant};
 
 #[test]
 fn clock() {
@@ -14,30 +13,13 @@ fn clock() {
 
     mock.enter(|handle| {
         let deadline = Instant::now() + Duration::from_secs(1);
-        let mut delay = delay(deadline);
+        let mut delay = task::spawn(delay(deadline));
 
-        assert_not_ready!(delay.poll());
+        assert_pending!(delay.poll());
 
         handle.advance(Duration::from_secs(2));
 
-        assert_ready!(delay.poll());
-    });
-}
-
-#[test]
-fn notify() {
-    let deadline = Instant::now() + Duration::from_secs(1);
-    let mut mock = MockClock::new();
-    let mut task = MockTask::new();
-
-    mock.enter(|handle| {
-        let mut delay = delay(deadline);
-
-        task.enter(|| assert_not_ready!(delay.poll()));
-
-        handle.advance(Duration::from_secs(1));
-
-        assert!(task.is_notified());
+        assert!(delay.is_woken());
         assert_ready!(delay.poll());
     });
 }