[ssip]: CLI UTs rev01.

Signed-off-by: Nazarii Hnydyn <[email protected]>

Author: nazariig

tests/syslog_input/config_mock.py

@@ -0,0 +1,70 @@
+"""
+Module holding IP/VRF mock data for config CLI command of the syslog_test.py
+"""
+
+VRF_LIST = '''
+[
+    {
+        "name": "mgmt"
+    },
+    {
+        "name": "Vrf-Data"
+    }
+]
+'''
+
+VRF_MGMT_MEMBERS = '''
+[
+    {
+        "ifname": "eth0"
+    }
+]
+'''
+
+VRF_DATA_MEMBERS = '''
+[
+    {
+        "ifname": "Ethernet0"
+    }
+]
+'''
+
+IP_ADDR_LIST = '''
+[
+    {
+        "ifname": "Ethernet0",
+        "addr_info": [
+            {
+                "local": "1111::1111"
+            }
+        ]
+    },
+    {
+        "ifname": "Loopback0",
+        "addr_info": [
+            {
+                "local": "1.1.1.1"
+            }
+        ]
+    },
+    {
+        "ifname": "eth0",
+        "addr_info": [
+            {
+                "local": "3.3.3.3"
+            }
+        ]
+    }
+]
+'''
+
+def exec_cmd_mock(cmd):
+    if cmd == 'ip --json vrf show':
+        return VRF_LIST
+    elif cmd == 'ip --json link show vrf mgmt':
+        return VRF_MGMT_MEMBERS
+    elif cmd == 'ip --json link show vrf Vrf-Data':
+        return VRF_DATA_MEMBERS
+    elif cmd == 'ip --json address show':
+        return IP_ADDR_LIST
+    raise Exception("{}: unknown command: {}".format(__name__, cmd))

tests/syslog_test.py

@@ -13,6 +13,7 @@
 from utilities_common.db import Db
 
 from .mock_tables import dbconnector
+from .syslog_input import config_mock
 from .syslog_input import assert_show_output
 
 
@@ -26,10 +27,6 @@
 ERROR_PATTERN_INVALID_VRF = "invalid choice"
 ERROR_PATTERN_NONEXISTENT_VRF = "VRF doesn't exist in Linux"
 
-VRF_LIST = ["mgmt", "Vrf-Data"]
-VRF_MEMBER_DICT = {"mgmt": ["eth0"], "Vrf-Data": ["Ethernet0"]}
-IP_ADDR_DICT = {"Ethernet0": ["1111::1111"], "Loopback0": ["1.1.1.1"], "eth0": ["3.3.3.3"]}
-
 SUCCESS = 0
 ERROR2 = 2
 
@@ -77,9 +74,7 @@ def test_config_syslog_basic(self, server_ip):
         assert result.exit_code == SUCCESS
 
     @mock.patch("utilities_common.cli.run_command", mock.MagicMock(return_value=None))
-    @mock.patch("config.syslog.get_vrf_list", mock.MagicMock(return_value=VRF_LIST))
-    @mock.patch("config.syslog.get_vrf_member_dict", mock.MagicMock(return_value=VRF_MEMBER_DICT))
-    @mock.patch("config.syslog.get_ip_addr_dict", mock.MagicMock(return_value=IP_ADDR_DICT))
+    @mock.patch("config.syslog.exec_cmd", mock.MagicMock(side_effect=config_mock.exec_cmd_mock))
     @pytest.mark.parametrize("server_ip,source_ip,port,vrf", [
         ("2.2.2.2", "1.1.1.1", "514", "default"),
         ("4.4.4.4", "3.3.3.3", "514", "mgmt"),