def test_Plugin_creates_adecuate_objects(self, monkeypatch): self.plugin = BurpPlugin() factory.register(Host) factory.register(Service) factory.register(Vuln) factory.register(VulnWeb) factory.register(Note) factory.register(Credential) pending_actions = Queue() self.plugin.set_actions_queue(pending_actions) monkeypatch.setattr(ModelBase, 'getID', lambda _: 1) self.plugin.processReport(self.cd + '/burp_xml') actions = defaultdict(list) while not pending_actions.empty(): action = self.plugin._pending_actions.get(block=True) actions[action[0]].append(action[1]) assert actions[2000][0].name == "200.20.20.201" assert actions.keys() == [2000, 20008, 2038] assert len(actions[20008]) == 14 assert len(actions[2038]) == 14 assert all( 'http' == name for name in map(lambda service: service.name, actions[20008])) assert all([80] == ports for ports in map( lambda service: service.getPorts(), actions[20008])) assert all('tcp' == protocol for protocol in map( lambda service: service.protocol, actions[20008])) assert all( 'open' for status in map(lambda service: service.status, actions[20008]))
def setUp(self): self.plugin = BurpPlugin() factory.register(Host) factory.register(Interface) factory.register(Service) factory.register(Vuln) factory.register(VulnWeb) factory.register(Note) factory.register(Credential)
def setUp(self): self.plugin = BurpPlugin() factory.register(Host) factory.register(Interface) factory.register(Service) factory.register(ModelObjectVuln) factory.register(ModelObjectVulnWeb) factory.register(ModelObjectNote) factory.register(ModelObjectCred)