def test_run(self, mock_get):
p = Producer()
q = Queue()
websites = ['http://ajankovic.com/blog/', 'http://test.com']
extracted = [BLOG_RESPONSE, TEST_RESPONSE]
p.run(websites, q)
for item in drain(q):
if item is None:
break
self.assertIn(item[1], extracted)
def test_run(self, mock_get):
c = Consumer()
markupq = Queue()
markupq.put(('http://ajankovic.com/blog/', BLOG_RESPONSE))
markupq.put(('http://test.com', TEST_RESPONSE))
markupq.put(None)
outq = Queue()
c.run(markupq, outq)
for item in drain(outq):
if item is None:
break
self.assertIn(item, EXTRACTED_URLS)
