f complete test

Author: TheBlueMatt

lightning/src/util/wakers.rs

@@ -447,8 +447,8 @@ mod tests {
 
 		// If we get a future and don't touch it we're definitely still notify-required.
 		notifier.get_future();
-		assert!(notifier.wait_timeout(Duration::from_secs(1)));
-		assert!(!notifier.wait_timeout(Duration::from_secs(1)));
+		assert!(notifier.wait_timeout(Duration::from_millis(1)));
+		assert!(!notifier.wait_timeout(Duration::from_millis(1)));
 
 		// Even if we poll'd once but didn't observe a `Ready`, we should be notify-required.
 		let mut future = notifier.get_future();
@@ -457,10 +457,16 @@ mod tests {
 
 		notifier.notify();
 		assert!(woken.load(Ordering::SeqCst));
-		assert!(notifier.wait_timeout(Duration::from_secs(1)));
+		assert!(notifier.wait_timeout(Duration::from_millis(1)));
 
-		notifier.notify();
+		// However, once we do poll `Ready` it should wipe the notify-required flag.
+		let mut future = notifier.get_future();
+		let (woken, waker) = create_waker();
+		assert_eq!(Pin::new(&mut future).poll(&mut Context::from_waker(&waker)), Poll::Pending);
 
 		notifier.notify();
+		assert!(woken.load(Ordering::SeqCst));
+		assert_eq!(Pin::new(&mut future).poll(&mut Context::from_waker(&waker)), Poll::Ready(()));
+		assert!(!notifier.wait_timeout(Duration::from_millis(1)));
 	}
 }