def testDelVulnFromService(self): """ This test case creates a host within the Model Controller context adds an interface to it then adds service then a Vuln, then removes the Vuln""" # When host = test_utils.create_host(self) interface = test_utils.create_interface(self, host) service = test_utils.create_service(self, host, interface) vuln = ModelObjectVuln(name='VulnTest', desc='TestDescription', severity='high') self.model_controller.addVulnToServiceSYNC(host.getID(), service.getID(), vuln) added_host = self.model_controller.getHost(host.getName()) added_interface = added_host.getInterface(interface.getID()) added_service = added_interface.getService(service.getID()) vulns = added_service.getVulns() self.assertIn(vuln, vulns, 'Vuln not added') # Then self.model_controller.delVulnFromServiceSYNC(host.getID(), service.getID(), vuln.getID()) added_host = self.model_controller.getHost(host.getName()) added_interface = added_host.getInterface(interface.getID()) added_service = added_interface.getService(service.getID()) vulns = added_service.getVulns() self.assertNotIn(vuln, vulns, 'Vuln not removed')
def testAddNoteToVulnGetsMapperDispatchSave(self): vuln = ModelObjectVuln('a vuln') note = ModelObjectNote("a_note") mappersManager = self.createMapperMock() objectMapper = mock() when(mappersManager).getMapper(note).thenReturn(objectMapper) when(objectMapper).saveObject(note).thenReturn(True) model_controller = controller.ModelController(mock(), mappersManager) model_controller.addNoteToServiceSYNC(None, vuln.getID(), note) verify(mappersManager).getMapper(note) verify(objectMapper).saveObject(note)
def testAddNoteToVulnGetsMapperDispatchSave(self): vuln = ModelObjectVuln('a vuln') note = ModelObjectNote("a_note") mappersManager = self.createMapperMock() objectMapper = mock() when(mappersManager).getMapper( note.class_signature).thenReturn(objectMapper) when(objectMapper).save(note).thenReturn(True) model_controller = controller.ModelController(mock(), mappersManager) model_controller.addNoteToServiceSYNC(None, vuln.getID(), note) verify(mappersManager).getMapper(note.class_signature) verify(objectMapper).save(note)
def testDeleteVulnFromHost(self): """ Creates a Host adds a Vuln then removes """ host1 = test_utils.create_host(self, "coquito") vuln = ModelObjectVuln(name='VulnTest', desc='TestDescription', severity='high') self.model_controller.addVulnToHostSYNC(host1.getID(), vuln) hosts_ids = [h.getID() for h in self.model_controller.getAllHosts()] self.assertIn(host1.getID(), hosts_ids, "Host not in controller") self.model_controller.delVulnFromHostSYNC(host1.getID(), vuln.getID()) added_host = self.model_controller.getHost(host1.getName()) self.assertNotIn(vuln, added_host.getVulns(), 'Vuln not removed')
def testDelVulnFromServiceASYNC(self): service = Service('coco') vuln = ModelObjectVuln("int_mock0") service.addChild(vuln.getID(), vuln) self.genericDelTest(service, vuln, controller.ModelController.delVulnFromServiceASYNC, process_pending=True)
def testDelVulnFromObjectSYNC(self): host = Host('coco') vuln = ModelObjectVuln("int_mock0") host.addChild(vuln.getID(), vuln) self.genericDelTest(host, vuln, controller.ModelController.delVulnSYNC)
def testDelVulnFromHostASYNC(self): host = Host('coco') vuln = ModelObjectVuln("int_mock0") host.addChild(vuln.getID(), vuln) self.genericDelTest(host, vuln, controller.ModelController.delVulnFromHostASYNC, process_pending=True)