def test_url_observer(self):
observer = Mock()
kb.add_url_observer(observer)
url = URL('http://w3af.org/')
kb.add_url(url)
self.assertEqual(observer.call_count, 1)
self.assertEqual(observer.call_args, call(url,))
self.assertIs(observer.call_args[0][0], url)
def test_url_observer_multiple(self):
observer_1 = Mock()
observer_2 = Mock()
kb.add_url_observer(observer_1)
kb.add_url_observer(observer_2)
url = URL('http://w3af.org/')
kb.add_url(url)
self.assertEqual(observer_1.call_count, 1)
self.assertEqual(observer_2.call_count, 1)
