def testRisk(self):
imcep = [MisuseCaseEnvironmentProperties(self.imc[0]["theEnvironmentProperties"][0]["theName"],self.imc[0]["theEnvironmentProperties"][0]["theDescription"])]
imcp = MisuseCase(int(0), self.imc[0]["theName"], imcep,self.imc[0]["theRisk"])
irp = RiskParameters(self.iRisks[0]["theName"],self.iRisks[0]["threatName"],self.iRisks[0]["vulName"], imcp,[])
b = Borg()
b.dbProxy.addRisk(irp)
oRisks = b.dbProxy.getRisks()
o = oRisks[self.iRisks[0]["theName"]]
self.assertEqual(irp.name(), o.name())
self.assertEqual(irp.threat(),o.threat())
self.assertEqual(irp.vulnerability(),o.vulnerability())
# scoreDetails = b.dbProxy.riskScore(self.iRisks[0]["threatName"],self.iRisks[0]["vulName"],self.iaeps1[0].name())
# preScore = scoreDetails[0][1]
# postScore = scoreDetails[0][2]
# self.assertEqual(preScore,9)
# self.assertEqual(postScore,9)
imcp.theName = 'Updated risk'
imcp.theId = o.misuseCase().id()
irp.theRiskName = 'Updated risk'
irp.theMisuseCase = imcp
irp.setId(o.id())
b.dbProxy.updateRisk(irp)
oRisks = b.dbProxy.getRisks()
o = oRisks['Updated risk']
self.assertEqual(o.name(),'Updated risk')
self.assertEqual(irp.threat(),o.threat())
self.assertEqual(irp.vulnerability(),o.vulnerability())
b.dbProxy.deleteRisk(o.id())
def update_object(self, risk, risk_name):
params = RiskParameters(riskName=risk.theName,threatName=risk.theThreatName,vulName=risk.theVulnerabilityName,mc=risk.theMisuseCase,rTags=risk.theTags)
try:
riskId = self.db_proxy.getDimensionId(risk_name,'risk')
params.setId(riskId)
self.db_proxy.updateRisk(params)
except DatabaseProxyException as ex:
self.close()
raise ARMHTTPError(ex)
except ARMException as ex:
self.close()
raise ARMHTTPError(ex)
def update_risk(self, risk_name, risk):
params = RiskParameters(riskName=risk.theName,threatName=risk.theThreatName,vulName=risk.theVulnerabilityName,mc=risk.theMisuseCase,rTags=risk.theTags)
try:
riskId = self.db_proxy.getDimensionId(risk_name,'risk')
params.setId(riskId)
self.db_proxy.updateRisk(params)
except DatabaseProxyException as ex:
self.close()
raise ARMHTTPError(ex)
except ARMException as ex:
self.close()
raise ARMHTTPError(ex)
def update_risk(self, risk_name, risk):
found_risk = self.get_risk_by_name(risk_name)
found_mc = self.get_misuse_case_by_risk_name(risk_name)
risk.theMisuseCase.theId = found_mc.theId
params = RiskParameters(riskName=risk.theName,threatName=risk.theThreatName,vulName=risk.theVulnerabilityName,mc=risk.theMisuseCase,rTags=risk.theTags)
params.setId(found_risk.theId)
try:
self.db_proxy.updateRisk(params)
except DatabaseProxyException as ex:
self.close()
raise ARMHTTPError(ex)
except ARMException as ex:
self.close()
raise ARMHTTPError(ex)
def testRisk(self):
imcep = [
MisuseCaseEnvironmentProperties(
self.imc[0]["theEnvironmentProperties"][0]["theName"],
self.imc[0]["theEnvironmentProperties"][0]["theDescription"])
]
imcp = MisuseCase(int(0), self.imc[0]["theName"], imcep,
self.imc[0]["theRisk"])
irp = RiskParameters(self.iRisks[0]["theName"],
self.iRisks[0]["threatName"],
self.iRisks[0]["vulName"], imcp, [])
b = Borg()
b.dbProxy.addRisk(irp)
oRisks = b.dbProxy.getRisks()
o = oRisks[self.iRisks[0]["theName"]]
self.assertEqual(irp.name(), o.name())
self.assertEqual(irp.threat(), o.threat())
self.assertEqual(irp.vulnerability(), o.vulnerability())
scoreDetails = b.dbProxy.riskScore(self.iRisks[0]["threatName"],
self.iRisks[0]["vulName"],
imcep[0].name())
preScore = scoreDetails[0][1]
postScore = scoreDetails[0][2]
self.assertEqual(preScore, 9)
self.assertEqual(postScore, 9)
imcp.theName = 'Updated risk'
imcp.theId = o.misuseCase().id()
irp.theRiskName = 'Updated risk'
irp.theMisuseCase = imcp
irp.setId(o.id())
b.dbProxy.updateRisk(irp)
oRisks = b.dbProxy.getRisks()
o = oRisks['Updated risk']
self.assertEqual(o.name(), 'Updated risk')
self.assertEqual(irp.threat(), o.threat())
self.assertEqual(irp.vulnerability(), o.vulnerability())
b.dbProxy.deleteRisk(o.id())
def update_risk(self, risk_name, risk):
found_risk = self.get_risk_by_name(risk_name)
found_mc = self.get_misuse_case_by_risk_name(risk_name)
risk.theMisuseCase.theId = found_mc.theId
params = RiskParameters(
riskName=risk.theName,
threatName=risk.theThreatName,
vulName=risk.theVulnerabilityName,
mc=risk.theMisuseCase,
rTags=risk.theTags
)
params.setId(found_risk.theId)
try:
self.db_proxy.updateRisk(params)
except DatabaseProxyException as ex:
self.close()
raise ARMHTTPError(ex)
except ARMException as ex:
self.close()
raise ARMHTTPError(ex)
def parameters(self): parameters = RiskParameters(self.theRiskName,self.theThreatName,self.theVulnerabilityName,self.theMisuseCase,self.theTags) parameters.setId(self.theRiskId) return parameters
def parameters(self):
parameters = RiskParameters(self.theRiskName, self.theThreatName,
self.theVulnerabilityName,
self.theMisuseCase, self.theTags)
parameters.setId(self.theRiskId)
return parameters
