added new HA fixtures to programme dash ha scope and ha set tables via… (#22390)

What is the motivation for this PR
Not provided in PR description.

How did you do it
Added new pytest fixtures.

How did you verify/test it
Ran the tests and verified the DPU Tables HA Scope and HA Set.

Signed-off-by: nnelluri-cisco <nnelluri@cisco.com>
Signed-off-by: nnelluri <nnelluri@cisco.com>

Author: nnelluri-cisco

tests/common/ha/dash_ha_scope_config_table.json

@@ -1,14 +1,20 @@
 {
-  "DASH_HA_SET_CONFIG_TABLE": {
-    "ha-set-1001": {
+  "DASH_HA_SCOPE_CONFIG_TABLE": {
+    "vdpu0_0:haset0_0": {
       "version": "1.0",
-      "vip_v4": "192.168.100.1",
-      "vip_v6": "fd00::1001",
-      "vdpu_ids": ["dpu0", "dpu1"],
-      "scope": "dpu",
-      "pinned_vdpu_bfd_probe_states": ["up", "down"],
-      "preferred_vdpu_id": "dpu0",
-      "preferred_standalone_vdpu_index": 0
+      "owner": "switch",
+      "disabled": "false",
+      "ha_set_id": "haset0_0",
+      "desired_ha_state": "active",
+      "approved_pending_operation_ids": ""
+    },
+    "vdpu1_0:haset0_0": {
+      "version": "1.0",
+      "owner": "switch",
+      "disabled": "false",
+      "ha_set_id": "haset0_0",
+      "desired_ha_state": "standalone",
+      "approved_pending_operation_ids": ""
     }
   }
 }

tests/common/ha/dash_ha_set_dpu_config_table.json

@@ -0,0 +1,14 @@
+{
+  "DASH_HA_SET_CONFIG_TABLE": {
+    "haset0_0": {
+      "version": "1.0",
+      "vip_v4": "3.2.1.0",
+      "vip_v6": "3:2::1:0",
+      "vdpu_ids": ["vdpu0_0", "vdpu1_0"],
+      "scope": "dpu",
+      "pinned_vdpu_bfd_probe_states": ["up", "down"],
+      "preferred_vdpu_id": "vdpu0_0",
+      "preferred_standalone_vdpu_index": 0
+    }
+  }
+}

tests/ha/conftest.py

@@ -1,20 +1,34 @@
+import logging
 import pytest
-import time
+from pathlib import Path
+from collections import defaultdict
+import os
 import json
+import time
 
 from tests.common.config_reload import config_reload
-from pathlib import Path
-from collections import defaultdict
 from tests.common.helpers.constants import DEFAULT_NAMESPACE
-from common.ha.smartswitch_ha_helper import PtfTcpTestAdapter
-from common.ha.smartswitch_ha_io import SmartSwitchHaTrafficTest
-from common.ha.smartswitch_ha_helper import (
+from tests.common.ha.smartswitch_ha_helper import PtfTcpTestAdapter
+from tests.common.ha.smartswitch_ha_io import SmartSwitchHaTrafficTest
+from tests.common.ha.smartswitch_ha_helper import (
     add_port_to_namespace,
     remove_namespace,
     add_static_route_to_ptf,
     add_static_route_to_dut
 )
 
+from tests.ha.ha_utils import (
+
+    build_dash_ha_scope_args,
+    wait_for_pending_operation_id,
+    build_dash_ha_scope_activate_args,
+    wait_for_ha_state,
+    build_dash_ha_set_args,
+    proto_utils_hset
+)
+
+logger = logging.getLogger(__name__)
+
 
 @pytest.fixture(scope="module")
 def copy_files(ptfhost):
@@ -408,3 +422,127 @@ def setup_ha_config(duthosts):
         final_cfg[f"DUT{switch_id}"] = cfg
 
     return final_cfg
+
+
+@pytest.fixture(scope="module")
+def setup_dash_ha_from_json(duthosts):
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    base_dir = os.path.join(current_dir, "..", "common", "ha")
+    ha_set_file = os.path.join(base_dir, "dash_ha_set_dpu_config_table.json")
+
+    with open(ha_set_file) as f:
+        ha_set_data = json.load(f)["DASH_HA_SET_CONFIG_TABLE"]
+
+    # -------------------------------------------------
+    # Step 1: Program HA SET on BOTH DUTs
+    # -------------------------------------------------
+    for duthost in duthosts:
+        for key, fields in ha_set_data.items():
+            proto_utils_hset(
+                duthost,
+                table="DASH_HA_SET_CONFIG_TABLE",
+                key=key,
+                args=build_dash_ha_set_args(fields),
+            )
+
+    # -------------------------------------------------
+    # Step 2: Initial HA SCOPE per DUT
+    # -------------------------------------------------
+    ha_scope_per_dut = [
+        (
+            "vdpu0_0:haset0_0",
+            {
+                "version": "1",
+                "disabled": "true",
+                "desired_ha_state": "active",
+                "ha_set_id": "haset0_0",
+                "owner": "dpu",
+            },
+        ),
+        (
+            "vdpu1_0:haset0_0",
+            {
+                "version": "1",
+                "disabled": "true",
+                "desired_ha_state": "unspecified",
+                "ha_set_id": "haset0_0",
+                "owner": "dpu",
+            },
+        ),
+    ]
+
+    for duthost, (key, fields) in zip(duthosts, ha_scope_per_dut):
+        proto_utils_hset(
+            duthost,
+            table="DASH_HA_SCOPE_CONFIG_TABLE",
+            key=key,
+            args=build_dash_ha_scope_args(fields),
+        )
+
+
+@pytest.fixture(scope="module")
+def activate_dash_ha_from_json(duthosts):
+    # -------------------------------------------------
+    # Step 4: Activate Role (using pending_operation_ids)
+    # -------------------------------------------------
+    activate_scope_per_dut = [
+        (
+            "vdpu0_0:haset0_0",
+            {
+                "version": "1",
+                "disabled": False,
+                "desired_ha_state": "active",
+                "ha_set_id": "haset0_0",
+                "owner": "dpu",
+            },
+        ),
+        (
+            "vdpu1_0:haset0_0",
+            {
+                "version": "1",
+                "disabled": False,
+                "desired_ha_state": "unspecified",
+                "ha_set_id": "haset0_0",
+                "owner": "dpu",
+            },
+        ),
+    ]
+    for duthost, (key, fields) in zip(duthosts, activate_scope_per_dut):
+        proto_utils_hset(
+            duthost,
+            table="DASH_HA_SCOPE_CONFIG_TABLE",
+            key=key,
+            args=build_dash_ha_scope_args(fields),
+        )
+    for idx, (duthost, (key, fields)) in enumerate(zip(duthosts, activate_scope_per_dut)):
+        pending_id = wait_for_pending_operation_id(
+            duthost,
+            scope_key=key,
+            expected_op_type="activate_role",
+            timeout=120,
+            interval=2
+        )
+        assert pending_id, (
+            f"Timed out waiting for active pending_operation_id "
+            f"for {duthost.hostname} scope {key}"
+        )
+
+        logger.info(f"DASH HA {duthost.hostname} found pending id {pending_id}")
+        proto_utils_hset(
+            duthost,
+            table="DASH_HA_SCOPE_CONFIG_TABLE",
+            key=key,
+            args=build_dash_ha_scope_activate_args(fields, pending_id),
+        )
+        # Verify HA state using fields
+        expected_state = "active" if idx == 0 else "standby"
+        assert wait_for_ha_state(
+            duthost,
+            scope_key=key,
+            expected_state=expected_state,
+            timeout=120,
+            interval=5,
+        ), f"HA did not reach expected state {expected_state} for {key} on {duthost.hostname}"
+        logger.info(f"DASH HA Step-4 Activate Role completed for {duthost.hostname}")
+    logger.info("DASH HA Step-4 Activate Role completed")
+    yield