def test_check_for_stop(self): """ Test checkForStop(self) """ sfp = SpiderFootPlugin() class DatabaseStub: def scanInstanceGet(self, scanId): return [None, None, None, None, None, status] sfp.__sfdb__ = DatabaseStub() # pseudo-parameterized test # TODO could include other statuses for completeness for status, expectedReturnValue in [("RUNNING", False), ("ABORT-REQUESTED", True)]: returnValue = sfp.checkForStop() self.assertEqual(returnValue, expectedReturnValue, status)
def test_notify_listeners(self): """ Test notifyListeners(self, sfEvent) Todo: Test with source event """ sfp = SpiderFootPlugin() class DatabaseStub: def scanInstanceGet(self, scanId): return [None, None, None, None, None, None] sfp.__sfdb__ = DatabaseStub() event_type = 'ROOT' event_data = 'test data' module = 'test module' source_event = None #SpiderFootEvent("ROOT", '', '', "ROOT") evt = SpiderFootEvent(event_type, event_data, module, source_event) sfp.notifyListeners(evt) self.assertEqual('TBD', 'TBD')