def testValueType(self):
ivt1 = ValueTypeParameters(self.iVtypes[0]["theName"],
self.iVtypes[0]["theDescription"],
self.iVtypes[0]["theType"],
self.iVtypes[0]['theEnvironmentName'])
ivt2 = ValueTypeParameters(self.iVtypes[1]["theName"],
self.iVtypes[1]["theDescription"],
self.iVtypes[1]["theType"],
self.iVtypes[1]['theEnvironmentName'])
b = Borg()
b.dbProxy.addValueType(ivt1)
b.dbProxy.addValueType(ivt2)
oVVtypes = b.dbProxy.getValueTypes('vulnerability_type')
ovt1 = oVVtypes[ivt1.id()]
self.assertEqual(ivt1.name(), ovt1.name())
self.assertEqual(ivt1.description(), ovt1.description())
self.assertEqual(ivt1.type(), ovt1.type())
oVTtypes = b.dbProxy.getValueTypes('threat_type')
ovt2 = oVTtypes[ivt2.id()]
self.assertEqual(ivt2.name(), ovt2.name())
self.assertEqual(ivt2.description(), ovt2.description())
self.assertEqual(ivt2.type(), ovt2.type())
b.dbProxy.deleteVulnerabilityType(ovt1.id())
b.dbProxy.deleteThreatType(ovt2.id())
def testValueType(self):
ivt1 = ValueTypeParameters(self.iVtypes[0]["theName"], self.iVtypes[0]["theDescription"], self.iVtypes[0]["theType"])
ivt2 = ValueTypeParameters(self.iVtypes[1]["theName"], self.iVtypes[1]["theDescription"], self.iVtypes[1]["theType"])
b = Borg()
b.dbProxy.addValueType(ivt1)
b.dbProxy.addValueType(ivt2)
oVVtypes = b.dbProxy.getValueTypes('vulnerability_type')
ovt1 = oVVtypes[ivt1.id()]
self.assertEqual(ivt1.name(), ovt1.name())
self.assertEqual(ivt1.description(),ovt1.description())
self.assertEqual(ivt1.type(),ovt1.type())
oVTtypes = b.dbProxy.getValueTypes('threat_type')
ovt2 = oVTtypes[ivt2.id()]
self.assertEqual(ivt2.name(), ovt2.name())
self.assertEqual(ivt2.description(),ovt2.description())
self.assertEqual(ivt2.type(),ovt2.type())
b.dbProxy.deleteVulnerabilityType(ovt1.id())
b.dbProxy.deleteThreatType(ovt2.id())
