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)