126 lines
4.9 KiB
Python
126 lines
4.9 KiB
Python
# Copyright 2013 IBM Corp.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import mock
|
|
|
|
from nova.db import api as db
|
|
from nova import exception
|
|
from nova.objects import instance_fault
|
|
from nova.tests.unit.objects import test_objects
|
|
from nova.tests import uuidsentinel as uuids
|
|
|
|
|
|
fake_faults = {
|
|
'fake-uuid': [
|
|
{'id': 1, 'instance_uuid': uuids.faults_instance, 'code': 123,
|
|
'message': 'msg1', 'details': 'details', 'host': 'host',
|
|
'deleted': False, 'created_at': None, 'updated_at': None,
|
|
'deleted_at': None},
|
|
{'id': 2, 'instance_uuid': uuids.faults_instance, 'code': 456,
|
|
'message': 'msg2', 'details': 'details', 'host': 'host',
|
|
'deleted': False, 'created_at': None, 'updated_at': None,
|
|
'deleted_at': None},
|
|
]
|
|
}
|
|
|
|
|
|
class _TestInstanceFault(object):
|
|
@mock.patch.object(db, 'instance_fault_get_by_instance_uuids',
|
|
return_value=fake_faults)
|
|
def test_get_latest_for_instance(self, get_mock):
|
|
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])
|
|
get_mock.assert_called_once_with(self.context, ['fake-uuid'])
|
|
|
|
@mock.patch.object(db, 'instance_fault_get_by_instance_uuids',
|
|
return_value={})
|
|
def test_get_latest_for_instance_with_none(self, get_mock):
|
|
fault = instance_fault.InstanceFault.get_latest_for_instance(
|
|
self.context, 'fake-uuid')
|
|
self.assertIsNone(fault)
|
|
get_mock.assert_called_once_with(self.context, ['fake-uuid'])
|
|
|
|
@mock.patch.object(db, 'instance_fault_get_by_instance_uuids',
|
|
return_value=fake_faults)
|
|
def test_get_by_instance(self, get_mock):
|
|
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])
|
|
get_mock.assert_called_once_with(self.context, ['fake-uuid'])
|
|
|
|
@mock.patch.object(db, 'instance_fault_get_by_instance_uuids',
|
|
return_value={})
|
|
def test_get_by_instance_with_none(self, get_mock):
|
|
faults = instance_fault.InstanceFaultList.get_by_instance_uuids(
|
|
self.context, ['fake-uuid'])
|
|
self.assertEqual(0, len(faults))
|
|
get_mock.assert_called_once_with(self.context, ['fake-uuid'])
|
|
|
|
@mock.patch('nova.cells.rpcapi.CellsAPI.instance_fault_create_at_top')
|
|
@mock.patch('nova.db.api.instance_fault_create')
|
|
def _test_create(self, update_cells, mock_create, cells_fault_create):
|
|
mock_create.return_value = fake_faults['fake-uuid'][1]
|
|
fault = instance_fault.InstanceFault(context=self.context)
|
|
fault.instance_uuid = uuids.faults_instance
|
|
fault.code = 456
|
|
fault.message = 'foo'
|
|
fault.details = 'you screwed up'
|
|
fault.host = 'myhost'
|
|
fault.create()
|
|
self.assertEqual(2, fault.id)
|
|
mock_create.assert_called_once_with(self.context,
|
|
{'instance_uuid': uuids.faults_instance,
|
|
'code': 456,
|
|
'message': 'foo',
|
|
'details': 'you screwed up',
|
|
'host': 'myhost'})
|
|
if update_cells:
|
|
cells_fault_create.assert_called_once_with(
|
|
self.context, fake_faults['fake-uuid'][1])
|
|
else:
|
|
self.assertFalse(cells_fault_create.called)
|
|
|
|
def test_create_no_cells(self):
|
|
self.flags(enable=False, group='cells')
|
|
self._test_create(False)
|
|
|
|
def test_create_api_cell(self):
|
|
self.flags(cell_type='api', enable=True, group='cells')
|
|
self._test_create(False)
|
|
|
|
def test_create_compute_cell(self):
|
|
self.flags(cell_type='compute', enable=True, group='cells')
|
|
self._test_create(True)
|
|
|
|
def test_create_already_created(self):
|
|
fault = instance_fault.InstanceFault(context=self.context)
|
|
fault.id = 1
|
|
self.assertRaises(exception.ObjectActionError,
|
|
fault.create)
|
|
|
|
|
|
class TestInstanceFault(test_objects._LocalTest,
|
|
_TestInstanceFault):
|
|
pass
|
|
|
|
|
|
class TestInstanceFaultRemote(test_objects._RemoteTest,
|
|
_TestInstanceFault):
|
|
pass
|