class ConsumerSignalTestCase(TestCase): def setUp(self): self.consumer = Consumer('') self.html_consumer = HTMLConsumer('') @patch('monocle.consumers.registry') def test_base_consumer_signal(self, registry): pre_cb = mock_receiver() post_cb = mock_receiver() pre_consume.connect(pre_cb) post_consume.connect(post_cb) self.consumer.devour('') self.assertTrue(pre_cb.called) self.assertTrue(post_cb.called) @patch('monocle.consumers.registry') def test_html_consumer_signals_once(self, registry): pre_cb = mock_receiver() post_cb = mock_receiver() pre_consume.connect(pre_cb) post_consume.connect(post_cb) self.consumer.devour('') self.assertEqual(pre_cb.call_count, 1) self.assertEqual(post_cb.call_count, 1)
class ConsumerTestCase(TestCase): def setUp(self): self.consumer = Consumer() def test_url_regex(self): urls = self.consumer.url_regex.findall(TEXT_CONTENT) expected = ["http://foo.com", "http://bar.com", "http://baz.com/foo?a=b&x=y"] self.assertEqual(expected, urls) @patch("monocle.consumers.registry") def test_enrich(self, registry): provider = Mock() provider.get_resource.return_value = provider provider.render.return_value = "RESOURCE" registry.match.return_value = provider result = self.consumer.enrich(TEXT_CONTENT) self.assertIn("RESOURCE, RESOURCE, and (RESOURCE)", result)
def setUp(self): self.consumer = Consumer()
def setUp(self): self.consumer = Consumer('') self.html_consumer = HTMLConsumer('')