9d2ce8f1fe
Tested methods consume_requests() and consume_requests() operate with InstanceNUMACell objects but because of mistake NUMACell objects were used instead. It may cause the problem if these methods will try to use some field which is not a field of NUMACell but is a field of InstanceNUMACell. Also some indents were removed from assertTrue/False Change-Id: Ib41c9b22bfd22e769fac729f09d95d6bc8ba8e6b
448 lines
19 KiB
Python
448 lines
19 KiB
Python
# Copyright (c) 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.
|
|
|
|
import mock
|
|
from oslo_config import cfg
|
|
import six
|
|
|
|
from nova import exception
|
|
from nova import objects
|
|
from nova.objects import fields
|
|
from nova.pci import stats
|
|
from nova.pci import whitelist
|
|
from nova import test
|
|
from nova.tests.unit.pci import fakes
|
|
|
|
CONF = cfg.CONF
|
|
fake_pci_1 = {
|
|
'compute_node_id': 1,
|
|
'address': '0000:00:00.1',
|
|
'product_id': 'p1',
|
|
'vendor_id': 'v1',
|
|
'status': 'available',
|
|
'extra_k1': 'v1',
|
|
'request_id': None,
|
|
'numa_node': 0,
|
|
'dev_type': fields.PciDeviceType.STANDARD,
|
|
'parent_addr': None,
|
|
}
|
|
|
|
|
|
fake_pci_2 = dict(fake_pci_1, vendor_id='v2',
|
|
product_id='p2',
|
|
address='0000:00:00.2',
|
|
numa_node=1)
|
|
|
|
|
|
fake_pci_3 = dict(fake_pci_1, address='0000:00:00.3')
|
|
|
|
fake_pci_4 = dict(fake_pci_1, vendor_id='v3',
|
|
product_id='p3',
|
|
address='0000:00:00.3',
|
|
numa_node= None)
|
|
|
|
pci_requests = [objects.InstancePCIRequest(count=1,
|
|
spec=[{'vendor_id': 'v1'}]),
|
|
objects.InstancePCIRequest(count=1,
|
|
spec=[{'vendor_id': 'v2'}])]
|
|
|
|
|
|
pci_requests_multiple = [objects.InstancePCIRequest(count=1,
|
|
spec=[{'vendor_id': 'v1'}]),
|
|
objects.InstancePCIRequest(count=3,
|
|
spec=[{'vendor_id': 'v2'}])]
|
|
|
|
|
|
class PciDeviceStatsTestCase(test.NoDBTestCase):
|
|
def _create_fake_devs(self):
|
|
self.fake_dev_1 = objects.PciDevice.create(None, fake_pci_1)
|
|
self.fake_dev_2 = objects.PciDevice.create(None, fake_pci_2)
|
|
self.fake_dev_3 = objects.PciDevice.create(None, fake_pci_3)
|
|
self.fake_dev_4 = objects.PciDevice.create(None, fake_pci_4)
|
|
|
|
for dev in [self.fake_dev_1, self.fake_dev_2,
|
|
self.fake_dev_3, self.fake_dev_4]:
|
|
self.pci_stats.add_device(dev)
|
|
|
|
def setUp(self):
|
|
super(PciDeviceStatsTestCase, self).setUp()
|
|
self.pci_stats = stats.PciDeviceStats()
|
|
# The following two calls need to be made before adding the devices.
|
|
patcher = fakes.fake_pci_whitelist()
|
|
self.addCleanup(patcher.stop)
|
|
self._create_fake_devs()
|
|
|
|
def test_add_device(self):
|
|
self.assertEqual(len(self.pci_stats.pools), 3)
|
|
self.assertEqual(set([d['vendor_id'] for d in self.pci_stats]),
|
|
set(['v1', 'v2', 'v3']))
|
|
self.assertEqual(set([d['count'] for d in self.pci_stats]),
|
|
set([1, 2]))
|
|
|
|
def test_remove_device(self):
|
|
self.pci_stats.remove_device(self.fake_dev_2)
|
|
self.assertEqual(len(self.pci_stats.pools), 2)
|
|
self.assertEqual(self.pci_stats.pools[0]['count'], 2)
|
|
self.assertEqual(self.pci_stats.pools[0]['vendor_id'], 'v1')
|
|
|
|
def test_remove_device_exception(self):
|
|
self.pci_stats.remove_device(self.fake_dev_2)
|
|
self.assertRaises(exception.PciDevicePoolEmpty,
|
|
self.pci_stats.remove_device,
|
|
self.fake_dev_2)
|
|
|
|
def test_pci_stats_equivalent(self):
|
|
pci_stats2 = stats.PciDeviceStats()
|
|
for dev in [self.fake_dev_1,
|
|
self.fake_dev_2,
|
|
self.fake_dev_3,
|
|
self.fake_dev_4]:
|
|
pci_stats2.add_device(dev)
|
|
self.assertEqual(self.pci_stats, pci_stats2)
|
|
|
|
def test_pci_stats_not_equivalent(self):
|
|
pci_stats2 = stats.PciDeviceStats()
|
|
for dev in [self.fake_dev_1,
|
|
self.fake_dev_2,
|
|
self.fake_dev_3]:
|
|
pci_stats2.add_device(dev)
|
|
self.assertNotEqual(self.pci_stats, pci_stats2)
|
|
|
|
def test_object_create(self):
|
|
m = self.pci_stats.to_device_pools_obj()
|
|
new_stats = stats.PciDeviceStats(m)
|
|
|
|
self.assertEqual(len(new_stats.pools), 3)
|
|
self.assertEqual(set([d['count'] for d in new_stats]),
|
|
set([1, 2]))
|
|
self.assertEqual(set([d['vendor_id'] for d in new_stats]),
|
|
set(['v1', 'v2', 'v3']))
|
|
|
|
def test_support_requests(self):
|
|
self.assertTrue(self.pci_stats.support_requests(pci_requests))
|
|
self.assertEqual(len(self.pci_stats.pools), 3)
|
|
self.assertEqual(set([d['count'] for d in self.pci_stats]),
|
|
set((1, 2)))
|
|
|
|
def test_support_requests_failed(self):
|
|
self.assertFalse(
|
|
self.pci_stats.support_requests(pci_requests_multiple))
|
|
self.assertEqual(len(self.pci_stats.pools), 3)
|
|
self.assertEqual(set([d['count'] for d in self.pci_stats]),
|
|
set([1, 2]))
|
|
|
|
def test_apply_requests(self):
|
|
self.pci_stats.apply_requests(pci_requests)
|
|
self.assertEqual(len(self.pci_stats.pools), 2)
|
|
self.assertEqual(self.pci_stats.pools[0]['vendor_id'], 'v1')
|
|
self.assertEqual(self.pci_stats.pools[0]['count'], 1)
|
|
|
|
def test_apply_requests_failed(self):
|
|
self.assertRaises(exception.PciDeviceRequestFailed,
|
|
self.pci_stats.apply_requests,
|
|
pci_requests_multiple)
|
|
|
|
def test_consume_requests(self):
|
|
devs = self.pci_stats.consume_requests(pci_requests)
|
|
self.assertEqual(2, len(devs))
|
|
self.assertEqual(set(['v1', 'v2']),
|
|
set([dev.vendor_id for dev in devs]))
|
|
|
|
def test_consume_requests_empty(self):
|
|
devs = self.pci_stats.consume_requests([])
|
|
self.assertEqual(0, len(devs))
|
|
|
|
def test_consume_requests_failed(self):
|
|
self.assertIsNone(self.pci_stats.consume_requests(
|
|
pci_requests_multiple))
|
|
|
|
def test_support_requests_numa(self):
|
|
cells = [objects.InstanceNUMACell(id=0, cpuset=set(), memory=0),
|
|
objects.InstanceNUMACell(id=1, cpuset=set(), memory=0)]
|
|
self.assertTrue(self.pci_stats.support_requests(pci_requests, cells))
|
|
|
|
def test_support_requests_numa_failed(self):
|
|
cells = [objects.InstanceNUMACell(id=0, cpuset=set(), memory=0)]
|
|
self.assertFalse(self.pci_stats.support_requests(pci_requests, cells))
|
|
|
|
def test_support_requests_no_numa_info(self):
|
|
cells = [objects.InstanceNUMACell(id=0, cpuset=set(), memory=0)]
|
|
pci_request = [objects.InstancePCIRequest(count=1,
|
|
spec=[{'vendor_id': 'v3'}])]
|
|
self.assertTrue(self.pci_stats.support_requests(pci_request, cells))
|
|
|
|
def test_consume_requests_numa(self):
|
|
cells = [objects.InstanceNUMACell(id=0, cpuset=set(), memory=0),
|
|
objects.InstanceNUMACell(id=1, cpuset=set(), memory=0)]
|
|
devs = self.pci_stats.consume_requests(pci_requests, cells)
|
|
self.assertEqual(2, len(devs))
|
|
self.assertEqual(set(['v1', 'v2']),
|
|
set([dev.vendor_id for dev in devs]))
|
|
|
|
def test_consume_requests_numa_failed(self):
|
|
cells = [objects.InstanceNUMACell(id=0, cpuset=set(), memory=0)]
|
|
self.assertIsNone(self.pci_stats.consume_requests(pci_requests, cells))
|
|
|
|
def test_consume_requests_no_numa_info(self):
|
|
cells = [objects.InstanceNUMACell(id=0, cpuset=set(), memory=0)]
|
|
pci_request = [objects.InstancePCIRequest(count=1,
|
|
spec=[{'vendor_id': 'v3'}])]
|
|
devs = self.pci_stats.consume_requests(pci_request, cells)
|
|
self.assertEqual(1, len(devs))
|
|
self.assertEqual(set(['v3']),
|
|
set([dev.vendor_id for dev in devs]))
|
|
|
|
@mock.patch(
|
|
'nova.pci.whitelist.Whitelist._parse_white_list_from_config')
|
|
def test_white_list_parsing(self, mock_whitelist_parse):
|
|
white_list = '{"product_id":"0001", "vendor_id":"8086"}'
|
|
CONF.set_override('passthrough_whitelist', white_list, 'pci')
|
|
pci_stats = stats.PciDeviceStats()
|
|
pci_stats.add_device(self.fake_dev_2)
|
|
pci_stats.remove_device(self.fake_dev_2)
|
|
self.assertEqual(1, mock_whitelist_parse.call_count)
|
|
|
|
|
|
class PciDeviceStatsWithTagsTestCase(test.NoDBTestCase):
|
|
|
|
def setUp(self):
|
|
super(PciDeviceStatsWithTagsTestCase, self).setUp()
|
|
white_list = ['{"vendor_id":"1137","product_id":"0071",'
|
|
'"address":"*:0a:00.*","physical_network":"physnet1"}',
|
|
'{"vendor_id":"1137","product_id":"0072"}']
|
|
self.flags(passthrough_whitelist=white_list, group='pci')
|
|
dev_filter = whitelist.Whitelist(white_list)
|
|
self.pci_stats = stats.PciDeviceStats(dev_filter=dev_filter)
|
|
|
|
def _create_pci_devices(self):
|
|
self.pci_tagged_devices = []
|
|
for dev in range(4):
|
|
pci_dev = {'compute_node_id': 1,
|
|
'address': '0000:0a:00.%d' % dev,
|
|
'vendor_id': '1137',
|
|
'product_id': '0071',
|
|
'status': 'available',
|
|
'request_id': None,
|
|
'dev_type': 'type-PCI',
|
|
'parent_addr': None,
|
|
'numa_node': 0}
|
|
self.pci_tagged_devices.append(objects.PciDevice.create(None,
|
|
pci_dev))
|
|
|
|
self.pci_untagged_devices = []
|
|
for dev in range(3):
|
|
pci_dev = {'compute_node_id': 1,
|
|
'address': '0000:0b:00.%d' % dev,
|
|
'vendor_id': '1137',
|
|
'product_id': '0072',
|
|
'status': 'available',
|
|
'request_id': None,
|
|
'dev_type': 'type-PCI',
|
|
'parent_addr': None,
|
|
'numa_node': 0}
|
|
self.pci_untagged_devices.append(objects.PciDevice.create(None,
|
|
pci_dev))
|
|
|
|
for dev in self.pci_tagged_devices:
|
|
self.pci_stats.add_device(dev)
|
|
|
|
for dev in self.pci_untagged_devices:
|
|
self.pci_stats.add_device(dev)
|
|
|
|
def _assertPoolContent(self, pool, vendor_id, product_id, count, **tags):
|
|
self.assertEqual(vendor_id, pool['vendor_id'])
|
|
self.assertEqual(product_id, pool['product_id'])
|
|
self.assertEqual(count, pool['count'])
|
|
if tags:
|
|
for k, v in six.iteritems(tags):
|
|
self.assertEqual(v, pool[k])
|
|
|
|
def _assertPools(self):
|
|
# Pools are ordered based on the number of keys. 'product_id',
|
|
# 'vendor_id' are always part of the keys. When tags are present,
|
|
# they are also part of the keys. In this test class, we have
|
|
# two pools with the second one having the tag 'physical_network'
|
|
# and the value 'physnet1'
|
|
self.assertEqual(2, len(self.pci_stats.pools))
|
|
self._assertPoolContent(self.pci_stats.pools[0], '1137', '0072',
|
|
len(self.pci_untagged_devices))
|
|
self.assertEqual(self.pci_untagged_devices,
|
|
self.pci_stats.pools[0]['devices'])
|
|
self._assertPoolContent(self.pci_stats.pools[1], '1137', '0071',
|
|
len(self.pci_tagged_devices),
|
|
physical_network='physnet1')
|
|
self.assertEqual(self.pci_tagged_devices,
|
|
self.pci_stats.pools[1]['devices'])
|
|
|
|
def test_add_devices(self):
|
|
self._create_pci_devices()
|
|
self._assertPools()
|
|
|
|
def test_consume_reqeusts(self):
|
|
self._create_pci_devices()
|
|
pci_requests = [objects.InstancePCIRequest(count=1,
|
|
spec=[{'physical_network': 'physnet1'}]),
|
|
objects.InstancePCIRequest(count=1,
|
|
spec=[{'vendor_id': '1137',
|
|
'product_id': '0072'}])]
|
|
devs = self.pci_stats.consume_requests(pci_requests)
|
|
self.assertEqual(2, len(devs))
|
|
self.assertEqual(set(['0071', '0072']),
|
|
set([dev.product_id for dev in devs]))
|
|
self._assertPoolContent(self.pci_stats.pools[0], '1137', '0072', 2)
|
|
self._assertPoolContent(self.pci_stats.pools[1], '1137', '0071', 3,
|
|
physical_network='physnet1')
|
|
|
|
def test_add_device_no_devspec(self):
|
|
self._create_pci_devices()
|
|
pci_dev = {'compute_node_id': 1,
|
|
'address': '0000:0c:00.1',
|
|
'vendor_id': '2345',
|
|
'product_id': '0172',
|
|
'status': 'available',
|
|
'parent_addr': None,
|
|
'request_id': None}
|
|
pci_dev_obj = objects.PciDevice.create(None, pci_dev)
|
|
self.pci_stats.add_device(pci_dev_obj)
|
|
# There should be no change
|
|
self.assertIsNone(
|
|
self.pci_stats._create_pool_keys_from_dev(pci_dev_obj))
|
|
self._assertPools()
|
|
|
|
def test_remove_device_no_devspec(self):
|
|
self._create_pci_devices()
|
|
pci_dev = {'compute_node_id': 1,
|
|
'address': '0000:0c:00.1',
|
|
'vendor_id': '2345',
|
|
'product_id': '0172',
|
|
'status': 'available',
|
|
'parent_addr': None,
|
|
'request_id': None}
|
|
pci_dev_obj = objects.PciDevice.create(None, pci_dev)
|
|
self.pci_stats.remove_device(pci_dev_obj)
|
|
# There should be no change
|
|
self.assertIsNone(
|
|
self.pci_stats._create_pool_keys_from_dev(pci_dev_obj))
|
|
self._assertPools()
|
|
|
|
def test_remove_device(self):
|
|
self._create_pci_devices()
|
|
dev1 = self.pci_untagged_devices.pop()
|
|
self.pci_stats.remove_device(dev1)
|
|
dev2 = self.pci_tagged_devices.pop()
|
|
self.pci_stats.remove_device(dev2)
|
|
self._assertPools()
|
|
|
|
|
|
class PciDeviceVFPFStatsTestCase(test.NoDBTestCase):
|
|
|
|
def setUp(self):
|
|
super(PciDeviceVFPFStatsTestCase, self).setUp()
|
|
white_list = ['{"vendor_id":"8086","product_id":"1528"}',
|
|
'{"vendor_id":"8086","product_id":"1515"}']
|
|
self.flags(passthrough_whitelist=white_list, group='pci')
|
|
self.pci_stats = stats.PciDeviceStats()
|
|
|
|
def _create_pci_devices(self, vf_product_id=1515, pf_product_id=1528):
|
|
self.sriov_pf_devices = []
|
|
for dev in range(2):
|
|
pci_dev = {'compute_node_id': 1,
|
|
'address': '0000:81:00.%d' % dev,
|
|
'vendor_id': '8086',
|
|
'product_id': '%d' % pf_product_id,
|
|
'status': 'available',
|
|
'request_id': None,
|
|
'dev_type': fields.PciDeviceType.SRIOV_PF,
|
|
'parent_addr': None,
|
|
'numa_node': 0}
|
|
dev_obj = objects.PciDevice.create(None, pci_dev)
|
|
dev_obj.child_devices = []
|
|
self.sriov_pf_devices.append(dev_obj)
|
|
|
|
self.sriov_vf_devices = []
|
|
for dev in range(8):
|
|
pci_dev = {'compute_node_id': 1,
|
|
'address': '0000:81:10.%d' % dev,
|
|
'vendor_id': '8086',
|
|
'product_id': '%d' % vf_product_id,
|
|
'status': 'available',
|
|
'request_id': None,
|
|
'dev_type': fields.PciDeviceType.SRIOV_VF,
|
|
'parent_addr': '0000:81:00.%d' % int(dev / 4),
|
|
'numa_node': 0}
|
|
dev_obj = objects.PciDevice.create(None, pci_dev)
|
|
dev_obj.parent_device = self.sriov_pf_devices[int(dev / 4)]
|
|
dev_obj.parent_device.child_devices.append(dev_obj)
|
|
self.sriov_vf_devices.append(dev_obj)
|
|
|
|
list(map(self.pci_stats.add_device, self.sriov_pf_devices))
|
|
list(map(self.pci_stats.add_device, self.sriov_vf_devices))
|
|
|
|
def test_consume_VF_requests(self):
|
|
self._create_pci_devices()
|
|
pci_requests = [objects.InstancePCIRequest(count=2,
|
|
spec=[{'product_id': '1515'}])]
|
|
devs = self.pci_stats.consume_requests(pci_requests)
|
|
self.assertEqual(2, len(devs))
|
|
self.assertEqual(set(['1515']),
|
|
set([dev.product_id for dev in devs]))
|
|
free_devs = self.pci_stats.get_free_devs()
|
|
# Validate that the parents of these VFs has been removed
|
|
# from pools.
|
|
for dev in devs:
|
|
self.assertNotIn(dev.parent_addr,
|
|
[free_dev.address for free_dev in free_devs])
|
|
|
|
def test_consume_PF_requests(self):
|
|
self._create_pci_devices()
|
|
pci_requests = [objects.InstancePCIRequest(count=2,
|
|
spec=[{'product_id': '1528',
|
|
'dev_type': 'type-PF'}])]
|
|
devs = self.pci_stats.consume_requests(pci_requests)
|
|
self.assertEqual(2, len(devs))
|
|
self.assertEqual(set(['1528']),
|
|
set([dev.product_id for dev in devs]))
|
|
free_devs = self.pci_stats.get_free_devs()
|
|
# Validate that there are no free devices left, as when allocating
|
|
# both available PFs, its VFs should not be available.
|
|
self.assertEqual(0, len(free_devs))
|
|
|
|
def test_consume_VF_and_PF_requests(self):
|
|
self._create_pci_devices()
|
|
pci_requests = [objects.InstancePCIRequest(count=2,
|
|
spec=[{'product_id': '1515'}]),
|
|
objects.InstancePCIRequest(count=1,
|
|
spec=[{'product_id': '1528',
|
|
'dev_type': 'type-PF'}])]
|
|
devs = self.pci_stats.consume_requests(pci_requests)
|
|
self.assertEqual(3, len(devs))
|
|
self.assertEqual(set(['1528', '1515']),
|
|
set([dev.product_id for dev in devs]))
|
|
|
|
def test_consume_VF_and_PF_requests_failed(self):
|
|
self._create_pci_devices()
|
|
pci_requests = [objects.InstancePCIRequest(count=5,
|
|
spec=[{'product_id': '1515'}]),
|
|
objects.InstancePCIRequest(count=1,
|
|
spec=[{'product_id': '1528',
|
|
'dev_type': 'type-PF'}])]
|
|
self.assertIsNone(self.pci_stats.consume_requests(pci_requests))
|
|
|
|
def test_consume_VF_and_PF_same_prodict_id_failed(self):
|
|
self._create_pci_devices(pf_product_id=1515)
|
|
pci_requests = [objects.InstancePCIRequest(count=9,
|
|
spec=[{'product_id': '1515'}])]
|
|
self.assertIsNone(self.pci_stats.consume_requests(pci_requests))
|