def test_load_from_msg(self):
for msg in self.cases.keys():
cr = CloudSummaryRecord()
cr.load_from_msg(msg)
cont = cr._record_content
for key in self.cases[msg].keys():
self.assertEqual(cont[key], self.cases[msg][key], "%s != %s for key %s" % (cont[key], self.cases[msg][key], key))
def test_mandatory_fields(self):
record = CloudSummaryRecord()
record.set_field('SiteName', 'MySite')
record.set_field('Month', 1)
record.set_field('Year', 2012)
record.set_field('GlobalUserName', 'me')
record.set_field('VO', 'myVO')
record.set_field('VOGroup', 'myVOGroup')
record.set_field('VORole', 'myVORole')
record.set_field('Status', 'completed')
record.set_field('CloudType', 'OpenStack')
record.set_field('ImageId', 'img-01')
record.set_field('NumberOfVMs', 1)
try:
record._check_fields()
except Exception, e:
self.fail('_check_fields method failed: %s [%s]' % (str(e), str(type(e))))
