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())
b.dbProxy.deleteCountermeasure(ocm.id())
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())
