Test cases for multiqueue

src/tests/test_multiqueue.py

@@ -0,0 +1,65 @@
+"""Test cases for multiqueue"""
+
+import unittest
+from pybitmessage.multiqueue import MultiQueue
+
+
+class TestMultiQueue(unittest.TestCase):
+    """Test cases for multiqueue"""
+
+    def test_queue_creation(self):
+        """Check if the queueCount matches the specified value"""
+        mqsize = 3
+        multiqueue = MultiQueue(count=mqsize)
+        self.assertEqual(multiqueue.queueCount, mqsize)
+
+    def test_empty_queue(self):
+        """Check for empty queue"""
+        multiqueue = MultiQueue(count=5)
+        self.assertEqual(multiqueue.totalSize(), 0)
+
+    def test_put_get_count(self):
+        """check if put & get count is equal"""
+        multiqueue = MultiQueue(count=5)
+        put_count = 6
+        for i in range(put_count):
+            multiqueue.put(i)
+
+        get_count = 0
+        while multiqueue.totalSize() != 0:
+            if multiqueue.qsize() > 0:
+                multiqueue.get()
+                get_count += 1
+            multiqueue.iterate()
+
+        self.assertEqual(get_count, put_count)
+
+    def test_put_and_get(self):
+        """Testing Put and Get"""
+        item = 400
+        multiqueue = MultiQueue(count=3)
+        multiqueue.put(item)
+        result = None
+        for _ in multiqueue.queues:
+            if multiqueue.qsize() > 0:
+                result = multiqueue.get()
+                break
+            multiqueue.iterate()
+        self.assertEqual(result, item)
+
+    def test_iteration(self):
+        """Check if the iteration wraps around correctly"""
+        mqsize = 3
+        iteroffset = 1
+        multiqueue = MultiQueue(count=mqsize)
+        for _ in range(mqsize + iteroffset):
+            multiqueue.iterate()
+        self.assertEqual(multiqueue.iter, iteroffset)
+
+    def test_total_size(self):
+        """Check if the total size matches the expected value"""
+        multiqueue = MultiQueue(count=3)
+        put_count = 5
+        for i in range(put_count):
+            multiqueue.put(i)
+        self.assertEqual(multiqueue.totalSize(), put_count)