def testThreat(self): iteps = [ThreatEnvironmentProperties(self.iThreats[0]["theEnvironmentProperties"][0]["theName"],self.iThreats[0]["theEnvironmentProperties"][0]["theLikelihood"],self.iThreats[0]["theEnvironmentProperties"][0]["theAssets"],self.iThreats[0]["theEnvironmentProperties"][0]["theAttackers"],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1])] itps = ThreatParameters(self.iThreats[0]["theName"],self.iThreats[0]["theType"],self.iThreats[0]["theMethod"],[],iteps) b = Borg() b.dbProxy.addThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats[self.iThreats[0]["theName"]] self.assertEqual(itps.name(), o.name()) self.assertEqual(itps.type(),o.type()) self.assertEqual(itps.method(),o.method()) oteps = o.environmentProperties() self.assertEqual(iteps[0].name(), oteps[0].name()) self.assertEqual(iteps[0].likelihood()[0], oteps[0].likelihood()[0]) self.assertEqual(str(iteps[0].assets()[0]), str(oteps[0].assets()[0])) self.assertEqual(str(iteps[0].attackers()[0]), str(oteps[0].attackers()[0])) self.assertEqual(str(iteps[0].rationale()[0]), str(oteps[0].rationale()[0])) itps.theName = 'Updated threat' itps.setId(o.id()) b.dbProxy.updateThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats['Updated threat'] self.assertEqual(o.name(),'Updated threat') b.dbProxy.deleteThreat(o.id())
def testThreat(self): iteps = [ThreatEnvironmentProperties(self.iThreats[0]["theEnvironmentProperties"][0]["theName"],self.iThreats[0]["theEnvironmentProperties"][0]["theLikelihood"],self.iThreats[0]["theEnvironmentProperties"][0]["theAssets"],self.iThreats[0]["theEnvironmentProperties"][0]["theAttackers"],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1]),ThreatEnvironmentProperties(self.iThreats[0]["theEnvironmentProperties"][1]["theName"],self.iThreats[0]["theEnvironmentProperties"][1]["theLikelihood"],self.iThreats[0]["theEnvironmentProperties"][1]["theAssets"],self.iThreats[0]["theEnvironmentProperties"][1]["theAttackers"],self.iThreats[0]["theEnvironmentProperties"][1]["theProperties"][0][1],self.iThreats[0]["theEnvironmentProperties"][1]["theProperties"][0][1])] itps = ThreatParameters(self.iThreats[0]["theName"],self.iThreats[0]["theType"],self.iThreats[0]["theMethod"],[],iteps) b = Borg() b.dbProxy.addThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats[self.iThreats[0]["theName"]] self.assertEqual(itps.name(), o.name()) self.assertEqual(itps.type(),o.type()) self.assertEqual(itps.method(),o.method()) oteps = o.environmentProperties() self.assertEqual(iteps[0].name(), oteps[0].name()) self.assertEqual(iteps[0].likelihood(), oteps[0].likelihood()) self.assertEqual(iteps[0].likelihood(),o.likelihood('Day','Maximise','None')) self.assertEqual(iteps[0].likelihood(),o.likelihood('','Maximise','None')) self.assertEqual(iteps[0].likelihood(),o.likelihood('','Override','Day')) self.assertEqual(iteps[0].assets(), oteps[0].assets()) self.assertEqual(iteps[0].assets(), o.assets('Day','')) self.assertEqual(iteps[0].assets(), list(o.assets('','Maximise'))) self.assertEqual(iteps[0].attackers(), oteps[0].attackers()) self.assertEqual(iteps[0].attackers(), o.attackers('Day','')) self.assertEqual(iteps[0].attackers(), list(o.attackers('','Maximise'))) self.assertEqual(str(iteps[0].rationale()[0]), oteps[0].rationale()[0]) self.assertEqual(str(iteps[0].rationale()[1]), oteps[0].rationale()[1]) self.assertEqual(str(iteps[0].rationale()[2]), oteps[0].rationale()[2]) self.assertEqual(str(iteps[0].rationale()[3]), oteps[0].rationale()[3]) self.assertEqual(str(iteps[0].rationale()[4]), oteps[0].rationale()[4]) self.assertEqual(str(iteps[0].rationale()[5]), oteps[0].rationale()[5]) self.assertEqual(str(iteps[0].rationale()[6]), oteps[0].rationale()[6]) self.assertEqual(str(iteps[0].rationale()[7]), oteps[0].rationale()[7]) self.assertEqual(array([0,2,0,0,0,0,0,0]).tolist(),o.securityProperties('','Maximise','').tolist()) self.assertEqual(array([0,2,0,0,0,0,0,0]).tolist(),o.securityProperties('','Override','Day').tolist()) self.assertEqual([['Integrity','Medium']],o.propertyList('','Maximise','')) self.assertEqual([['Integrity','Medium']],o.propertyList('','Override','Day')) envName = self.iThreats[0]["theEnvironmentProperties"][0]["theName"] self.assertEqual(iteps[0].likelihood(), o.likelihood(envName,'',envName)) self.assertEqual(iteps[0].assets(), o.assets(envName,'')) self.assertEqual(iteps[0].attackers(), o.attackers(envName,'')) itps.theName = 'Updated threat' itps.setId(o.id()) b.dbProxy.updateThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats['Updated threat'] self.assertEqual(o.name(),'Updated threat') b.dbProxy.deleteThreat(o.id())
def testThreat(self): iteps = [ThreatEnvironmentProperties(self.iThreats[0]["theEnvironmentProperties"][0]["theName"],self.iThreats[0]["theEnvironmentProperties"][0]["theLikelihood"],self.iThreats[0]["theEnvironmentProperties"][0]["theAssets"],self.iThreats[0]["theEnvironmentProperties"][0]["theAttackers"],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1])] itps = ThreatParameters(self.iThreats[0]["theName"],self.iThreats[0]["theType"],self.iThreats[0]["theMethod"],[],iteps) b = Borg() b.dbProxy.addThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats[self.iThreats[0]["theName"]] self.assertEqual(itps.name(), o.name()) self.assertEqual(itps.type(),o.type()) self.assertEqual(itps.method(),o.method()) oteps = o.environmentProperties() self.assertEqual(iteps[0].name(), oteps[0].name()) self.assertEqual(iteps[0].likelihood()[0], oteps[0].likelihood()[0]) self.assertEqual(str(iteps[0].assets()[0]), str(oteps[0].assets()[0])) self.assertEqual(str(iteps[0].attackers()[0]), str(oteps[0].attackers()[0])) self.assertEqual(str(iteps[0].rationale()[0]), str(oteps[0].rationale()[0])) b.dbProxy.deleteThreat(o.id())
def testThreat(self): iteps = [ThreatEnvironmentProperties(self.iThreats[0]["theEnvironmentProperties"][0]["theName"],self.iThreats[0]["theEnvironmentProperties"][0]["theLikelihood"],self.iThreats[0]["theEnvironmentProperties"][0]["theAssets"],self.iThreats[0]["theEnvironmentProperties"][0]["theAttackers"],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1],self.iThreats[0]["theEnvironmentProperties"][0]["theProperties"][0][1])] itps = ThreatParameters(self.iThreats[0]["theName"],self.iThreats[0]["theType"],self.iThreats[0]["theMethod"],[],iteps) b = Borg() b.dbProxy.addThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats[self.iThreats[0]["theName"]] self.assertEqual(itps.name(), o.name()) self.assertEqual(itps.type(),o.type()) self.assertEqual(itps.method(),o.method()) oteps = o.environmentProperties() self.assertEqual(iteps[0].name(), oteps[0].name()) self.assertEqual(iteps[0].likelihood()[0], oteps[0].likelihood()[0]) self.assertEqual(iteps[0].assets(), oteps[0].assets()) self.assertEqual(iteps[0].attackers(), oteps[0].attackers()) self.assertEqual(str(iteps[0].rationale()[0]), oteps[0].rationale()[0]) self.assertEqual(str(iteps[0].rationale()[1]), oteps[0].rationale()[1]) self.assertEqual(str(iteps[0].rationale()[2]), oteps[0].rationale()[2]) self.assertEqual(str(iteps[0].rationale()[3]), oteps[0].rationale()[3]) self.assertEqual(str(iteps[0].rationale()[4]), oteps[0].rationale()[4]) self.assertEqual(str(iteps[0].rationale()[5]), oteps[0].rationale()[5]) self.assertEqual(str(iteps[0].rationale()[6]), oteps[0].rationale()[6]) self.assertEqual(str(iteps[0].rationale()[7]), oteps[0].rationale()[7]) envName = self.iThreats[0]["theEnvironmentProperties"][0]["theName"] self.assertEqual(iteps[0].likelihood(), o.likelihood(envName,'',envName)) self.assertEqual(iteps[0].assets(), o.assets(envName,'')) self.assertEqual(iteps[0].attackers(), o.attackers(envName,'')) itps.theName = 'Updated threat' itps.setId(o.id()) b.dbProxy.updateThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats['Updated threat'] self.assertEqual(o.name(),'Updated threat') b.dbProxy.deleteThreat(o.id())
def testThreat(self): iteps = [ ThreatEnvironmentProperties( self.iThreats[0]["theEnvironmentProperties"][0]["theName"], self.iThreats[0]["theEnvironmentProperties"][0] ["theLikelihood"], self.iThreats[0]["theEnvironmentProperties"][0]["theAssets"], self.iThreats[0]["theEnvironmentProperties"][0] ["theAttackers"], self.iThreats[0]["theEnvironmentProperties"] [0]["theProperties"][0][1], self.iThreats[0] ["theEnvironmentProperties"][0]["theProperties"][0][1]), ThreatEnvironmentProperties( self.iThreats[0]["theEnvironmentProperties"][1]["theName"], self.iThreats[0]["theEnvironmentProperties"][1] ["theLikelihood"], self.iThreats[0]["theEnvironmentProperties"][1]["theAssets"], self.iThreats[0]["theEnvironmentProperties"][1] ["theAttackers"], self.iThreats[0]["theEnvironmentProperties"] [1]["theProperties"][0][1], self.iThreats[0] ["theEnvironmentProperties"][1]["theProperties"][0][1]) ] itps = ThreatParameters(self.iThreats[0]["theName"], self.iThreats[0]["theType"], self.iThreats[0]["theMethod"], [], iteps) b = Borg() b.dbProxy.addThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats[self.iThreats[0]["theName"]] self.assertEqual(itps.name(), o.name()) self.assertEqual(itps.type(), o.type()) self.assertEqual(itps.method(), o.method()) oteps = o.environmentProperties() self.assertEqual(iteps[0].name(), oteps[0].name()) self.assertEqual(iteps[0].likelihood(), oteps[0].likelihood()) self.assertEqual(iteps[0].likelihood(), o.likelihood('Day', 'Maximise', 'None')) self.assertEqual(iteps[0].likelihood(), o.likelihood('', 'Maximise', 'None')) self.assertEqual(iteps[0].likelihood(), o.likelihood('', 'Override', 'Day')) self.assertEqual(iteps[0].assets(), oteps[0].assets()) self.assertEqual(iteps[0].assets(), o.assets('Day', '')) self.assertEqual(iteps[0].assets(), list(o.assets('', 'Maximise'))) self.assertEqual(iteps[0].attackers(), oteps[0].attackers()) self.assertEqual(iteps[0].attackers(), o.attackers('Day', '')) self.assertEqual(iteps[0].attackers(), list(o.attackers('', 'Maximise'))) self.assertEqual(str(iteps[0].rationale()[0]), oteps[0].rationale()[0]) self.assertEqual(str(iteps[0].rationale()[1]), oteps[0].rationale()[1]) self.assertEqual(str(iteps[0].rationale()[2]), oteps[0].rationale()[2]) self.assertEqual(str(iteps[0].rationale()[3]), oteps[0].rationale()[3]) self.assertEqual(str(iteps[0].rationale()[4]), oteps[0].rationale()[4]) self.assertEqual(str(iteps[0].rationale()[5]), oteps[0].rationale()[5]) self.assertEqual(str(iteps[0].rationale()[6]), oteps[0].rationale()[6]) self.assertEqual(str(iteps[0].rationale()[7]), oteps[0].rationale()[7]) self.assertEqual( array([0, 2, 0, 0, 0, 0, 0, 0]).tolist(), o.securityProperties('', 'Maximise', '').tolist()) self.assertEqual( array([0, 2, 0, 0, 0, 0, 0, 0]).tolist(), o.securityProperties('', 'Override', 'Day').tolist()) self.assertEqual([['Integrity', 'Medium']], o.propertyList('', 'Maximise', '')) self.assertEqual([['Integrity', 'Medium']], o.propertyList('', 'Override', 'Day')) envName = self.iThreats[0]["theEnvironmentProperties"][0]["theName"] self.assertEqual(iteps[0].likelihood(), o.likelihood(envName, '', envName)) self.assertEqual(iteps[0].assets(), o.assets(envName, '')) self.assertEqual(iteps[0].attackers(), o.attackers(envName, '')) itps.theName = 'Updated threat' itps.setId(o.id()) b.dbProxy.updateThreat(itps) oThreats = b.dbProxy.getThreats() o = oThreats['Updated threat'] self.assertEqual(o.name(), 'Updated threat') b.dbProxy.deleteThreat(o.id())