wlb: T7977: Updated smoketest to validate nft vmap weight buckets

1 files changed, 19 insertions, 22 deletions

diff --git a/smoketest/scripts/cli/test_load-balancing_wan.py b/smoketest/scripts/cli/test_load-balancing_wan.py
index 255203e91..3b956a12e 100755
--- a/smoketest/scripts/cli/test_load-balancing_wan.py
+++ b/smoketest/scripts/cli/test_load-balancing_wan.py
@@ -442,20 +442,17 @@ echo "$ifname - $state" > {hook_output_path}
         self.cli_set(['interfaces', 'ethernet', 'eth0', 'vif', '201', 'address', '203.0.113.14/30'])
         self.cli_set(['interfaces', 'ethernet', 'eth0', 'vif', '202', 'address', '203.0.113.18/30'])
         self.cli_set(['interfaces', 'ethernet', 'eth0', 'vif', '203', 'address', '203.0.113.22/30'])
+        self.cli_set(['interfaces', 'ethernet', 'eth0', 'vif', '204', 'address', '203.0.113.26/30'])
+        self.cli_set(['interfaces', 'ethernet', 'eth0', 'vif', '205', 'address', '203.0.113.30/30'])
+
 
         self.cli_set(['interfaces', 'ethernet', lan_iface, 'vif', '100', 'address', '198.51.100.2/30'])
         self.cli_set(['interfaces', 'ethernet', lan_iface, 'vif', '200', 'address', '198.51.100.6/30'])
 
         # Health checks for equal weight test
         self.cli_set(base_path + ['wan', 'interface-health', 'eth0.101', 'nexthop', '203.0.113.2'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.101', 'success-count', '1'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.101', 'failure-count', '1'])
         self.cli_set(base_path + ['wan', 'interface-health', 'eth0.102', 'nexthop', '203.0.113.6'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.102', 'success-count', '1'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.102', 'failure-count', '1'])
         self.cli_set(base_path + ['wan', 'interface-health', 'eth0.103', 'nexthop', '203.0.113.10'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.103', 'success-count', '1'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.103', 'failure-count', '1'])
         self.cli_set(base_path + ['wan', 'rule', '10', 'inbound-interface', f'{lan_iface}.100'])
         self.cli_set(base_path + ['wan', 'rule', '10', 'interface', 'eth0.101'])
         self.cli_set(base_path + ['wan', 'rule', '10', 'interface', 'eth0.102'])
@@ -463,18 +460,16 @@ echo "$ifname - $state" > {hook_output_path}
 
         # Health checks for unequal weight test
         self.cli_set(base_path + ['wan', 'interface-health', 'eth0.201', 'nexthop', '203.0.113.14'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.201', 'success-count', '1'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.201', 'failure-count', '1'])
         self.cli_set(base_path + ['wan', 'interface-health', 'eth0.202', 'nexthop', '203.0.113.18'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.202', 'success-count', '1'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.202', 'failure-count', '1'])
         self.cli_set(base_path + ['wan', 'interface-health', 'eth0.203', 'nexthop', '203.0.113.22'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.203', 'success-count', '1'])
-        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.203', 'failure-count', '1'])
+        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.204', 'nexthop', '203.0.113.26'])
+        self.cli_set(base_path + ['wan', 'interface-health', 'eth0.205', 'nexthop', '203.0.113.30'])
         self.cli_set(base_path + ['wan', 'rule', '20', 'inbound-interface', f'{lan_iface}.200'])
-        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.201'])
-        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.202', 'weight', '2'])
-        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.203', 'weight', '3'])
+        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.201', 'weight', '2'])
+        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.202', 'weight', '4'])
+        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.203', 'weight', '4'])
+        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.204', 'weight', '7'])
+        self.cli_set(base_path + ['wan', 'rule', '20', 'interface', 'eth0.205', 'weight', '4'])
 
         # commit changes
         self.cli_commit()
@@ -490,17 +485,19 @@ echo "$ifname - $state" > {hook_output_path}
             # check total count of 'jump' across all matching lines
             total_jumps = sum(l.count('jump') for l in lines)
 
-            return total_jumps == 6
+            return total_jumps == 8
 
         wait_for(check_wlb_status)
 
         nftables_search = [
-            ['jump wlb_mangle_isp_eth0.101'],
-            ['jump wlb_mangle_isp_eth0.102'],
-            ['jump wlb_mangle_isp_eth0.103'],
-            ['jump wlb_mangle_isp_eth0.201'],
-            ['jump wlb_mangle_isp_eth0.202'],
-            ['jump wlb_mangle_isp_eth0.203'],
+            ['0 : jump wlb_mangle_isp_eth0.101',
+            '1 : jump wlb_mangle_isp_eth0.102',
+            '2 : jump wlb_mangle_isp_eth0.103'],
+            ['0-1 : jump wlb_mangle_isp_eth0.201',
+            '2-5 : jump wlb_mangle_isp_eth0.202',
+            '6-9 : jump wlb_mangle_isp_eth0.203',
+            '10-13 : jump wlb_mangle_isp_eth0.205',
+            '14-20 : jump wlb_mangle_isp_eth0.204'],
         ]
 
         self.verify_nftables_chain(nftables_search, 'ip vyos_wanloadbalance', 'wlb_mangle_prerouting')