From bb67cfd955c0b53640a938bb71f4afeb210b6c47 Mon Sep 17 00:00:00 2001 From: Pablo Hoffman Date: Fri, 5 Aug 2011 20:32:22 -0300 Subject: [PATCH] added MarshalDiskQueue unittests --- scrapy/tests/test_squeue.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 scrapy/tests/test_squeue.py diff --git a/scrapy/tests/test_squeue.py b/scrapy/tests/test_squeue.py new file mode 100644 index 000000000..818888d65 --- /dev/null +++ b/scrapy/tests/test_squeue.py @@ -0,0 +1,31 @@ +from scrapy.tests import test_utils_queue as t +from scrapy.squeue import MarshalDiskQueue + + +class MarshalDiskQueueTest(t.DiskQueueTest): + + chunksize = 100000 + + def queue(self): + return MarshalDiskQueue(self.qdir, chunksize=self.chunksize) + + def test_serialize(self): + q = self.queue() + q.push('a') + q.push(123) + q.push({'a': 'dict'}) + self.assertEqual(q.pop(), 'a') + self.assertEqual(q.pop(), 123) + self.assertEqual(q.pop(), {'a': 'dict'}) + +class ChunkSize1MarshalDiskQueueTest(MarshalDiskQueueTest): + chunksize = 1 + +class ChunkSize2MarshalDiskQueueTest(MarshalDiskQueueTest): + chunksize = 2 + +class ChunkSize3MarshalDiskQueueTest(MarshalDiskQueueTest): + chunksize = 3 + +class ChunkSize4MarshalDiskQueueTest(MarshalDiskQueueTest): + chunksize = 4