portbindings: use constants for extension keys

Use the constants defined in the portbindings extension.

TrivialFix

Change-Id: I90c3deeae9018d4e5b23283f64657d9f5d5723c7
This commit is contained in:
Gary Kotton 2015-12-21 10:21:57 -08:00
parent 6c5ecdc0e6
commit 49d25b3992
4 changed files with 7 additions and 7 deletions

View File

@ -71,8 +71,8 @@ class SriovNicSwitchRpcCallbacks(sg_rpc.SecurityGroupAgentRpcCallbackMixin):
# processed in the same order as the relevant API requests.
mac = port['mac_address']
pci_slot = None
if port.get('binding:profile'):
pci_slot = port['binding:profile'].get('pci_slot')
if port.get(portbindings.PROFILE):
pci_slot = port[portbindings.PROFILE].get('pci_slot')
if pci_slot:
self.agent.updated_devices.add((mac, pci_slot))

View File

@ -53,7 +53,7 @@ class FakePortContext(api.PortContext):
@property
def current(self):
return {'id': PORT_ID,
'binding:vnic_type': self._bound_vnic_type}
portbindings.VNIC_TYPE: self._bound_vnic_type}
@property
def original(self):

View File

@ -286,7 +286,7 @@ class TestSriovNicSwitchRpcCallbacks(base.BaseTestCase):
def _create_fake_port(self):
return {'id': uuidutils.generate_uuid(),
'binding:profile': {'pci_slot': PCI_SLOT},
portbindings.PROFILE: {'pci_slot': PCI_SLOT},
'mac_address': DEVICE_MAC}
def test_port_update_with_pci_slot(self):
@ -305,7 +305,7 @@ class TestSriovNicSwitchRpcCallbacks(base.BaseTestCase):
def test_port_update_without_pci_slot(self):
port = self._create_fake_port()
port['binding:profile'] = None
port[portbindings.PROFILE] = None
kwargs = {'context': self.context, 'port': port}
self.sriov_rpc_callback.port_update(**kwargs)
self.assertEqual(set(), self.agent.updated_devices)

View File

@ -45,8 +45,8 @@ class TestFakePortContext(base.FakePortContext):
@property
def current(self):
return {'id': base.PORT_ID,
'binding:vnic_type': self._bound_vnic_type,
'binding:profile': self._bound_profile}
portbindings.VNIC_TYPE: self._bound_vnic_type,
portbindings.PROFILE: self._bound_profile}
def set_binding(self, segment_id, vif_type, vif_details, state):
self._bound_segment_id = segment_id