def test_observer_add_url(self): observer1 = Mock() url = URL('http://www.w3af.org/') kb.add_observer(observer1) kb.add_url(url) observer1.add_url.assert_called_once_with(url)
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)