def update_countermeasure(self, countermeasure, name): countermeasure_params = CountermeasureParameters( cmName=countermeasure.name(), cmDesc=countermeasure.description(), cmType=countermeasure.type(), tags=countermeasure.tags(), cProps=countermeasure.environmentProperties() ) try: cmId = self.db_proxy.getDimensionId(name,'countermeasure') countermeasure_params.setId(cmId) self.db_proxy.updateCountermeasure(countermeasure_params) except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def update_countermeasure(self, countermeasure, name): found_countermeasure = self.get_countermeasure_by_name(name, simplify=False) countermeasure_params = CountermeasureParameters( cmName=countermeasure.name(), cmDesc=countermeasure.description(), cmType=countermeasure.type(), tags=countermeasure.tags(), cProps=countermeasure.environmentProperties() ) countermeasure_params.setId(found_countermeasure.id()) try: self.db_proxy.updateCountermeasure(countermeasure_params) except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def update_countermeasure(self, countermeasure, name): countermeasure_params = CountermeasureParameters( cmName=countermeasure.name(), cmDesc=countermeasure.description(), cmType=countermeasure.type(), tags=countermeasure.tags(), cProps=countermeasure.environmentProperties() ) try: cmId = self.db_proxy.getDimensionId(name,'countermeasure') countermeasure_params.setId(cmId) self.db_proxy.updateCountermeasure(countermeasure_params) except ObjectNotFound as ex: self.close() raise ObjectNotFoundHTTPError('The provided countermeasure name') except DatabaseProxyException as ex: self.close() raise ARMHTTPError(ex) except ARMException as ex: self.close() raise ARMHTTPError(ex)
def testCountermeasure(self): icep = CountermeasureEnvironmentProperties(self.iCountermeasures[0]["theEnvironmentProperties"][0][0],[self.iCountermeasures[0]["theEnvironmentProperties"][0][2]],[Target(self.iCountermeasures[0]["theEnvironmentProperties"][0][3],self.iCountermeasures[0]["theEnvironmentProperties"][0][4],"None")],[0,0,0,0,0,0,0,0],['None','None','None','None','None','None','None','None'],self.iCountermeasures[0]["theEnvironmentProperties"][0][1],[self.iCountermeasures[0]["theEnvironmentProperties"][0][5]]) icm = CountermeasureParameters(self.iCountermeasures[0]["theName"],self.iCountermeasures[0]["theDescription"],self.iCountermeasures[0]["theType"],[],[icep]) b = Borg() b.dbProxy.addCountermeasure(icm) ocms = b.dbProxy.getCountermeasures() ocm = ocms[self.iCountermeasures[0]["theName"]] self.assertEqual(icm.name(), ocm.name()) self.assertEqual(icm.type(),ocm.type()) self.assertEqual(icm.description(),ocm.description()) icm.theName = 'Updated countermeasure' icm.setId(ocm.id()) b.dbProxy.updateCountermeasure(icm) ocms = b.dbProxy.getCountermeasures() ocm = ocms['Updated countermeasure'] self.assertEqual('Updated countermeasure', ocm.name()) self.assertEqual(icm.type(),ocm.type()) self.assertEqual(icm.description(),ocm.description()) b.dbProxy.deleteCountermeasure(ocm.id())