From 8a2c619537011e7dc66e98f1a4aac74162273885 Mon Sep 17 00:00:00 2001 From: yaqiangz Date: Wed, 15 Feb 2023 13:49:46 +0000 Subject: [PATCH 1/2] [dhcp_relay] Remove add field of vlanid to DHCP_RELAY table while adding vlan --- config/vlan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config/vlan.py b/config/vlan.py index 48096d2ebd..ed13fabaff 100644 --- a/config/vlan.py +++ b/config/vlan.py @@ -44,7 +44,7 @@ def add_vlan(db, vid): set_dhcp_relay_table('VLAN', db.cfgdb, vlan, {'vlanid': str(vid)}) # set dhcpv6_relay table - set_dhcp_relay_table('DHCP_RELAY', db.cfgdb, vlan, {'vlanid': str(vid)}) + set_dhcp_relay_table('DHCP_RELAY', db.cfgdb, vlan, None) # We need to restart dhcp_relay service after dhcpv6_relay config change dhcp_relay_util.handle_restart_dhcp_relay_service() From 9908efc6d6dd9d4fb9ae91f4c98cc17917703b21 Mon Sep 17 00:00:00 2001 From: yaqiangz Date: Wed, 15 Feb 2023 14:20:34 +0000 Subject: [PATCH 2/2] Fix unit test failed --- tests/vlan_test.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/vlan_test.py b/tests/vlan_test.py index 85673c5020..f582d0e3ba 100644 --- a/tests/vlan_test.py +++ b/tests/vlan_test.py @@ -19,9 +19,6 @@ "table": "DHCP_RELAY" } } -DHCP_RELAY_TABLE_ENTRY = { - "vlanid": "1001" -} show_vlan_brief_output="""\ +-----------+-----------------+-----------------+----------------+-------------+ @@ -610,7 +607,8 @@ def test_config_add_del_vlan_dhcp_relay(self, ip_version, mock_restart_dhcp_rela print(result.output) assert result.exit_code == 0 - assert db.cfgdb.get_entry(IP_VERSION_PARAMS_MAP[ip_version]["table"], "Vlan1001") == DHCP_RELAY_TABLE_ENTRY + exp_output = {"vlanid": "1001"} if ip_version == "ipv4" else {} + assert db.cfgdb.get_entry(IP_VERSION_PARAMS_MAP[ip_version]["table"], "Vlan1001") == exp_output # del vlan 1001 result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1001"], obj=db)