Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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)
Exemplo n.º 3
0
 def setUp(self):
     self.consumer = Consumer()
Exemplo n.º 4
0
 def setUp(self):
     self.consumer = Consumer('')
     self.html_consumer = HTMLConsumer('')