segments: fix rp inventory update

The patch https://review.opendev.org/663980 made resource provider
inventory update failing with the assumption that inventory update
expects a dict with a key of the resource class, like resource provider
inventories update.
See the placement API-ref:
https://developer.openstack.org/api-ref/placement/#update-resource-provider-inventory
https://developer.openstack.org/api-ref/placement/#update-resource-provider-inventories

Change-Id: I7de1a947b864eb5ac57ebaca895f827d2e667443
Closes-Bug: #1836037
Related-Bug: #1828543
This commit is contained in:
elajkat 2019-07-10 17:20:22 +02:00
parent 28721b4c8e
commit 12ab7c4cb1
2 changed files with 23 additions and 29 deletions

View File

@ -217,10 +217,9 @@ class NovaSegmentNotifier(object):
ipv4_inventory = self.p_client.get_inventory(event.segment_id, ipv4_inventory = self.p_client.get_inventory(event.segment_id,
IPV4_RESOURCE_CLASS) IPV4_RESOURCE_CLASS)
if event.total: if event.total:
ipv4_inventory[IPV4_RESOURCE_CLASS]['total'] += event.total ipv4_inventory['total'] += event.total
if event.reserved: if event.reserved:
ipv4_inventory[IPV4_RESOURCE_CLASS]['reserved'] += \ ipv4_inventory['reserved'] += event.reserved
event.reserved
try: try:
self.p_client.update_resource_provider_inventory( self.p_client.update_resource_provider_inventory(
event.segment_id, ipv4_inventory, IPV4_RESOURCE_CLASS) event.segment_id, ipv4_inventory, IPV4_RESOURCE_CLASS)

View File

