[Dash] [UT] Add ZMQ test case for dash (#2967)

liuh-80 · web-flow · commit 920959cfcdc2 · 2023-12-07T09:53:33.000-08:00
* Add ZMQ test case for dash
diff --git a/tests/create_appliance.py b/tests/create_appliance.py
@@ -0,0 +1,42 @@
+#!/usr/bin/python3
+
+"""
+    Connect to Dash orch with ZMQ and send create appliance request.
+    usage:
+        python3 create_appliance.py [appliance ID]
+    Example:
+        python3 create_appliance.py 1234
+"""
+
+from swsscommon import swsscommon
+from dash_api.appliance_pb2 import *
+import typing
+import ipaddress
+import socket
+import sys
+
+def to_string(value):
+    if isinstance(value, bool):
+        return "true" if value else "false"
+    elif isinstance(value, bytes):
+        return value
+    return str(value)
+
+# connect to Dash ZMQ endpoint
+db_connection = swsscommon.DBConnector("APPL_DB", 0)
+zmq_client = swsscommon.ZmqClient("tcp://127.0.0.1:8100")
+app_dash_appliance_table = swsscommon.ZmqProducerStateTable(
+                db_connection,
+                "DASH_APPLIANCE_TABLE",
+                zmq_client,
+                True)
+
+# prepare create appliance request
+pairs_str = []
+pb = Appliance()
+pb.sip.ipv4 = socket.htonl(int(ipaddress.ip_address("10.0.0.1")))
+pb.vm_vni = int(sys.argv[1])
+pairs_str.append(("pb", pb.SerializeToString()))
+
+# send create appliance request via ZMQ
+app_dash_appliance_table.set("100", pairs_str)
diff --git a/tests/test_zmq.py b/tests/test_zmq.py
@@ -0,0 +1,104 @@
+from swsscommon import swsscommon
+
+from dash_api.appliance_pb2 import *
+from dash_api.vnet_pb2 import *
+from dash_api.eni_pb2 import *
+from dash_api.route_pb2 import *
+from dash_api.route_rule_pb2 import *
+from dash_api.vnet_mapping_pb2 import *
+from dash_api.route_type_pb2 import *
+from dash_api.types_pb2 import *
+
+import typing
+import time
+import binascii
+import uuid
+import ipaddress
+import sys
+import socket
+import logging
+import pytest
+
+logging.basicConfig(level=logging.INFO)
+zmq_logger = logging.getLogger(__name__)
+
+DVS_ENV = ["HWSKU=DPU-2P"]
+NUM_PORTS = 2
+
+class Table(object):
+    def __init__(self, database, table_name: str):
+        self.table_name = table_name
+        self.table = swsscommon.Table(database.db_connection, self.table_name)
+
+    def __getitem__(self, key: str):
+        exists, result = self.table.get(str(key))
+        if not exists:
+            return None
+        else:
+            return dict(result)
+
+    def get_keys(self):
+        return self.table.getKeys()
+
+    def get_newly_created_oid(self, old_oids):
+        new_oids = self.asic_db.wait_for_n_keys(table, len(old_oids) + 1)
+        oid = [ids for ids in new_oids if ids not in old_oids]
+        return oid[0]
+
+class DashZmq(object):
+    def __init__(self, dvs):
+        self.dvs = dvs
+        self.asic_direction_lookup_table = Table(
+            self.dvs.get_asic_db(), "ASIC_STATE:SAI_OBJECT_TYPE_DIRECTION_LOOKUP_ENTRY")
+        self.asic_vip_table = Table(
+            self.dvs.get_asic_db(), "ASIC_STATE:SAI_OBJECT_TYPE_VIP_ENTRY")
+
+class TestZmqDash(object):
+    @pytest.fixture(scope="class")
+    def enable_orchagent_zmq(self, dvs):
+        # change orchagent to use ZMQ
+        dvs.runcmd("cp /usr/bin/orchagent.sh /usr/bin/orchagent.sh_zmq_ut_backup")
+        dvs.runcmd("sed -i.bak 's/\/usr\/bin\/orchagent /\/usr\/bin\/orchagent -q tcp:\/\/127.0.0.1:8100 /g' /usr/bin/orchagent.sh")
+        dvs.stop_swss()
+        dvs.start_swss()
+
+        process_statue = dvs.runcmd("ps -ef")
+        zmq_logger.debug("Process status: {}".format(process_statue))
+
+        yield
+
+        # revert change
+        dvs.runcmd("cp /usr/bin/orchagent.sh_zmq_ut_backup /usr/bin/orchagent.sh")
+        dvs.stop_swss()
+        dvs.start_swss()
+
+    @pytest.mark.usefixtures("enable_orchagent_zmq")
+    def test_appliance(self, dvs):
+        # upload test script to test container and create applicance with it
+        dvs.copy_file("/", "create_appliance.py")
+        dvs.runcmd(['sh', '-c', "python3 create_appliance.py {}".format(1234)])
+        time.sleep(3)
+
+        asic_direction_lookup_table = Table(
+            dvs.get_asic_db(), "ASIC_STATE:SAI_OBJECT_TYPE_DIRECTION_LOOKUP_ENTRY")
+        direction_entries = asic_direction_lookup_table.get_keys()
+        zmq_logger.info("Keys from asic_direction_lookup_table: {}".format(direction_entries))
+
+        assert direction_entries
+        fvs = asic_direction_lookup_table[direction_entries[0]]
+        zmq_logger.info("Data from asic_direction_lookup_table: {}={}".format(direction_entries[0], fvs))
+        for fv in fvs.items():
+            if fv[0] == "SAI_DIRECTION_LOOKUP_ENTRY_ATTR_ACTION":
+                assert fv[1] == "SAI_DIRECTION_LOOKUP_ENTRY_ACTION_SET_OUTBOUND_DIRECTION"
+
+        asic_vip_table = Table(
+            dvs.get_asic_db(), "ASIC_STATE:SAI_OBJECT_TYPE_VIP_ENTRY")
+        vip_entries = asic_vip_table.get_keys()
+        zmq_logger.info("Keys from asic_vip_table: {}".format(direction_entries))
+
+        assert vip_entries
+        fvs = asic_vip_table[vip_entries[0]]
+        zmq_logger.info("Data from asic_vip_table: {}={}".format(vip_entries[0], fvs))
+        for fv in fvs.items():
+            if fv[0] == "SAI_VIP_ENTRY_ATTR_ACTION":
+                assert fv[1] == "SAI_VIP_ENTRY_ACTION_ACCEPT"
