Ausgabe der neuen DB Einträge

venv/lib/python3.9/site-packages/queuelib/tests/test_rrqueue.py

@@ -0,0 +1,144 @@
+import os
+from queuelib.rrqueue import RoundRobinQueue
+from queuelib.queue import (
+    FifoMemoryQueue, LifoMemoryQueue, FifoDiskQueue, LifoDiskQueue,
+    FifoSQLiteQueue, LifoSQLiteQueue,
+)
+from queuelib.tests import (QueuelibTestCase, track_closed)
+
+
+# hack to prevent py.test from discovering base test class
+class base:
+    class RRQueueTestBase(QueuelibTestCase):
+
+        def setUp(self):
+            QueuelibTestCase.setUp(self)
+            self.q = RoundRobinQueue(self.qfactory)
+
+        def qfactory(self, key):
+            raise NotImplementedError
+
+        def test_len_nonzero(self):
+            assert not self.q
+            self.assertEqual(len(self.q), 0)
+            self.q.push(b'a', '3')
+            assert self.q
+            self.q.push(b'b', '1')
+            self.q.push(b'c', '2')
+            self.q.push(b'd', '1')
+            self.assertEqual(len(self.q), 4)
+            self.q.pop()
+            self.q.pop()
+            self.q.pop()
+            self.q.pop()
+            assert not self.q
+            self.assertEqual(len(self.q), 0)
+
+        def test_close(self):
+            self.q.push(b'a', '3')
+            self.q.push(b'b', '1')
+            self.q.push(b'c', '2')
+            self.q.push(b'd', '1')
+            iqueues = self.q.queues.values()
+            self.assertEqual(sorted(self.q.close()), ['1', '2', '3'])
+            assert all(q.closed for q in iqueues)
+
+        def test_close_return_active(self):
+            self.q.push(b'b', '1')
+            self.q.push(b'c', '2')
+            self.q.push(b'a', '3')
+            self.q.pop()
+            self.assertEqual(sorted(self.q.close()), ['2', '3'])
+
+
+class FifoTestMixin(object):
+    def test_push_pop_key(self):
+        self.q.push(b'a', '1')
+        self.q.push(b'b', '1')
+        self.q.push(b'c', '2')
+        self.q.push(b'd', '2')
+        self.assertEqual(self.q.pop(), b'a')
+        self.assertEqual(self.q.pop(), b'c')
+        self.assertEqual(self.q.pop(), b'b')
+        self.assertEqual(self.q.pop(), b'd')
+        self.assertEqual(self.q.pop(), None)
+
+
+class LifoTestMixin(object):
+
+    def test_push_pop_key(self):
+        self.q.push(b'a', '1')
+        self.q.push(b'b', '1')
+        self.q.push(b'c', '2')
+        self.q.push(b'd', '2')
+        self.assertEqual(self.q.pop(), b'b')
+        self.assertEqual(self.q.pop(), b'd')
+        self.assertEqual(self.q.pop(), b'a')
+        self.assertEqual(self.q.pop(), b'c')
+        self.assertEqual(self.q.pop(), None)
+
+
+class FifoMemoryRRQueueTest(FifoTestMixin, base.RRQueueTestBase):
+
+    def qfactory(self, key):
+        return track_closed(FifoMemoryQueue)()
+
+
+class LifoMemoryRRQueueTest(LifoTestMixin, base.RRQueueTestBase):
+
+    def qfactory(self, key):
+        return track_closed(LifoMemoryQueue)()
+
+
+class DiskTestMixin(object):
+
+    def test_nonserializable_object_one(self):
+        self.assertRaises(TypeError, self.q.push, lambda x: x, '0')
+        self.assertEqual(self.q.close(), [])
+
+    def test_nonserializable_object_many_close(self):
+        self.q.push(b'a', '3')
+        self.q.push(b'b', '1')
+        self.assertRaises(TypeError, self.q.push, lambda x: x, '0')
+        self.q.push(b'c', '2')
+        self.assertEqual(self.q.pop(), b'a')
+        self.assertEqual(sorted(self.q.close()), ['1', '2'])
+
+    def test_nonserializable_object_many_pop(self):
+        self.q.push(b'a', '3')
+        self.q.push(b'b', '1')
+        self.assertRaises(TypeError, self.q.push, lambda x: x, '0')
+        self.q.push(b'c', '2')
+        self.assertEqual(self.q.pop(), b'a')
+        self.assertEqual(self.q.pop(), b'b')
+        self.assertEqual(self.q.pop(), b'c')
+        self.assertEqual(self.q.pop(), None)
+        self.assertEqual(self.q.close(), [])
+
+
+class FifoDiskRRQueueTest(FifoTestMixin, DiskTestMixin, base.RRQueueTestBase):
+
+    def qfactory(self, key):
+        path = os.path.join(self.qdir, str(key))
+        return track_closed(FifoDiskQueue)(path)
+
+
+class LifoDiskRRQueueTest(LifoTestMixin, DiskTestMixin, base.RRQueueTestBase):
+
+    def qfactory(self, key):
+        path = os.path.join(self.qdir, str(key))
+        return track_closed(LifoDiskQueue)(path)
+
+
+class FifoSQLiteRRQueueTest(FifoTestMixin, DiskTestMixin, base.RRQueueTestBase):
+
+    def qfactory(self, key):
+        path = os.path.join(self.qdir, str(key))
+        return track_closed(FifoSQLiteQueue)(path)
+
+
+class LifoSQLiteRRQueueTest(LifoTestMixin, DiskTestMixin, base.RRQueueTestBase):
+
+    def qfactory(self, key):
+        path = os.path.join(self.qdir, str(key))
+        return track_closed(LifoSQLiteQueue)(path)