def test_task_get_fallback_handler(self): class TestSpider(Spider): def zz(self, task): pass def task_bar_fallback(self, task): pass t1 = Task('foo', url='http://foo.com/', fallback_name='zz') t2 = Task('bar', url='http://foo.com/') t3 = Task(url='http://foo.com/') bot = build_spider(TestSpider, ) self.assertEqual(t1.get_fallback_handler(bot), bot.zz) self.assertEqual(t2.get_fallback_handler(bot), bot.task_bar_fallback) self.assertEqual(t3.get_fallback_handler(bot), None)
def test_task_get_fallback_handler(self): class TestSpider(Spider): def do_smth(self, task): pass def task_bar_fallback(self, task): pass task1 = Task('foo', url='http://foo.com/', fallback_name='do_smth') task2 = Task('bar', url='http://foo.com/') task3 = Task(url='http://foo.com/') bot = build_spider(TestSpider, ) self.assertEqual(task1.get_fallback_handler(bot), bot.do_smth) self.assertEqual(task2.get_fallback_handler(bot), bot.task_bar_fallback) self.assertEqual(task3.get_fallback_handler(bot), None)