Exemplos de Task.get_fallback_handler em Python

Exemplo n.º 1

    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)

Exemplo n.º 2

Arquivo: spider_task.py Projeto: target-v/grab

    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)

Exemplo n.º 3

Arquivo: spider_task.py Projeto: yegorov-p/grab

    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)