def testAddServiceToInterface(self): interface = Interface('coco') serv = Service('cuca') interface.addChild(serv) self.assertIn(serv, interface.childs.values(), 'Service not in childs') self.assertIn(serv, interface.getAllServices(), 'Service not accessible')
def testInterfaceWithMultipleChildTypes(self): inter = Interface('coco') serv = Service('cuca') vuln = ModelObjectVuln('vuln') inter.addChild(serv) inter.addChild(vuln) self.assertEquals(len(inter.getVulns()), 1, "Vulns added is not 1") self.assertIn(vuln, inter.getVulns(), "Vuln not accessible") self.assertEquals(len(inter.getAllServices()), 1, "Services added is not 1")