Move TestSameUrlSpider to test method

This commit is contained in:
Stas Glubokiy 2018-08-15 20:36:10 +03:00
parent 4de493efdd
commit 2cb4decb6a
1 changed files with 22 additions and 23 deletions

View File

@ -103,29 +103,6 @@ class TestSpider(Spider):
pass
class TestSameUrlSpider(Spider):
name = 'test_same_url'
def __init__(self, *args, **kwargs):
super(TestSameUrlSpider, self).__init__(*args, **kwargs)
self.visited = 0
def parse_first(self, response):
"""first callback
@url http://scrapy.org
"""
self.visited += 1
return TestItem()
def parse_second(self, response):
"""second callback
@url http://scrapy.org
"""
self.visited += 1
return TestItem()
class ContractsManagerTest(unittest.TestCase):
contracts = [UrlContract, ReturnsContract, ScrapesContract]
@ -228,6 +205,28 @@ class ContractsManagerTest(unittest.TestCase):
@defer.inlineCallbacks
def test_same_url(self):
class TestSameUrlSpider(Spider):
name = 'test_same_url'
def __init__(self, *args, **kwargs):
super(TestSameUrlSpider, self).__init__(*args, **kwargs)
self.visited = 0
def parse_first(self, response):
"""first callback
@url http://scrapy.org
"""
self.visited += 1
return TestItem()
def parse_second(self, response):
"""second callback
@url http://scrapy.org
"""
self.visited += 1
return TestItem()
TestSameUrlSpider.start_requests = lambda s: self.conman.from_spider(s, self.results)
crawler = CrawlerRunner().create_crawler(TestSameUrlSpider)