def test_ReturnsLinksFromPostProcessor(self):
links = set()
mock_links_post_processor = MagicMock()
mock_links_post_processor.apply_transforms_and_filters = MagicMock(return_value=links)
sut = responseProcessor.ResponseProcessor(None, mock_links_post_processor)
sut._process_markup = MagicMock()
expected = links
actual = sut.process_response(None, None, None, None)
self.assertEqual(expected, actual)
def test_InvokesPostProcessor(self):
dummy_link = "link"
dummy_processing_context = {
"current_link_url": dummy_link
}
dummy_links_from_markup = set()
mock_links_post_processor = MagicMock()
mock_links_post_processor.apply_transforms_and_filters = MagicMock()
sut = responseProcessor.ResponseProcessor(None, mock_links_post_processor)
sut._process_markup = MagicMock(return_value=dummy_links_from_markup)
sut.process_response(None, dummy_link, None, None)
mock_links_post_processor.apply_transforms_and_filters.assert_called_with(
dummy_links_from_markup, dummy_processing_context)
