diff options
Diffstat (limited to 'nova/tests/unit/compute/test_compute_mgr.py')
-rw-r--r-- | nova/tests/unit/compute/test_compute_mgr.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/nova/tests/unit/compute/test_compute_mgr.py b/nova/tests/unit/compute/test_compute_mgr.py index 2d3d8bc7c4..f5b1df0664 100644 --- a/nova/tests/unit/compute/test_compute_mgr.py +++ b/nova/tests/unit/compute/test_compute_mgr.py @@ -212,6 +212,7 @@ class ComputeManagerUnitTestCase(test.NoDBTestCase): self.assertTrue(log_mock.info.called) self.assertIsNone(self.compute._resource_tracker) + @mock.patch.object(manager.ComputeManager, '_get_resource_tracker') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'delete_resource_provider') @mock.patch.object(manager.ComputeManager, @@ -219,7 +220,7 @@ class ComputeManagerUnitTestCase(test.NoDBTestCase): @mock.patch.object(fake_driver.FakeDriver, 'get_available_nodes') @mock.patch.object(manager.ComputeManager, '_get_compute_nodes_in_db') def test_update_available_resource(self, get_db_nodes, get_avail_nodes, - update_mock, del_rp_mock): + update_mock, del_rp_mock, mock_get_rt): db_nodes = [self._make_compute_node('node%s' % i, i) for i in range(1, 5)] avail_nodes = set(['node2', 'node3', 'node4', 'node5']) @@ -239,6 +240,8 @@ class ComputeManagerUnitTestCase(test.NoDBTestCase): db_node.destroy.assert_called_once_with() del_rp_mock.assert_called_once_with(self.context, db_node, cascade=True) + mock_get_rt.return_value.remove_node.assert_called_once_with( + 'node1') else: self.assertFalse(db_node.destroy.called) |