Added Portchannel test case.

Author: AkhileshSamineni

tests/dhcp_relay_test.py

@@ -16,6 +16,10 @@
 |                  | 12.0.0.2              |
 |                  | 12.0.0.3              |
 +------------------+-----------------------+
+| PortChannel0001  | 11.0.0.1              |
+|                  | 11.0.0.2              |
+|                  | 11.0.0.3              |
++------------------+-----------------------+
 | Vlan1000         | 192.0.0.1             |
 |                  | 192.0.0.2             |
 |                  | 192.0.0.3             |
@@ -48,6 +52,16 @@
 Restarting DHCP relay service...
 """
 
+config_portchannel_add_dhcp_relay_output="""\
+Added DHCP relay address 11.0.0.4 to PortChannel0001
+Restarting DHCP relay service...
+"""
+
+config_portchannel_rem_dhcp_relay_output="""\
+Removed DHCP relay address 11.0.0.3 from PortChannel0001
+Restarting DHCP relay service...
+"""
+
 class TestDhcpRelay(object):
     @classmethod
     def setup_class(cls):
@@ -114,6 +128,35 @@ def test_config_add_delete_dhcp_server_on_vlan_interface(self):
             assert result.output == config_vlan_rem_dhcp_relay_output
             assert mock_run_command.call_count == 3
 
+    def test_config_add_delete_dhcp_server_on_portchannel_interface(self):
+        runner = CliRunner()
+        db = Db()
+        obj = {'db':db.cfgdb}
+
+        with mock.patch("utilities_common.cli.run_command") as mock_run_command:
+            # add a same dhcp relay server address to PortChannel0001
+            result = runner.invoke(config.config.commands["interface"].commands["ip"].commands["dhcp-relay"].commands["add"], ["PortChannel0001", "11.0.0.3"], obj=obj)
+            print(result.exit_code)
+            print(result.output)
+            assert result.exit_code != 0
+            assert "Error: IP address 11.0.0.3 is already configured" in result.output
+
+            # remove dhcp relay server address from PortChannel0001
+            result = runner.invoke(config.config.commands["interface"].commands["ip"].commands["dhcp-relay"].commands["remove"], ["PortChannel0001", "11.0.0.3"], obj=obj)
+            print(result.exit_code)
+            print(result.output)
+            assert result.exit_code == 0
+            assert result.output == config_portchannel_rem_dhcp_relay_output
+            assert mock_run_command.call_count == 3
+
+            # add a new dhcp relay server address to PortChannel0001
+            result = runner.invoke(config.config.commands["interface"].commands["ip"].commands["dhcp-relay"].commands["add"], ["PortChannel0001", "11.0.0.4"], obj=obj)
+            print(result.exit_code)
+            print(result.output)
+            assert result.exit_code == 0
+            assert result.output == config_portchannel_add_dhcp_relay_output
+            assert mock_run_command.call_count == 6
+
     @classmethod
     def teardown_class(cls):
         os.environ['UTILITIES_UNIT_TESTING'] = "0"

tests/mock_tables/config_db.json

@@ -520,7 +520,7 @@
     },
     "PORTCHANNEL_INTERFACE|PortChannel0001": {
         "NULL": "NULL",
-        "dhcp_servers@": "11.0.0.1,11.0.0.2,11.0.0.3",
+        "dhcp_servers@": "11.0.0.1,11.0.0.2,11.0.0.3"
     },
     "PORTCHANNEL_INTERFACE|PortChannel0002": {
         "NULL": "NULL"