147 lines
5.5 KiB
Python
147 lines
5.5 KiB
Python
# Copyright 2011 OpenStack LLC.
|
|
# 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 Query Drivers
|
|
"""
|
|
|
|
import json
|
|
|
|
from nova import exception
|
|
from nova import flags
|
|
from nova import test
|
|
from nova.scheduler import query
|
|
|
|
FLAGS = flags.FLAGS
|
|
|
|
class FakeZoneManager:
|
|
pass
|
|
|
|
class QueryTestCase(test.TestCase):
|
|
"""Test case for query drivers."""
|
|
|
|
def _host_caps(self, multiplier):
|
|
# Returns host capabilities in the following way:
|
|
# host0 = memory:free 10 (100max)
|
|
# disk:available 100 (1000max)
|
|
# hostN = memory:free 10 + 10N
|
|
# disk:available 100 + 100N
|
|
# in other words: hostN has more resources than host0
|
|
# which means ... don't go above 10 hosts.
|
|
return {'host_name-description':'XenServer %s' % multiplier,
|
|
'host_hostname':'xs-%s' % multiplier,
|
|
'host_memory':{'total': 100,
|
|
'overhead': 10,
|
|
'free': 10 + multiplier * 10,
|
|
'free-computed': 10 + multiplier * 10},
|
|
'host_other-config':{},
|
|
'host_ip_address':'192.168.1.%d' % (100 + multiplier),
|
|
'host_cpu_info':{},
|
|
'disk':{'available': 100 + multiplier * 100,
|
|
'total': 1000,
|
|
'used': 0},
|
|
'host_uuid':'xxx-%d' % multiplier,
|
|
'host_name-label':'xs-%s' % multiplier}
|
|
|
|
def setUp(self):
|
|
self.old_flag = FLAGS.default_query_engine
|
|
FLAGS.default_query_engine = 'nova.scheduler.query.AllHostsQuery'
|
|
self.instance_type = dict(name= 'tiny',
|
|
memory_mb= 50,
|
|
vcpus= 10,
|
|
local_gb= 500,
|
|
flavorid= 1,
|
|
swap= 500,
|
|
rxtx_quota= 30000,
|
|
rxtx_cap= 200)
|
|
|
|
self.zone_manager = FakeZoneManager()
|
|
states = {}
|
|
for x in xrange(10):
|
|
states['host%s' % x] = {'compute': self._host_caps(x)}
|
|
self.zone_manager.service_states = states
|
|
|
|
def tearDown(self):
|
|
FLAGS.default_query_engine = self.old_flag
|
|
|
|
def test_choose_driver(self):
|
|
# Test default driver ...
|
|
driver = query.choose_driver()
|
|
self.assertEquals(str(driver), 'nova.scheduler.query.AllHostsQuery')
|
|
# Test valid driver ...
|
|
driver = query.choose_driver('nova.scheduler.query.FlavorQuery')
|
|
self.assertEquals(str(driver), 'nova.scheduler.query.FlavorQuery')
|
|
# Test invalid driver ...
|
|
try:
|
|
query.choose_driver('does not exist')
|
|
self.fail("Should not find driver")
|
|
except exception.SchedulerQueryDriverNotFound:
|
|
pass
|
|
|
|
def test_all_host_driver(self):
|
|
driver = query.AllHostsQuery()
|
|
cooked = driver.instance_type_to_query(self.instance_type)
|
|
hosts = driver.filter_hosts(self.zone_manager, cooked)
|
|
self.assertEquals(10, len(hosts))
|
|
for host, capabilities in hosts:
|
|
self.assertTrue(host.startswith('host'))
|
|
|
|
def test_flavor_driver(self):
|
|
driver = query.FlavorQuery()
|
|
# filter all hosts that can support 50 ram and 500 disk
|
|
name, cooked = driver.instance_type_to_query(self.instance_type)
|
|
self.assertEquals('nova.scheduler.query.FlavorQuery', name)
|
|
hosts = driver.filter_hosts(self.zone_manager, cooked)
|
|
self.assertEquals(6, len(hosts))
|
|
just_hosts = [host for host, caps in hosts]
|
|
just_hosts.sort()
|
|
self.assertEquals('host4', just_hosts[0])
|
|
self.assertEquals('host9', just_hosts[5])
|
|
|
|
def test_json_driver(self):
|
|
driver = query.JsonQuery()
|
|
# filter all hosts that can support 50 ram and 500 disk
|
|
name, cooked = driver.instance_type_to_query(self.instance_type)
|
|
self.assertEquals('nova.scheduler.query.JsonQuery', name)
|
|
hosts = driver.filter_hosts(self.zone_manager, cooked)
|
|
self.assertEquals(6, len(hosts))
|
|
just_hosts = [host for host, caps in hosts]
|
|
just_hosts.sort()
|
|
self.assertEquals('host4', just_hosts[0])
|
|
self.assertEquals('host9', just_hosts[5])
|
|
|
|
# Try some custom queries
|
|
|
|
raw = ['or',
|
|
['and',
|
|
['<', '$compute.host_memory.free', 30],
|
|
['<', '$compute.disk.available', 300]
|
|
],
|
|
['and',
|
|
['>', '$compute.host_memory.free', 70],
|
|
['>', '$compute.disk.available', 700]
|
|
]
|
|
]
|
|
cooked = json.dumps(raw)
|
|
hosts = driver.filter_hosts(self.zone_manager, cooked)
|
|
|
|
self.assertEquals(5, len(hosts))
|
|
just_hosts = [host for host, caps in hosts]
|
|
just_hosts.sort()
|
|
self.assertEquals('host0', just_hosts[0])
|
|
self.assertEquals('host1', just_hosts[1])
|
|
self.assertEquals('host7', just_hosts[2])
|
|
self.assertEquals('host8', just_hosts[3])
|
|
self.assertEquals('host9', just_hosts[4])
|