f moar test

Author: TheBlueMatt

lightning/src/util/wakers.rs

@@ -575,4 +575,28 @@ mod tests {
 		assert!(woken.load(Ordering::SeqCst));
 		assert_eq!(Pin::new(&mut future).poll(&mut Context::from_waker(&waker)), Poll::Ready(()));
 	}
+
+	#[test]
+	fn test_multi_future_sleep() {
+		// Tests the `Sleeper` with multiple futures.
+		let notifier_a = Notifier::new();
+		let notifier_b = Notifier::new();
+
+		// Set both notifiers as woken without sleeping yet.
+		notifier_a.notify();
+		notifier_b.notify();
+		Sleeper::from_two_futures(notifier_a.get_future(), notifier_b.get_future()).wait();
+
+		// One future has woken us up, but the other should still have a pending notification.
+		Sleeper::from_two_futures(notifier_a.get_future(), notifier_b.get_future()).wait();
+
+		// However once we've slept twice, we should no longer have any pending notifications
+		#[cfg(feature = "std")]
+		assert!(!Sleeper::from_two_futures(notifier_a.get_future(), notifier_b.get_future())
+			.wait_timeout(Duration::from_millis(10)));
+
+		// Test ordering somewhat more.
+		notifier_a.notify();
+		Sleeper::from_two_futures(notifier_a.get_future(), notifier_b.get_future()).wait();
+	}
 }