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 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 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)
