Merge "Fix aggregate placement sync issue"

This commit is contained in:
Zuul 2020-06-27 05:31:10 +00:00 committed by Gerrit Code Review
commit 279aa24b54
4 changed files with 183 additions and 28 deletions

View File

@ -6029,9 +6029,11 @@ class AggregateAPI(base.Base):
aggregate.add_host(host_name)
self.query_client.update_aggregates(context, [aggregate])
nodes = objects.ComputeNodeList.get_all_by_host(context, host_name)
node_name = nodes[0].hypervisor_hostname
try:
self.placement_client.aggregate_add_host(
context, aggregate.uuid, host_name=host_name)
context, aggregate.uuid, host_name=node_name)
except (exception.ResourceProviderNotFound,
exception.ResourceProviderAggregateRetrievalFailed,
exception.ResourceProviderUpdateFailed,
@ -6045,7 +6047,7 @@ class AggregateAPI(base.Base):
LOG.warning("Failed to associate %s with a placement "
"aggregate: %s. This may be corrected after running "
"nova-manage placement sync_aggregates.",
host_name, err)
node_name, err)
self._update_az_cache_for_host(context, host_name, aggregate.metadata)
# NOTE(jogo): Send message to host to support resource pools
self.compute_rpcapi.add_aggregate_host(context,
@ -6083,11 +6085,13 @@ class AggregateAPI(base.Base):
# we change anything on the nova side because if we did the nova stuff
# first we can't re-attempt this from the compute API if cleaning up
# placement fails.
nodes = objects.ComputeNodeList.get_all_by_host(context, host_name)
node_name = nodes[0].hypervisor_hostname
try:
# Anything else this raises is handled in the route handler as
# either a 409 (ResourceProviderUpdateConflict) or 500.
self.placement_client.aggregate_remove_host(
context, aggregate.uuid, host_name)
context, aggregate.uuid, node_name)
except exception.ResourceProviderNotFound as err:
# If the resource provider is not found then it's likely not part
# of the aggregate anymore anyway since provider aggregates are
@ -6095,7 +6099,7 @@ class AggregateAPI(base.Base):
# are just a grouping concept around resource providers. Log and
# continue.
LOG.warning("Failed to remove association of %s with a placement "
"aggregate: %s.", host_name, err)
"aggregate: %s.", node_name, err)
aggregate.delete_host(host_name)
self.query_client.update_aggregates(context, [aggregate])

View File

