Add basic testing of EventQueue

Author: tnull

src/event.rs

@@ -28,8 +28,8 @@ pub(crate) const EVENTS_PERSISTENCE_KEY: &str = "events";
 /// An event emitted by [`LdkLite`] that should be handled by the user.
 ///
 /// [`LdkLite`]: [`crate::LdkLite`]
-#[derive(Debug, Clone)]
-pub enum LdkLiteEvent {
+#[derive(Debug, Clone, PartialEq)]
+pub enum Event {
 	/// A payment we sent was successful.
 	PaymentSuccessful {
 		/// The hash of the payment.
@@ -368,6 +368,9 @@ where
 
 				if let Some(preimage) = payment_preimage {
 					self.channel_manager.claim_funds(preimage);
+					self.event_queue
+						.add_event(Event::PaymentReceived { payment_hash, amount_msat })
+						.expect("Failed to push to event queue");
 				} else {
 					log_error!(
 						self.logger,
@@ -406,9 +409,6 @@ where
 						});
 					}
 				}
-				self.event_queue
-					.add_event(Event::PaymentReceived { payment_hash, amount_msat })
-					.expect("Failed to push to event queue");
 			}
 			LdkEvent::PaymentSent { payment_preimage, payment_hash, fee_paid_msat, .. } => {
 				let mut payments = self.outbound_payments.lock().unwrap();

src/test_utils.rs

@@ -0,0 +1,26 @@
+use lightning::util::persist::KVStorePersister;
+use lightning::util::ser::Writeable;
+
+use std::sync::atomic::{AtomicBool, Ordering};
+
+pub(crate) struct TestPersister {
+	pending_persist: AtomicBool,
+}
+
+impl TestPersister {
+	pub fn new() -> Self {
+		let pending_persist = AtomicBool::new(false);
+		Self { pending_persist }
+	}
+
+	pub fn get_and_clear_pending_persist(&self) -> bool {
+		self.pending_persist.swap(false, Ordering::SeqCst)
+	}
+}
+
+impl KVStorePersister for TestPersister {
+	fn persist<W: Writeable>(&self, _key: &str, _object: &W) -> std::io::Result<()> {
+		self.pending_persist.store(true, Ordering::SeqCst);
+		Ok(())
+	}
+}