236 lines
9.6 KiB
Python
236 lines
9.6 KiB
Python
# Copyright 2012 OpenStack Foundation # All Rights Reserved.
|
|
#
|
|
# 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.
|
|
"""
|
|
Tests For Scheduler Host Filters.
|
|
"""
|
|
|
|
import inspect
|
|
|
|
import mock
|
|
from oslo_utils.fixture import uuidsentinel as uuids
|
|
|
|
from nova import filters
|
|
from nova import loadables
|
|
from nova import objects
|
|
from nova import test
|
|
|
|
|
|
class Filter1(filters.BaseFilter):
|
|
"""Test Filter class #1."""
|
|
pass
|
|
|
|
|
|
class Filter2(filters.BaseFilter):
|
|
"""Test Filter class #2."""
|
|
pass
|
|
|
|
|
|
class FiltersTestCase(test.NoDBTestCase):
|
|
|
|
def setUp(self):
|
|
super(FiltersTestCase, self).setUp()
|
|
with mock.patch.object(loadables.BaseLoader, "__init__") as mock_load:
|
|
mock_load.return_value = None
|
|
self.filter_handler = filters.BaseFilterHandler(filters.BaseFilter)
|
|
|
|
@mock.patch('nova.filters.BaseFilter._filter_one')
|
|
def test_filter_all(self, mock_filter_one):
|
|
mock_filter_one.side_effect = [True, False, True]
|
|
filter_obj_list = ['obj1', 'obj2', 'obj3']
|
|
spec_obj = objects.RequestSpec()
|
|
base_filter = filters.BaseFilter()
|
|
|
|
result = base_filter.filter_all(filter_obj_list, spec_obj)
|
|
self.assertTrue(inspect.isgenerator(result))
|
|
self.assertEqual(['obj1', 'obj3'], list(result))
|
|
|
|
@mock.patch('nova.filters.BaseFilter._filter_one')
|
|
def test_filter_all_recursive_yields(self, mock_filter_one):
|
|
# Test filter_all() allows generators from previous filter_all()s.
|
|
# filter_all() yields results. We want to make sure that we can
|
|
# call filter_all() with generators returned from previous calls
|
|
# to filter_all().
|
|
filter_obj_list = ['obj1', 'obj2', 'obj3']
|
|
spec_obj = objects.RequestSpec()
|
|
base_filter = filters.BaseFilter()
|
|
|
|
# The order that _filter_one is going to get called gets
|
|
# confusing because we will be recursively yielding things..
|
|
# We are going to simulate the first call to filter_all()
|
|
# returning False for 'obj2'. So, 'obj1' will get yielded
|
|
# 'total_iterations' number of times before the first filter_all()
|
|
# call gets to processing 'obj2'. We then return 'False' for it.
|
|
# After that, 'obj3' gets yielded 'total_iterations' number of
|
|
# times.
|
|
mock_results = []
|
|
total_iterations = 200
|
|
for x in range(total_iterations):
|
|
mock_results.append(True)
|
|
mock_results.append(False)
|
|
for x in range(total_iterations):
|
|
mock_results.append(True)
|
|
mock_filter_one.side_effect = mock_results
|
|
|
|
objs = iter(filter_obj_list)
|
|
for x in range(total_iterations):
|
|
# Pass in generators returned from previous calls.
|
|
objs = base_filter.filter_all(objs, spec_obj)
|
|
self.assertTrue(inspect.isgenerator(objs))
|
|
self.assertEqual(['obj1', 'obj3'], list(objs))
|
|
|
|
def test_get_filtered_objects(self):
|
|
filter_objs_initial = ['initial', 'filter1', 'objects1']
|
|
filter_objs_second = ['second', 'filter2', 'objects2']
|
|
filter_objs_last = ['last', 'filter3', 'objects3']
|
|
spec_obj = objects.RequestSpec()
|
|
|
|
def _fake_base_loader_init(*args, **kwargs):
|
|
pass
|
|
|
|
self.stub_out('nova.loadables.BaseLoader.__init__',
|
|
_fake_base_loader_init)
|
|
|
|
filt1_mock = mock.Mock(Filter1)
|
|
filt1_mock.run_filter_for_index.return_value = True
|
|
filt1_mock.filter_all.return_value = filter_objs_second
|
|
filt2_mock = mock.Mock(Filter2)
|
|
filt2_mock.run_filter_for_index.return_value = True
|
|
filt2_mock.filter_all.return_value = filter_objs_last
|
|
|
|
filter_handler = filters.BaseFilterHandler(filters.BaseFilter)
|
|
filter_mocks = [filt1_mock, filt2_mock]
|
|
result = filter_handler.get_filtered_objects(filter_mocks,
|
|
filter_objs_initial,
|
|
spec_obj)
|
|
self.assertEqual(filter_objs_last, result)
|
|
filt1_mock.filter_all.assert_called_once_with(filter_objs_initial,
|
|
spec_obj)
|
|
filt2_mock.filter_all.assert_called_once_with(filter_objs_second,
|
|
spec_obj)
|
|
|
|
def test_get_filtered_objects_for_index(self):
|
|
"""Test that we don't call a filter when its
|
|
run_filter_for_index() method returns false
|
|
"""
|
|
filter_objs_initial = ['initial', 'filter1', 'objects1']
|
|
filter_objs_second = ['second', 'filter2', 'objects2']
|
|
spec_obj = objects.RequestSpec()
|
|
|
|
def _fake_base_loader_init(*args, **kwargs):
|
|
pass
|
|
|
|
self.stub_out('nova.loadables.BaseLoader.__init__',
|
|
_fake_base_loader_init)
|
|
|
|
filt1_mock = mock.Mock(Filter1)
|
|
filt1_mock.run_filter_for_index.return_value = True
|
|
filt1_mock.filter_all.return_value = filter_objs_second
|
|
filt2_mock = mock.Mock(Filter2)
|
|
filt2_mock.run_filter_for_index.return_value = False
|
|
|
|
filter_handler = filters.BaseFilterHandler(filters.BaseFilter)
|
|
filter_mocks = [filt1_mock, filt2_mock]
|
|
result = filter_handler.get_filtered_objects(filter_mocks,
|
|
filter_objs_initial,
|
|
spec_obj)
|
|
self.assertEqual(filter_objs_second, result)
|
|
filt1_mock.filter_all.assert_called_once_with(filter_objs_initial,
|
|
spec_obj)
|
|
filt2_mock.filter_all.assert_not_called()
|
|
|
|
def test_get_filtered_objects_none_response(self):
|
|
filter_objs_initial = ['initial', 'filter1', 'objects1']
|
|
spec_obj = objects.RequestSpec()
|
|
|
|
def _fake_base_loader_init(*args, **kwargs):
|
|
pass
|
|
|
|
self.stub_out('nova.loadables.BaseLoader.__init__',
|
|
_fake_base_loader_init)
|
|
|
|
filt1_mock = mock.Mock(Filter1)
|
|
filt1_mock.run_filter_for_index.return_value = True
|
|
filt1_mock.filter_all.return_value = None
|
|
filt2_mock = mock.Mock(Filter2)
|
|
|
|
filter_handler = filters.BaseFilterHandler(filters.BaseFilter)
|
|
filter_mocks = [filt1_mock, filt2_mock]
|
|
result = filter_handler.get_filtered_objects(filter_mocks,
|
|
filter_objs_initial,
|
|
spec_obj)
|
|
self.assertIsNone(result)
|
|
filt1_mock.filter_all.assert_called_once_with(filter_objs_initial,
|
|
spec_obj)
|
|
filt2_mock.filter_all.assert_not_called()
|
|
|
|
def test_get_filtered_objects_info_log_none_returned(self):
|
|
LOG = filters.LOG
|
|
|
|
class FilterA(filters.BaseFilter):
|
|
def filter_all(self, list_objs, spec_obj):
|
|
# return all but the first object
|
|
return list_objs[1:]
|
|
|
|
class FilterB(filters.BaseFilter):
|
|
def filter_all(self, list_objs, spec_obj):
|
|
# return an empty list
|
|
return []
|
|
|
|
filter_a = FilterA()
|
|
filter_b = FilterB()
|
|
all_filters = [filter_a, filter_b]
|
|
hosts = ["Host0", "Host1", "Host2"]
|
|
fake_uuid = uuids.instance
|
|
spec_obj = objects.RequestSpec(instance_uuid=fake_uuid)
|
|
with mock.patch.object(LOG, "info") as mock_log:
|
|
result = self.filter_handler.get_filtered_objects(
|
|
all_filters, hosts, spec_obj)
|
|
self.assertFalse(result)
|
|
# FilterA should leave Host1 and Host2; FilterB should leave None.
|
|
exp_output = ("['FilterA: (start: 3, end: 2)', "
|
|
"'FilterB: (start: 2, end: 0)']")
|
|
cargs = mock_log.call_args[0][0]
|
|
self.assertIn("with instance ID '%s'" % fake_uuid, cargs)
|
|
self.assertIn(exp_output, cargs)
|
|
|
|
def test_get_filtered_objects_debug_log_none_returned(self):
|
|
LOG = filters.LOG
|
|
|
|
class FilterA(filters.BaseFilter):
|
|
def filter_all(self, list_objs, spec_obj):
|
|
# return all but the first object
|
|
return list_objs[1:]
|
|
|
|
class FilterB(filters.BaseFilter):
|
|
def filter_all(self, list_objs, spec_obj):
|
|
# return an empty list
|
|
return []
|
|
|
|
filter_a = FilterA()
|
|
filter_b = FilterB()
|
|
all_filters = [filter_a, filter_b]
|
|
hosts = ["Host0", "Host1", "Host2"]
|
|
fake_uuid = uuids.instance
|
|
spec_obj = objects.RequestSpec(instance_uuid=fake_uuid)
|
|
with mock.patch.object(LOG, "debug") as mock_log:
|
|
result = self.filter_handler.get_filtered_objects(
|
|
all_filters, hosts, spec_obj)
|
|
self.assertFalse(result)
|
|
# FilterA should leave Host1 and Host2; FilterB should leave None.
|
|
exp_output = ("[('FilterA', [('Host1', ''), ('Host2', '')]), " +
|
|
"('FilterB', None)]")
|
|
cargs = mock_log.call_args[0][0]
|
|
self.assertIn("with instance ID '%s'" % fake_uuid, cargs)
|
|
self.assertIn(exp_output, cargs)
|