@ -11558,14 +11558,22 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertRaises(exception.AggregateNotFound,
self.api.delete_aggregate, self.context, aggr.id)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_check_az_for_aggregate(self, mock_add_host):
def test_check_az_for_aggregate(self, mock_add_host, mock_get_all_by_host):
# Ensure all conflict hosts can be returned
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host1 = values[0][1][0]
fake_host2 = values[0][1][1]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host1,
hypervisor_hostname=fake_host1),
objects.ComputeNode(
host=fake_host2,
hypervisor_hostname=fake_host2)])
aggr1 = self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host1)
aggr1 = self._init_aggregate_with_host(aggr1, None, None, fake_host2)
@ -11593,15 +11601,20 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertEqual(msg.event_type,
'aggregate.updateprop.end')
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_update_aggregate_no_az(self, mock_add_host):
def test_update_aggregate_no_az(self, mock_add_host, mock_get_all_by_host):
# Ensure metadata without availability zone can be
# updated,even the aggregate contains hosts belong
# to another availability zone
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host)
aggr2 = self._init_aggregate_with_host(None, 'fake_aggregate2', None,
@ -11617,15 +11630,21 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertEqual(msg.event_type,
'aggregate.updateprop.end')
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_update_aggregate_az_change(self, mock_add_host):
def test_update_aggregate_az_change(self, mock_add_host,
mock_get_all_by_host):
# Ensure availability zone can be updated,
# when the aggregate is the only one with
# availability zone
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
aggr1 = self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host)
self._init_aggregate_with_host(None, 'fake_aggregate2', None,
@ -11641,15 +11660,21 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertEqual(msg.event_type,
'aggregate.updatemetadata.end')
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_update_aggregate_az_fails(self, mock_add_host):
def test_update_aggregate_az_fails(self, mock_add_host,
mock_get_all_by_host):
# Ensure aggregate's availability zone can't be updated,
# when aggregate has hosts in other availability zone
fake_notifier.NOTIFICATIONS = []
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host)
aggr2 = self._init_aggregate_with_host(None, 'fake_aggregate2', None,
@ -11659,6 +11684,10 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.api.update_aggregate,
self.context, aggr2.id, metadata)
fake_host2 = values[0][1][1]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host2,
hypervisor_hostname=fake_host2)])
aggr3 = self._init_aggregate_with_host(None, 'fake_aggregate3',
None, fake_host2)
metadata = {'availability_zone': fake_zone}
@ -11676,14 +11705,20 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.api.update_aggregate, self.context,
aggr4.id, metadata)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_update_aggregate_az_fails_with_nova_az(self, mock_add_host):
def test_update_aggregate_az_fails_with_nova_az(self, mock_add_host,
mock_get_all_by_host):
# Ensure aggregate's availability zone can't be updated,
# when aggregate has hosts in other availability zone
fake_notifier.NOTIFICATIONS = []
values = _create_service_entries(self.context)
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
self._init_aggregate_with_host(None, 'fake_aggregate1',
CONF.default_availability_zone,
fake_host)
@ -11738,16 +11773,22 @@ class ComputeAPIAggrTestCase(BaseTestCase):
matchers.DictMatches({'availability_zone': 'fake_zone',
'foo_key2': 'foo_value2'}))
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
@mock.patch('nova.compute.utils.notify_about_aggregate_action')
def test_update_aggregate_metadata_no_az(self, mock_notify, mock_add_host):
def test_update_aggregate_metadata_no_az(self, mock_notify, mock_add_host,
mock_get_all_by_host):
# Ensure metadata without availability zone can be
# updated,even the aggregate contains hosts belong
# to another availability zone
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host)
aggr2 = self._init_aggregate_with_host(None, 'fake_aggregate2', None,
@ -11771,15 +11812,21 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertThat(aggr2.metadata,
matchers.DictMatches({'foo_key2': 'foo_value3'}))
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_update_aggregate_metadata_az_change(self, mock_add_host):
def test_update_aggregate_metadata_az_change(self, mock_add_host,
mock_get_all_by_host):
# Ensure availability zone can be updated,
# when the aggregate is the only one with
# availability zone
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
aggr1 = self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host)
self._init_aggregate_with_host(None, 'fake_aggregate2', None,
@ -11811,15 +11858,21 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertThat(aggr.metadata, matchers.DictMatches(
{'availability_zone': 'new_fake_zone', 'foo_key1': 'foo_value1'}))
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_update_aggregate_metadata_az_fails(self, mock_add_host):
def test_update_aggregate_metadata_az_fails(self, mock_add_host,
mock_get_all_by_host):
# Ensure aggregate's availability zone can't be updated,
# when aggregate has hosts in other availability zone
fake_notifier.NOTIFICATIONS = []
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
self._init_aggregate_with_host(None, 'fake_aggregate1',
fake_zone, fake_host)
aggr2 = self._init_aggregate_with_host(None, 'fake_aggregate2', None,
@ -11892,33 +11945,46 @@ class ComputeAPIAggrTestCase(BaseTestCase):
mock.call(context=self.context, aggregate=AggregateIdMatcher(aggr),
action='delete', phase='end')])
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_delete_non_empty_aggregate(self, mock_add_host, mock_remove_host):
def test_delete_non_empty_aggregate(self, mock_add_host, mock_remove_host,
mock_get_all_by_host):
# Ensure InvalidAggregateAction is raised when non empty aggregate.
_create_service_entries(self.context,
[['fake_availability_zone', ['fake_host']]])
aggr = self.api.create_aggregate(self.context, 'fake_aggregate',
'fake_availability_zone')
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host='fake_host',
hypervisor_hostname='fake_host')])
self.api.add_host_to_aggregate(self.context, aggr.id, 'fake_host')
self.assertRaises(exception.InvalidAggregateActionDelete,
self.api.delete_aggregate, self.context, aggr.id)
mock_remove_host.assert_not_called()
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
@mock.patch('nova.compute.utils.notify_about_aggregate_action')
@mock.patch.object(availability_zones,
'update_host_availability_zone_cache')
def test_add_host_to_aggregate(self, mock_az, mock_notify, mock_add_host):
def test_add_host_to_aggregate(self, mock_az, mock_notify, mock_add_host,
mock_get_all_by_host):
# Ensure we can add a host to an aggregate.
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
aggr = self.api.create_aggregate(self.context,
'fake_aggregate', fake_zone)
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
def fake_add_aggregate_host(*args, **kwargs):
hosts = kwargs["aggregate"].hosts
@ -11948,12 +12014,18 @@ class ComputeAPIAggrTestCase(BaseTestCase):
mock_add_host.assert_called_once_with(
self.context, aggr.uuid, host_name=fake_host)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_add_host_to_aggr_with_no_az(self, mock_add_host):
def test_add_host_to_aggr_with_no_az(self, mock_add_host,
mock_get_all_by_host):
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
aggr = self.api.create_aggregate(self.context,
'fake_aggregate', fake_zone)
aggr = self.api.add_host_to_aggregate(self.context, aggr.id,
@ -11966,13 +12038,18 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertIn(fake_host, aggr.hosts)
self.assertIn(fake_host, aggr_no_az.hosts)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_add_host_to_multi_az(self, mock_add_host):
def test_add_host_to_multi_az(self, mock_add_host, mock_get_all_by_host):
# Ensure we can't add a host to different availability zone
values = _create_service_entries(self.context)
fake_zone = values[0][0]
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
aggr = self.api.create_aggregate(self.context,
'fake_aggregate', fake_zone)
aggr = self.api.add_host_to_aggregate(self.context,
@ -11986,13 +12063,19 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.context, aggr2.id, fake_host)
self.assertEqual(1, mock_add_host.call_count)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_add_host_to_multi_az_with_nova_agg(self, mock_add_host):
def test_add_host_to_multi_az_with_nova_agg(self, mock_add_host,
mock_get_all_by_host):
# Ensure we can't add a host if already existing in an agg with AZ set
# to default
values = _create_service_entries(self.context)
fake_host = values[0][1][0]
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
aggr = self.api.create_aggregate(self.context,
'fake_aggregate',
CONF.default_availability_zone)
@ -12007,15 +12090,21 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.context, aggr2.id, fake_host)
self.assertEqual(1, mock_add_host.call_count)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_add_host_to_aggregate_multiple(self, mock_add_host):
def test_add_host_to_aggregate_multiple(self, mock_add_host,
mock_get_all_by_host):
# Ensure we can add multiple hosts to an aggregate.
values = _create_service_entries(self.context)
fake_zone = values[0][0]
aggr = self.api.create_aggregate(self.context,
'fake_aggregate', fake_zone)
for host in values[0][1]:
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=host,
hypervisor_hostname=host)])
aggr = self.api.add_host_to_aggregate(self.context,
aggr.id, host)
self.assertEqual(len(aggr.hosts), len(values[0][1]))
@ -12074,6 +12163,7 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.context, aggr.id, 'invalid_host')
mock_add_host.assert_not_called()
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
@ -12082,13 +12172,18 @@ class ComputeAPIAggrTestCase(BaseTestCase):
@mock.patch.object(availability_zones,
'update_host_availability_zone_cache')
def test_remove_host_from_aggregate_active(
self, mock_az, mock_notify, mock_add_host, mock_remove_host):
self, mock_az, mock_notify, mock_add_host, mock_remove_host,
mock_get_all_by_host):
# Ensure we can remove a host from an aggregate.
values = _create_service_entries(self.context)
fake_zone = values[0][0]
aggr = self.api.create_aggregate(self.context,
'fake_aggregate', fake_zone)
for host in values[0][1]:
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=host,
hypervisor_hostname=host)])
aggr = self.api.add_host_to_aggregate(self.context,
aggr.id, host)
host_to_remove = values[0][1][0]
@ -12102,6 +12197,11 @@ class ComputeAPIAggrTestCase(BaseTestCase):
fake_notifier.NOTIFICATIONS = []
mock_notify.reset_mock()
mock_get_all_by_host.reset_mock()
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=host_to_remove,
hypervisor_hostname=host_to_remove)])
expected = self.api.remove_host_from_aggregate(self.context,
aggr.id,
host_to_remove)
@ -12136,6 +12236,7 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.context, aggr.id, 'invalid_host')
mock_remove_host.assert_not_called()
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.objects.Service.get_by_compute_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
@ -12146,10 +12247,14 @@ class ComputeAPIAggrTestCase(BaseTestCase):
'update_host_availability_zone_cache')
def test_remove_host_from_aggregate_no_host_mapping_service_exists(
self, mock_az, mock_notify, mock_add_host, mock_rm_host,
mock_get_service):
mock_get_service, mock_get_all_by_host):
# Ensure ComputeHostNotFound is not raised when adding a host with a
# hostname that doesn't have host mapping but has a service entry.
fake_host = 'fake_host'
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=fake_host,
hypervisor_hostname=fake_host)])
# This is called 4 times, during addition to aggregate for cell0 and
# cell1, and during deletion for cell0 and cell1 as well
mock_get_service.side_effect = [
@ -12213,14 +12318,20 @@ class ComputeAPIAggrTestCase(BaseTestCase):
self.assertEqual('foo_value1', test_agg_meta['foo_key1'])
self.assertEqual('foo_value2', test_agg_meta['foo_key2'])
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_aggregate_list_with_hosts(self, mock_add_host):
def test_aggregate_list_with_hosts(self, mock_add_host,
mock_get_all_by_host):
values = _create_service_entries(self.context)
fake_zone = values[0][0]
host_aggregate = self.api.create_aggregate(self.context,
'fake_aggregate',
fake_zone)
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=values[0][1][0],
hypervisor_hostname=values[0][1][0])])
self.api.add_host_to_aggregate(self.context, host_aggregate.id,
values[0][1][0])
aggregate_list = self.api.get_aggregate_list(self.context)
@ -12292,6 +12403,7 @@ class ComputeAPIAggrCallsSchedulerTestCase(test.NoDBTestCase):
self.api.delete_aggregate(self.context, 1)
delete_aggregate.assert_called_once_with(self.context, agg)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
@mock.patch('nova.compute.utils.notify_about_aggregate_action')
@ -12299,11 +12411,16 @@ class ComputeAPIAggrCallsSchedulerTestCase(test.NoDBTestCase):
@mock.patch('nova.scheduler.client.query.SchedulerQueryClient.'
'update_aggregates')
def test_add_host_to_aggregate(self, update_aggregates, mock_add_agg,
mock_notify, mock_add_host):
mock_notify, mock_add_host,
mock_get_all_by_host):
self.api.is_safe_to_update_az = mock.Mock()
self.api._update_az_cache_for_host = mock.Mock()
agg = objects.Aggregate(name='fake', metadata={}, uuid=uuids.agg)
agg.add_host = mock.Mock()
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host='fakehost',
hypervisor_hostname='fakehost')])
with test.nested(
mock.patch.object(objects.Service, 'get_by_compute_host',
return_value=objects.Service(
@ -12318,6 +12435,7 @@ class ComputeAPIAggrCallsSchedulerTestCase(test.NoDBTestCase):
mock_add_host.assert_called_once_with(
self.context, agg.uuid, host_name='fakehost')
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
@mock.patch('nova.compute.utils.notify_about_aggregate_action')
@ -12326,10 +12444,15 @@ class ComputeAPIAggrCallsSchedulerTestCase(test.NoDBTestCase):
'update_aggregates')
def test_remove_host_from_aggregate(self, update_aggregates,
mock_remove_agg, mock_notify,
mock_remove_host):
mock_remove_host,
mock_get_all_by_host):
self.api._update_az_cache_for_host = mock.Mock()
agg = objects.Aggregate(name='fake', metadata={}, uuid=uuids.agg)
agg.delete_host = mock.Mock()
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host='fakehost',
hypervisor_hostname='fakehost')])
with test.nested(
mock.patch.object(objects.Service, 'get_by_compute_host'),
mock.patch.object(objects.Aggregate, 'get_by_id',

View File

@ -592,12 +592,17 @@ class ComputeAggregateAPITestCase(test.TestCase):
mock_service_get_by_compute_host.return_value = (
objects.Service(host='fake-host'))
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
@mock.patch.object(compute.LOG, 'warning')
def test_aggregate_add_host_placement_missing_provider(
self, mock_log, mock_pc_add_host):
self, mock_log, mock_pc_add_host, mock_get_all_by_host):
hostname = 'fake-host'
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=hostname,
hypervisor_hostname=hostname)])
err = exception.ResourceProviderNotFound(name_or_uuid=hostname)
mock_pc_add_host.side_effect = err
aggregate = self.aggregate_api.create_aggregate(
@ -610,10 +615,16 @@ class ComputeAggregateAPITestCase(test.TestCase):
"nova-manage placement sync_aggregates.")
mock_log.assert_called_with(msg, hostname, err)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_aggregate_add_host_bad_placement(self, mock_pc_add_host):
def test_aggregate_add_host_bad_placement(self, mock_pc_add_host,
mock_get_all_by_host):
hostname = 'fake-host'
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=hostname,
hypervisor_hostname=hostname)])
mock_pc_add_host.side_effect = exception.PlacementAPIConnectFailure
aggregate = self.aggregate_api.create_aggregate(
self.ctxt, 'aggregate', None)
@ -624,12 +635,18 @@ class ComputeAggregateAPITestCase(test.TestCase):
mock_pc_add_host.assert_called_once_with(
self.ctxt, agg_uuid, host_name=hostname)
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.objects.Aggregate.delete_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
def test_aggregate_remove_host_bad_placement(
self, mock_pc_remove_host, mock_agg_obj_delete_host):
self, mock_pc_remove_host, mock_agg_obj_delete_host,
mock_get_all_by_host):
hostname = 'fake-host'
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=hostname,
hypervisor_hostname=hostname)])
mock_pc_remove_host.side_effect = exception.PlacementAPIConnectFailure
aggregate = self.aggregate_api.create_aggregate(
self.ctxt, 'aggregate', None)
@ -643,13 +660,19 @@ class ComputeAggregateAPITestCase(test.TestCase):
# should be tried first and failed with a server failure.
mock_agg_obj_delete_host.assert_not_called()
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.objects.Aggregate.delete_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
@mock.patch.object(compute.LOG, 'warning')
def test_aggregate_remove_host_placement_missing_provider(
self, mock_log, mock_pc_remove_host, mock_agg_obj_delete_host):
self, mock_log, mock_pc_remove_host, mock_agg_obj_delete_host,
mock_get_all_by_host):
hostname = 'fake-host'
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=hostname,
hypervisor_hostname=hostname)])
err = exception.ResourceProviderNotFound(name_or_uuid=hostname)
mock_pc_remove_host.side_effect = err
aggregate = self.aggregate_api.create_aggregate(

View File

@ -2990,12 +2990,13 @@ class XenAPIAggregateTestCase(stubs.XenAPITestBase):
aggregate, 'fake_host')
self.assertIn('aggregate in error', str(ex))
@mock.patch.object(objects.ComputeNodeList, 'get_all_by_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_remove_host')
@mock.patch('nova.scheduler.client.report.SchedulerReportClient.'
'aggregate_add_host')
def test_remove_host_from_aggregate_error(
self, mock_add_host, mock_remove_host):
self, mock_add_host, mock_remove_host, mock_get_all_by_host):
# Ensure we can remove a host from an aggregate even if in error.
values = _create_service_entries(self.context)
fake_zone = list(values.keys())[0]
@ -3008,6 +3009,10 @@ class XenAPIAggregateTestCase(stubs.XenAPITestBase):
aggr.id,
metadata)
for aggregate_host in values[fake_zone]:
mock_get_all_by_host.return_value = objects.ComputeNodeList(
objects=[objects.ComputeNode(
host=aggregate_host,
hypervisor_hostname=aggregate_host)])
aggr = self.api.add_host_to_aggregate(self.context,
aggr.id, aggregate_host)
# let's mock the fact that the aggregate is in error!