def test_get_by_instance_with_none(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(self.context,
['fake-uuid']).AndReturn({})
self.mox.ReplayAll()
faults = instance_fault.InstanceFaultList.get_by_instance_uuids(
self.context, ['fake-uuid'])
self.assertEqual(0, len(faults))
def test_get_latest_for_instance_with_none(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(self.context,
['fake-uuid']).AndReturn({})
self.mox.ReplayAll()
fault = instance_fault.InstanceFault.get_latest_for_instance(
self.context, 'fake-uuid')
self.assertIsNone(fault)
def test_get_by_instance_with_none(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(self.context, ['fake-uuid']
).AndReturn({})
self.mox.ReplayAll()
faults = instance_fault.InstanceFaultList.get_by_instance_uuids(
self.context, ['fake-uuid'])
self.assertEqual(0, len(faults))
def test_get_latest_for_instance_with_none(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(self.context, ['fake-uuid']
).AndReturn({})
self.mox.ReplayAll()
fault = instance_fault.InstanceFault.get_latest_for_instance(
self.context, 'fake-uuid')
self.assertIsNone(fault)
def test_get_latest_for_instance(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(
self.context, ['fake-uuid']).AndReturn(fake_faults)
self.mox.ReplayAll()
fault = instance_fault.InstanceFault.get_latest_for_instance(
self.context, 'fake-uuid')
for key in fake_faults['fake-uuid'][0]:
self.assertEqual(fake_faults['fake-uuid'][0][key], fault[key])
def test_get_latest_for_instance(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(self.context, ['fake-uuid']
).AndReturn(fake_faults)
self.mox.ReplayAll()
fault = instance_fault.InstanceFault.get_latest_for_instance(
self.context, 'fake-uuid')
for key in fake_faults['fake-uuid'][0]:
self.assertEqual(fake_faults['fake-uuid'][0][key], fault[key])
def test_get_by_instance(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(
self.context, ['fake-uuid']).AndReturn(fake_faults)
self.mox.ReplayAll()
faults = instance_fault.InstanceFaultList.get_by_instance_uuids(
self.context, ['fake-uuid'])
for index, db_fault in enumerate(fake_faults['fake-uuid']):
for key in db_fault:
self.assertEqual(fake_faults['fake-uuid'][index][key],
faults[index][key])
def test_get_by_instance(self):
self.mox.StubOutWithMock(db, 'instance_fault_get_by_instance_uuids')
db.instance_fault_get_by_instance_uuids(self.context, ['fake-uuid']
).AndReturn(fake_faults)
self.mox.ReplayAll()
faults = instance_fault.InstanceFaultList.get_by_instance_uuids(
self.context, ['fake-uuid'])
for index, db_fault in enumerate(fake_faults['fake-uuid']):
for key in db_fault:
self.assertEqual(fake_faults['fake-uuid'][index][key],
faults[index][key])
def get_latest_for_instance(cls, context, instance_uuid):
db_faults = db.instance_fault_get_by_instance_uuids(
context, [instance_uuid])
if instance_uuid in db_faults and db_faults[instance_uuid]:
return cls._from_db_object(context, cls(),
db_faults[instance_uuid][0])
def get_by_instance_uuids(cls, context, instance_uuids):
db_faultdict = db.instance_fault_get_by_instance_uuids(
context, instance_uuids)
db_faultlist = itertools.chain(*db_faultdict.values())
return base.obj_make_list(context, cls(context), objects.InstanceFault,
db_faultlist)
