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)
