144 lines
4.4 KiB
Python
144 lines
4.4 KiB
Python
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)
|