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