@ -1714,17 +1714,17 @@ class TestNovaSegmentNotifier(SegmentAwareIpamTestCase):
total, reserved = self._calculate_inventory_total_and_reserved( total, reserved = self._calculate_inventory_total_and_reserved(
subnet['subnet']) subnet['subnet'])
inventory, _ = self._get_inventory(total, reserved) inventory, _ = self._get_inventory(total, reserved)
inventory_dict = inventory[seg_plugin.IPV4_RESOURCE_CLASS] ipv4_classname = seg_plugin.IPV4_RESOURCE_CLASS
self.mock_p_client.update_resource_provider_inventories.\ self.mock_p_client.update_resource_provider_inventories.\
assert_called_with(segment_id, inventory) assert_called_with(segment_id, {ipv4_classname: inventory})
self.assertEqual( self.assertEqual(
inventory_dict['total'], inventory['total'],
self.mock_p_client.update_resource_provider_inventories. self.mock_p_client.update_resource_provider_inventories.
call_args[0][1][seg_plugin.IPV4_RESOURCE_CLASS]['total']) call_args[0][1][ipv4_classname]['total'])
self.assertEqual( self.assertEqual(
inventory_dict['reserved'], inventory['reserved'],
self.mock_p_client.update_resource_provider_inventories. self.mock_p_client.update_resource_provider_inventories.
call_args[0][1][seg_plugin.IPV4_RESOURCE_CLASS]['reserved']) call_args[0][1][ipv4_classname]['reserved'])
self.mock_p_client.reset_mock() self.mock_p_client.reset_mock()
self.mock_p_client.get_inventory.side_effect = None self.mock_p_client.get_inventory.side_effect = None
self.mock_n_client.reset_mock() self.mock_n_client.reset_mock()
@ -1789,29 +1789,25 @@ class TestNovaSegmentNotifier(SegmentAwareIpamTestCase):
if subnet: if subnet:
total, reserved = self._calculate_inventory_total_and_reserved( total, reserved = self._calculate_inventory_total_and_reserved(
subnet) subnet)
ipv4_classname = seg_plugin.IPV4_RESOURCE_CLASS inventory['total'] += total - original_total
inventory[ipv4_classname]['total'] += total - original_total inventory['reserved'] += reserved - original_reserved
inventory[ipv4_classname]['reserved'] += reserved - original_reserved
self.mock_p_client.update_resource_provider_inventory.\ self.mock_p_client.update_resource_provider_inventory.\
assert_called_with(segment_id, inventory, assert_called_with(segment_id, inventory,
seg_plugin.IPV4_RESOURCE_CLASS) seg_plugin.IPV4_RESOURCE_CLASS)
self.assertEqual( self.assertEqual(
inventory[ipv4_classname]['total'], inventory['total'],
self.mock_p_client.update_resource_provider_inventory. self.mock_p_client.update_resource_provider_inventory.
call_args[0][1][ipv4_classname]['total']) call_args[0][1]['total'])
self.assertEqual( self.assertEqual(
inventory[ipv4_classname]['reserved'], inventory['reserved'],
self.mock_p_client.update_resource_provider_inventory. self.mock_p_client.update_resource_provider_inventory.
call_args[0][1][ipv4_classname]['reserved']) call_args[0][1]['reserved'])
self.mock_p_client.reset_mock() self.mock_p_client.reset_mock()
self.mock_n_client.reset_mock() self.mock_n_client.reset_mock()
def _get_inventory(self, total, reserved): def _get_inventory(self, total, reserved):
inventory = { inventory = {'total': total, 'reserved': reserved, 'min_unit': 1,
seg_plugin.IPV4_RESOURCE_CLASS: { 'max_unit': 1, 'step_size': 1, 'allocation_ratio': 1.0}
'total': total, 'reserved': reserved, 'min_unit': 1,
'max_unit': 1, 'step_size': 1, 'allocation_ratio': 1.0
}}
return inventory, copy.deepcopy(inventory) return inventory, copy.deepcopy(inventory)
def _test_second_subnet_association_with_segment(self): def _test_second_subnet_association_with_segment(self):
@ -2050,20 +2046,20 @@ class TestNovaSegmentNotifier(SegmentAwareIpamTestCase):
def _assert_inventory_update_port(self, segment_id, inventory, def _assert_inventory_update_port(self, segment_id, inventory,
num_fixed_ips): num_fixed_ips):
inventory[seg_plugin.IPV4_RESOURCE_CLASS]['reserved'] += num_fixed_ips inventory['reserved'] += num_fixed_ips
self.mock_p_client.get_inventory.assert_called_with( self.mock_p_client.get_inventory.assert_called_with(
segment_id, seg_plugin.IPV4_RESOURCE_CLASS) segment_id, seg_plugin.IPV4_RESOURCE_CLASS)
self.mock_p_client.update_resource_provider_inventory.\ self.mock_p_client.update_resource_provider_inventory.\
assert_called_with(segment_id, inventory, assert_called_with(segment_id, inventory,
seg_plugin.IPV4_RESOURCE_CLASS) seg_plugin.IPV4_RESOURCE_CLASS)
self.assertEqual( self.assertEqual(
inventory[seg_plugin.IPV4_RESOURCE_CLASS]['total'], inventory['total'],
self.mock_p_client.update_resource_provider_inventory. self.mock_p_client.update_resource_provider_inventory.
call_args[0][1][seg_plugin.IPV4_RESOURCE_CLASS]['total']) call_args[0][1]['total'])
self.assertEqual( self.assertEqual(
inventory[seg_plugin.IPV4_RESOURCE_CLASS]['reserved'], inventory['reserved'],
self.mock_p_client.update_resource_provider_inventory. self.mock_p_client.update_resource_provider_inventory.
call_args[0][1][seg_plugin.IPV4_RESOURCE_CLASS]['reserved']) call_args[0][1]['reserved'])
self.mock_p_client.reset_mock() self.mock_p_client.reset_mock()
self.mock_n_client.reset_mock() self.mock_n_client.reset_mock()
@ -2117,10 +2113,9 @@ class TestNovaSegmentNotifier(SegmentAwareIpamTestCase):
segment_id, before_create_inventory, port = self._test_create_port( segment_id, before_create_inventory, port = self._test_create_port(
**kwargs) **kwargs)
self.mock_p_client.reset_mock() self.mock_p_client.reset_mock()
ipv4_classname = seg_plugin.IPV4_RESOURCE_CLASS
inventory, original_inventory = self._get_inventory( inventory, original_inventory = self._get_inventory(
before_create_inventory[ipv4_classname]['total'], before_create_inventory['total'],
before_create_inventory[ipv4_classname]['reserved'] + 1) before_create_inventory['reserved'] + 1)
self.mock_p_client.get_inventory.return_value = inventory self.mock_p_client.get_inventory.return_value = inventory
self._delete('ports', port['port']['id']) self._delete('ports', port['port']['id'])
self.batch_notifier._notify() self.batch_notifier._notify()