def testAttacker(self): iatkeps = [AttackerEnvironmentProperties(self.iAttackers[0]["theEnvironmentProperties"][0]["theName"],self.iAttackers[0]["theEnvironmentProperties"][0]["theRoles"],self.iAttackers[0]["theEnvironmentProperties"][0]["theMotives"],self.iAttackers[0]["theEnvironmentProperties"][0]["theCapabilities"]),AttackerEnvironmentProperties(self.iAttackers[0]["theEnvironmentProperties"][1]["theName"],self.iAttackers[0]["theEnvironmentProperties"][1]["theRoles"],self.iAttackers[0]["theEnvironmentProperties"][1]["theMotives"],self.iAttackers[0]["theEnvironmentProperties"][1]["theCapabilities"])] iatk = AttackerParameters(self.iAttackers[0]["theName"], self.iAttackers[0]["theDescription"], self.iAttackers[0]["theImage"],[],iatkeps) b = Borg() b.dbProxy.addAttacker(iatk) oAttackers = b.dbProxy.getAttackers() o = oAttackers[self.iAttackers[0]["theName"]] self.assertEqual(iatk.name(), o.name()) self.assertEqual(iatk.description(),o.description()) self.assertEqual(iatk.image(),o.image()) oatkeps = o.environmentProperties() self.assertEqual(iatkeps[0].name(), oatkeps[0].name()) self.assertEqual(str(iatkeps[0].roles()[0]), str(oatkeps[0].roles()[0])) self.assertEqual(str(iatkeps[0].roles()[0]), o.roles('Day','')[0]) self.assertEqual(iatkeps[0].roles(), list(o.roles('','Maximise'))) self.assertEqual(str(iatkeps[0].motives()[0]), str(oatkeps[0].motives()[0])) self.assertEqual(str(iatkeps[0].motives()[0]), str(o.motives('Day','')[0])) self.assertEqual(iatkeps[0].motives(), list(o.motives('','Maximise'))) self.assertEqual(str(iatkeps[0].capabilities()[0][0]), str(oatkeps[0].capabilities()[0][0])) self.assertEqual(str(iatkeps[0].capabilities()[0][1]), str(oatkeps[0].capabilities()[0][1])) self.assertEqual(iatkeps[0].capabilities()[0][0], o.capability('Day','')[0][0]) self.assertEqual(iatkeps[0].capabilities()[0][0], list(o.capability('','Maximise'))[0][0]) iatk.theName = 'Updated name' iatk.setId(o.id()) b.dbProxy.updateAttacker(iatk) oAttackers = b.dbProxy.getAttackers() o = oAttackers["Updated name"] self.assertEqual(o.name(),'Updated name') b.dbProxy.deleteAttacker(o.id())
def parameters(self): parameters = AttackerParameters(self.theAttackerName, self.theAttackerDescription, self.theAttackerImage, self.theTags, self.theEnvironmentProperties) parameters.setId(self.theAttackerId) return parameters
def testAttacker(self): iatkeps = [ AttackerEnvironmentProperties( self.iAttackers[0]["theEnvironmentProperties"][0]["theName"], self.iAttackers[0]["theEnvironmentProperties"][0]["theRoles"], self.iAttackers[0]["theEnvironmentProperties"][0] ["theMotives"], self.iAttackers[0]["theEnvironmentProperties"] [0]["theCapabilities"]), AttackerEnvironmentProperties( self.iAttackers[0]["theEnvironmentProperties"][1]["theName"], self.iAttackers[0]["theEnvironmentProperties"][1]["theRoles"], self.iAttackers[0]["theEnvironmentProperties"][1] ["theMotives"], self.iAttackers[0]["theEnvironmentProperties"] [1]["theCapabilities"]) ] iatk = AttackerParameters(self.iAttackers[0]["theName"], self.iAttackers[0]["theDescription"], self.iAttackers[0]["theImage"], [], iatkeps) b = Borg() b.dbProxy.addAttacker(iatk) oAttackers = b.dbProxy.getAttackers() o = oAttackers[self.iAttackers[0]["theName"]] self.assertEqual(iatk.name(), o.name()) self.assertEqual(iatk.description(), o.description()) self.assertEqual(iatk.image(), o.image()) oatkeps = o.environmentProperties() self.assertEqual(iatkeps[0].name(), oatkeps[0].name()) self.assertEqual(str(iatkeps[0].roles()[0]), str(oatkeps[0].roles()[0])) self.assertEqual(str(iatkeps[0].roles()[0]), o.roles('Day', '')[0]) self.assertEqual(iatkeps[0].roles(), list(o.roles('', 'Maximise'))) self.assertEqual(str(iatkeps[0].motives()[0]), str(oatkeps[0].motives()[0])) self.assertEqual(str(iatkeps[0].motives()[0]), str(o.motives('Day', '')[0])) self.assertEqual(iatkeps[0].motives(), list(o.motives('', 'Maximise'))) self.assertEqual(str(iatkeps[0].capabilities()[0][0]), str(oatkeps[0].capabilities()[0][0])) self.assertEqual(str(iatkeps[0].capabilities()[0][1]), str(oatkeps[0].capabilities()[0][1])) self.assertEqual(iatkeps[0].capabilities()[0][0], o.capability('Day', '')[0][0]) self.assertEqual(iatkeps[0].capabilities()[0][0], list(o.capability('', 'Maximise'))[0][0]) iatk.theName = 'Updated name' iatk.setId(o.id()) b.dbProxy.updateAttacker(iatk) oAttackers = b.dbProxy.getAttackers() o = oAttackers["Updated name"] self.assertEqual(o.name(), 'Updated name') b.dbProxy.deleteAttacker(o.id())
def update_attacker(self, attacker, name): attacker_params = AttackerParameters( name=attacker.theName, desc=attacker.theDescription, image=attacker.theImage, tags=attacker.theTags, properties=attacker.theEnvironmentProperties) try: attackerId = self.db_proxy.getDimensionId(name, 'attacker') attacker_params.setId(attackerId) self.db_proxy.updateAttacker(attacker_params) except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def update_attacker(self, attacker, name): attacker_params = AttackerParameters( name=attacker.theName, desc=attacker.theDescription, image=attacker.theImage, tags=attacker.theTags, properties=attacker.theEnvironmentProperties ) try: attackerId = self.db_proxy.getDimensionId(name,'attacker') attacker_params.setId(attackerId) self.db_proxy.updateAttacker(attacker_params) except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def update_attacker(self, attacker, name): found_attacker = self.get_attacker_by_name(name, simplify=False) attacker_params = AttackerParameters( name=attacker.theName, desc=attacker.theDescription, image=attacker.theImage, tags=attacker.theTags, properties=attacker.theEnvironmentProperties) attacker_params.setId(found_attacker.theId) try: self.db_proxy.updateAttacker(attacker_params) except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def update_attacker(self, attacker, name): found_attacker = self.get_attacker_by_name(name, simplify=False) attacker_params = AttackerParameters( name=attacker.theName, desc=attacker.theDescription, image=attacker.theImage, tags=attacker.theTags, properties=attacker.theEnvironmentProperties ) attacker_params.setId(found_attacker.theId) try: self.db_proxy.updateAttacker(attacker_params) except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def parameters(self): parameters = AttackerParameters(self.theAttackerName,self.theAttackerDescription,self.theAttackerImage,self.theTags,self.theEnvironmentProperties) parameters.setId(self.theAttackerId) return parameters