From 5586e2d46fa2c9231d46943ddb946d3fb0db8406 Mon Sep 17 00:00:00 2001
From: Veritius <veritiusgaming@gmail.com>
Date: Fri, 5 Jul 2024 23:40:02 +1000
Subject: [PATCH] Message ordering test

---
 stardust/src/messages/queue.rs | 40 ++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/stardust/src/messages/queue.rs b/stardust/src/messages/queue.rs
index bd6a35ec5..743ba1e4e 100644
--- a/stardust/src/messages/queue.rs
+++ b/stardust/src/messages/queue.rs
@@ -252,4 +252,44 @@ impl<'a> ExactSizeIterator for MessageIter<'a> {
     fn len(&self) -> usize {
         self.indexes.len()
     }
+}
+
+#[test]
+fn message_queue_ordering_test() {
+    let mut queue = MessageQueue::new();
+
+    fn map_messages(set: &'static [&'static [u8]]) -> impl Iterator<Item = Message> {
+        set.iter().map(|v| Message::from_bytes(Bytes::from_static(v)))
+    }
+
+    const MESSAGE_SET_A: &[&[u8]] = &[
+        b"Hello, world!",
+        b"It's a very nice day, isn't it?",
+        b"Yeah, I agree!",
+    ];
+
+    const MESSAGE_SET_B: &[&[u8]] = &[
+        b"Hello, world!",
+        b"It's a miserable day, isn't it?",
+        b"No, I think it's fine!",
+    ];
+
+    const MESSAGE_SET_C: &[&[u8]] = &[
+        b"Hello, world!",
+        b"It's an alright day, isn't it?",
+        b"That's a good way of putting it.",
+    ];
+
+    // Add all the messages to the channel
+    queue.push_channel(ChannelId::from(0), map_messages(MESSAGE_SET_A));
+    queue.push_channel(ChannelId::from(0), map_messages(MESSAGE_SET_B));
+    queue.push_channel(ChannelId::from(1), map_messages(MESSAGE_SET_C));
+
+    queue.iter_channel(ChannelId::from(0))
+    .zip(MESSAGE_SET_A.iter().chain(MESSAGE_SET_B))
+    .for_each(|(a, b)| assert_eq!(a.as_slice(), *b));
+
+    queue.iter_channel(ChannelId::from(1))
+    .zip(MESSAGE_SET_C)
+    .for_each(|(a, b)| assert_eq!(a.as_slice(), *b));
 }
\ No newline at end of file