diff options
Diffstat (limited to 'nova/tests/unit/compute/test_host_api.py')
-rw-r--r-- | nova/tests/unit/compute/test_host_api.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/nova/tests/unit/compute/test_host_api.py b/nova/tests/unit/compute/test_host_api.py index e9b364c63b..48e2dd9747 100644 --- a/nova/tests/unit/compute/test_host_api.py +++ b/nova/tests/unit/compute/test_host_api.py @@ -721,8 +721,12 @@ class ComputeAggregateAPITestCase(test.TestCase): fixtures.MockPatch('nova.objects.HostMapping.get_by_host')) self.useFixture( fixtures.MockPatch('nova.context.set_target_cell')) - self.useFixture( - fixtures.MockPatch('nova.objects.Service.get_by_compute_host')) + mock_service_get_by_compute_host = ( + self.useFixture( + fixtures.MockPatch( + 'nova.objects.Service.get_by_compute_host')).mock) + mock_service_get_by_compute_host.return_value = ( + objects.Service(host='fake-host')) @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'aggregate_add_host') |