From 43ae7bdd8992debea24a725e739729094c84abda Mon Sep 17 00:00:00 2001 From: Pablo Hoffman Date: Sun, 11 Sep 2011 08:27:05 -0300 Subject: [PATCH] added tests for SpiderState extension --- scrapy/tests/test_contrib_spiderstate.py | 27 ++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 scrapy/tests/test_contrib_spiderstate.py diff --git a/scrapy/tests/test_contrib_spiderstate.py b/scrapy/tests/test_contrib_spiderstate.py new file mode 100644 index 000000000..e340b489a --- /dev/null +++ b/scrapy/tests/test_contrib_spiderstate.py @@ -0,0 +1,27 @@ +import os +from datetime import datetime +from twisted.trial import unittest + +from scrapy.contrib.spiderstate import SpiderState +from scrapy.spider import BaseSpider + + +class SpiderStateTest(unittest.TestCase): + + def test_store_load(self): + jobdir = self.mktemp() + os.mkdir(jobdir) + spider = BaseSpider(name='default') + dt = datetime.now() + + ss = SpiderState(jobdir) + ss.spider_opened(spider) + spider.state['one'] = 1 + spider.state['dt'] = dt + ss.spider_closed(spider) + + spider2 = BaseSpider(name='default') + ss2 = SpiderState(jobdir) + ss2.spider_opened(spider2) + self.assertEqual(spider.state, {'one': 1, 'dt': dt}) + ss2.spider_closed(spider2)