def test_on_interrupt_is_not_called_without_interrupt(self):
myCrawler = DomainCrawler()
callback = CallableMock()
myCrawler.on_interrupt(callback)
self.assertEqual(callback.call_count(), 0)
myCrawler.crawl("tiggers")
self.assertEqual(callback.call_count(), 0)
def test_calls_callback_on_interrupt(self):
parser.parse.raises(KeyboardInterrupt())
my_crawler = DomainCrawler()
callback = CallableMock()
my_crawler.on_interrupt(callback)
self.assertEqual(callback.call_count(), 0)
with self.assertRaises(KeyboardInterrupt):
my_crawler.crawl("tiggers")
self.assertEqual(callback.call_count(), 1)