diff --git a/kuryr_kubernetes/controller/drivers/nested_vlan_vif.py b/kuryr_kubernetes/controller/drivers/nested_vlan_vif.py index 4cd863922..e3dabea10 100644 --- a/kuryr_kubernetes/controller/drivers/nested_vlan_vif.py +++ b/kuryr_kubernetes/controller/drivers/nested_vlan_vif.py @@ -306,6 +306,6 @@ class NestedVlanPodVIFDriver(nested_vif.NestedPodVIFDriver): os_net = clients.get_network_client() trunk = os_net.get_trunk(trunk_id) for port in trunk.sub_ports: - vlan_ids.add(port.segmentation_id) + vlan_ids.add(port['segmentation_id']) return vlan_ids diff --git a/kuryr_kubernetes/tests/unit/controller/drivers/test_nested_vlan_vif.py b/kuryr_kubernetes/tests/unit/controller/drivers/test_nested_vlan_vif.py index a069e5874..b01c4c363 100644 --- a/kuryr_kubernetes/tests/unit/controller/drivers/test_nested_vlan_vif.py +++ b/kuryr_kubernetes/tests/unit/controller/drivers/test_nested_vlan_vif.py @@ -644,7 +644,7 @@ class TestNestedVlanPodVIFDriver(test_base.TestCase): trunk_id = mock.sentinel.trunk_id vlan_ids.add('100') - port = os_port.Port(segmentation_id='100') + port = {"segmentation_id": '100'} # Trunk.sub_ports is a list of dicts trunk_obj = os_trunk.Trunk(sub_ports=[port]) os_net.get_trunk.return_value = trunk_obj self.assertEqual(vlan_ids,