vlan_test.py

import os
import traceback
from unittest import mock

from click.testing import CliRunner

import config.main as config
import show.main as show
from utilities_common.db import Db

show_vlan_brief_output="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | Ethernet4       | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | Ethernet8       | untagged       | 192.0.0.2             |             |
|           |                 | Ethernet12      | untagged       | 192.0.0.3             |             |
|           |                 | Ethernet16      | untagged       | 192.0.0.4             |             |
|           |                 |                 |                | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | Ethernet24      | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | Ethernet28      | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

show_vlan_brief_in_alias_mode_output="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | etp2            | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | etp3            | untagged       | 192.0.0.2             |             |
|           |                 | etp4            | untagged       | 192.0.0.3             |             |
|           |                 | etp5            | untagged       | 192.0.0.4             |             |
|           |                 |                 |                | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | etp7            | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | etp8            | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

show_vlan_brief_empty_output="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      2000 | 192.168.0.10/21 | Ethernet24      | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | Ethernet28      | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

show_vlan_brief_with_portchannel_output="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | Ethernet4       | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | Ethernet8       | untagged       | 192.0.0.2             |             |
|           |                 | Ethernet12      | untagged       | 192.0.0.3             |             |
|           |                 | Ethernet16      | untagged       | 192.0.0.4             |             |
|           |                 | PortChannel1001 | untagged       | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | Ethernet24      | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | Ethernet28      | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

show_vlan_config_output="""\
Name        VID  Member           Mode
--------  -----  ---------------  --------
Vlan1000   1000  Ethernet4        untagged
Vlan1000   1000  Ethernet8        untagged
Vlan1000   1000  Ethernet12       untagged
Vlan1000   1000  Ethernet16       untagged
Vlan2000   2000  Ethernet24       untagged
Vlan2000   2000  Ethernet28       untagged
Vlan3000   3000
Vlan4000   4000  PortChannel1001  tagged
"""

show_vlan_config_in_alias_mode_output="""\
Name        VID  Member           Mode
--------  -----  ---------------  --------
Vlan1000   1000  etp2             untagged
Vlan1000   1000  etp3             untagged
Vlan1000   1000  etp4             untagged
Vlan1000   1000  etp5             untagged
Vlan2000   2000  etp7             untagged
Vlan2000   2000  etp8             untagged
Vlan3000   3000
Vlan4000   4000  PortChannel1001  tagged
"""

config_vlan_add_dhcp_relay_output="""\
Added DHCP relay destination address 192.0.0.100 to Vlan1000
Restarting DHCP relay service...
"""

config_vlan_del_dhcp_relay_output="""\
Removed DHCP relay destination address 192.0.0.100 from Vlan1000
Restarting DHCP relay service...
"""

config_vlan_add_dhcp_relayv6_output="""\
Added DHCP relay destination address fc02:2000::3 to Vlan1000
Restarting DHCP relay service...
"""

config_vlan_del_dhcp_relayv6_output="""\
Removed DHCP relay destination address fc02:2000::3 from Vlan1000
Restarting DHCP relay service...
"""

show_vlan_brief_output_with_new_dhcp_relay_address="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | Ethernet4       | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | Ethernet8       | untagged       | 192.0.0.2             |             |
|           |                 | Ethernet12      | untagged       | 192.0.0.3             |             |
|           |                 | Ethernet16      | untagged       | 192.0.0.4             |             |
|           |                 |                 |                | 192.0.0.100           |             |
|           |                 |                 |                | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | Ethernet24      | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | Ethernet28      | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

show_vlan_brief_output_with_new_dhcp_relayv6_address="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | Ethernet4       | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | Ethernet8       | untagged       | 192.0.0.2             |             |
|           |                 | Ethernet12      | untagged       | 192.0.0.3             |             |
|           |                 | Ethernet16      | untagged       | 192.0.0.4             |             |
|           |                 |                 |                | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
|           |                 |                 |                | fc02:2000::3          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | Ethernet24      | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | Ethernet28      | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

config_add_del_vlan_and_vlan_member_output="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | Ethernet4       | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | Ethernet8       | untagged       | 192.0.0.2             |             |
|           |                 | Ethernet12      | untagged       | 192.0.0.3             |             |
|           |                 | Ethernet16      | untagged       | 192.0.0.4             |             |
|           |                 |                 |                | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      1001 |                 | Ethernet20      | untagged       |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | Ethernet24      | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | Ethernet28      | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""

config_add_del_vlan_and_vlan_member_in_alias_mode_output="""\
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|   VLAN ID | IP Address      | Ports           | Port Tagging   | DHCP Helper Address   | Proxy ARP   |
+===========+=================+=================+================+=======================+=============+
|      1000 | 192.168.0.1/21  | etp2            | untagged       | 192.0.0.1             | disabled    |
|           | fc02:1000::1/64 | etp3            | untagged       | 192.0.0.2             |             |
|           |                 | etp4            | untagged       | 192.0.0.3             |             |
|           |                 | etp5            | untagged       | 192.0.0.4             |             |
|           |                 |                 |                | fc02:2000::1          |             |
|           |                 |                 |                | fc02:2000::2          |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      1001 |                 | etp6            | untagged       |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      2000 | 192.168.0.10/21 | etp7            | untagged       | 192.0.0.1             | enabled     |
|           | fc02:1011::1/64 | etp8            | untagged       | 192.0.0.2             |             |
|           |                 |                 |                | 192.0.0.3             |             |
|           |                 |                 |                | 192.0.0.4             |             |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      3000 |                 |                 |                |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
|      4000 |                 | PortChannel1001 | tagged         |                       | disabled    |
+-----------+-----------------+-----------------+----------------+-----------------------+-------------+
"""
class TestVlan(object):
    @classmethod
    def setup_class(cls):
        os.environ['UTILITIES_UNIT_TESTING'] = "1"
        print("SETUP")

    def test_show_vlan(self):
        runner = CliRunner()
        result = runner.invoke(show.cli.commands["vlan"], [])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

    def test_show_vlan_brief(self):
        runner = CliRunner()
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_output

    def test_show_vlan_brief_verbose(self):
        runner = CliRunner()
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], ["--verbose"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_output

    def test_show_vlan_brief_in_alias_mode(self):
        runner = CliRunner()
        os.environ['SONIC_CLI_IFACE_MODE'] = "alias"
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"])
        os.environ['SONIC_CLI_IFACE_MODE'] = "default"
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_in_alias_mode_output

    def test_show_vlan_brief_explicit_proxy_arp_disable(self):
        runner = CliRunner()
        db = Db()

        db.cfgdb.set_entry("VLAN_INTERFACE", "Vlan1000", {"proxy_arp": "disabled"})

    def test_show_vlan_config(self):
        runner = CliRunner()
        result = runner.invoke(show.cli.commands["vlan"].commands["config"], [])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_config_output

    def test_show_vlan_config_in_alias_mode(self):
        runner = CliRunner()
        os.environ['SONIC_CLI_IFACE_MODE'] = "alias"
        result = runner.invoke(show.cli.commands["vlan"].commands["config"], [])
        os.environ['SONIC_CLI_IFACE_MODE'] = "default"
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_config_in_alias_mode_output

    def test_config_vlan_add_vlan_with_invalid_vlanid(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["add"], ["4096"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Invalid VLAN ID 4096 (1-4094)" in result.output

    def test_config_vlan_add_vlan_with_exist_vlanid(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["add"], ["1000"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Vlan1000 already exists" in result.output

    def test_config_vlan_del_vlan_with_invalid_vlanid(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["4096"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Invalid VLAN ID 4096 (1-4094)" in result.output

    def test_config_vlan_del_vlan_with_nonexist_vlanid(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1001"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Vlan1001 does not exist" in result.output

    def test_config_vlan_add_member_with_invalid_vlanid(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], ["4096", "Ethernet4"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Invalid VLAN ID 4096 (1-4094)" in result.output

    def test_config_vlan_add_member_with_nonexist_vlanid(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], ["1001", "Ethernet4"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Vlan1001 does not exist" in result.output

    def test_config_vlan_add_exist_port_member(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], ["1000", "Ethernet4"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Ethernet4 is already a member of Vlan1000" in result.output

    def test_config_vlan_add_nonexist_port_member(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], ["1000", "Ethernet3"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Ethernet3 does not exist" in result.output

    def test_config_vlan_add_nonexist_portchannel_member(self):
        runner = CliRunner()
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], \
				["1000", "PortChannel1011"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: PortChannel1011 does not exist" in result.output

    def test_config_vlan_add_portchannel_member(self):
        runner = CliRunner()
        db = Db()

        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], \
				["1000", "PortChannel1001", "--untagged"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_with_portchannel_output

    def test_config_vlan_add_rif_portchannel_member(self):
        runner = CliRunner()
        db = Db()

        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], \
				["1000", "PortChannel0001", "--untagged"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: PortChannel0001 is a router interface!" in result.output

    def test_config_vlan_del_vlan(self):
        runner = CliRunner()
        db = Db()
        obj = {'config_db':db.cfgdb}

        # del vlan with IP
        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1000"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Vlan1000 can not be removed. First remove IP addresses assigned to this VLAN" in result.output

        # remove vlan IP`s
        result = runner.invoke(config.config.commands["interface"].commands["ip"].commands["remove"], ["Vlan1000", "192.168.0.1/21"], obj=obj)
        print(result.exit_code, result.output)
        assert result.exit_code != 0

        result = runner.invoke(config.config.commands["interface"].commands["ip"].commands["remove"], ["Vlan1000", "fc02:1000::1/64"], obj=obj)
        print(result.exit_code, result.output)
        assert result.exit_code != 0

        # del vlan with IP
        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1000"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: VLAN ID 1000 can not be removed. First remove all members assigned to this VLAN." in result.output

        vlan_member = db.cfgdb.get_table('VLAN_MEMBER')
        keys = [ (k, v) for k, v in vlan_member if k == 'Vlan{}'.format(1000) ]
        for k,v in keys:    
            result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["del"], ["1000", v], obj=db)
            print(result.exit_code)
            print(result.output)
            assert result.exit_code == 0

        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1000"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_empty_output

    def test_config_vlan_del_nonexist_vlan_member(self):
        runner = CliRunner()

        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["del"], \
				["1000", "Ethernet0"])
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Ethernet0 is not a member of Vlan1000" in result.output

    def test_config_add_del_vlan_and_vlan_member(self):
        runner = CliRunner()
        db = Db()

        # add vlan 1001
        result = runner.invoke(config.config.commands["vlan"].commands["add"], ["1001"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # add Ethernet20 to vlan 1001
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"],
                ["1001", "Ethernet20", "--untagged"], obj=db)
        print(result.exit_code)
        print(result.output)
        traceback.print_tb(result.exc_info[2])
        assert result.exit_code == 0

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.output)
        assert result.output == config_add_del_vlan_and_vlan_member_output

        # remove vlan member
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["del"],
                ["1001", "Ethernet20"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # add del 1001
        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1001"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_output

    def test_config_add_del_vlan_and_vlan_member_in_alias_mode(self):
        runner = CliRunner()
        db = Db()

        os.environ['SONIC_CLI_IFACE_MODE'] = "alias"

        # add vlan 1001
        result = runner.invoke(config.config.commands["vlan"].commands["add"], ["1001"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # add etp6 to vlan 1001
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"],
                ["1001", "etp6", "--untagged"], obj=db)
        print(result.exit_code)
        print(result.output)
        traceback.print_tb(result.exc_info[2])
        assert result.exit_code == 0

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.output == config_add_del_vlan_and_vlan_member_in_alias_mode_output

        # remove vlan member
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["del"],
                ["1001", "etp6"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # add del 1001
        result = runner.invoke(config.config.commands["vlan"].commands["del"], ["1001"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code == 0
        assert result.output == show_vlan_brief_in_alias_mode_output

        os.environ['SONIC_CLI_IFACE_MODE'] = "default"

    def test_config_vlan_add_dhcp_relay_with_nonexist_vlanid(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1001", "192.0.0.100"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: Vlan1001 doesn't exist" in result.output
            assert mock_run_command.call_count == 0

    def test_config_vlan_add_dhcp_relay_with_invalid_vlanid(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["4096", "192.0.0.100"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: Vlan4096 doesn't exist" in result.output
            assert mock_run_command.call_count == 0

    def test_config_vlan_add_dhcp_relay_with_invalid_ip(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1000", "192.0.0.1000"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: 192.0.0.1000 is invalid IP address" in result.output
            assert mock_run_command.call_count == 0
    
    def test_config_vlan_add_dhcp_relay_with_invalid_ipv6(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1000", "fe80:2030:31:24"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: fe80:2030:31:24 is invalid IP address" in result.output
            assert mock_run_command.call_count == 0

    def test_config_vlan_add_dhcp_relay_with_exist_ip(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1000", "192.0.0.1"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code == 0
            assert "192.0.0.1 is already a DHCP relay destination for Vlan1000" in result.output
            assert mock_run_command.call_count == 0
    
    def test_config_vlan_add_dhcp_relay_with_exist_ipv6(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1000", "fc02:2000::1"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code == 0
            assert "fc02:2000::1 is already a DHCP relay destination for Vlan1000" in result.output
            assert mock_run_command.call_count == 0

    def test_config_vlan_add_del_dhcp_relay_dest(self):
        runner = CliRunner()
        db = Db()

        # add new relay dest
        with mock.patch("utilities_common.cli.run_command") as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1000", "192.0.0.100"], obj=db)
            print(result.exit_code)
            print(result.output)
            assert result.exit_code == 0
            assert result.output == config_vlan_add_dhcp_relay_output
            assert mock_run_command.call_count == 3

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.output)
        assert result.output == show_vlan_brief_output_with_new_dhcp_relay_address

        # del relay dest
        with mock.patch("utilities_common.cli.run_command") as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["del"],
                                   ["1000", "192.0.0.100"], obj=db)
            print(result.exit_code)
            print(result.output)
            assert result.exit_code == 0
            assert result.output == config_vlan_del_dhcp_relay_output
            assert mock_run_command.call_count == 3

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.output)
        assert result.output == show_vlan_brief_output

    def test_config_vlan_add_del_dhcp_relayv6_dest(self):
        runner = CliRunner()
        db = Db()

        # add new relay dest
        with mock.patch("utilities_common.cli.run_command") as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["add"],
                                   ["1000", "fc02:2000::3"], obj=db)
            print(result.exit_code)
            print(result.output)
            assert result.exit_code == 0
            assert result.output == config_vlan_add_dhcp_relayv6_output
            assert mock_run_command.call_count == 3

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.output)
        assert result.output == show_vlan_brief_output_with_new_dhcp_relayv6_address

        # del relay dest
        with mock.patch("utilities_common.cli.run_command") as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["del"],
                                   ["1000", "fc02:2000::3"], obj=db)
            print(result.exit_code)
            print(result.output)
            assert result.exit_code == 0
            assert result.output == config_vlan_del_dhcp_relayv6_output
            assert mock_run_command.call_count == 3

        # show output
        result = runner.invoke(show.cli.commands["vlan"].commands["brief"], [], obj=db)
        print(result.output)
        assert result.output == show_vlan_brief_output

    def test_config_vlan_remove_nonexist_dhcp_relay_dest(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["del"],
                                   ["1000", "192.0.0.100"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: 192.0.0.100 is not a DHCP relay destination for Vlan1000" in result.output
            assert mock_run_command.call_count == 0
    
    def test_config_vlan_remove_nonexist_dhcp_relayv6_dest(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["del"],
                                   ["1000", "fc02:2000::3"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: fc02:2000::3 is not a DHCP relay destination for Vlan1000" in result.output
            assert mock_run_command.call_count == 0

    def test_config_vlan_remove_dhcp_relay_dest_with_nonexist_vlanid(self):
        runner = CliRunner()

        with mock.patch('utilities_common.cli.run_command') as mock_run_command:
            result = runner.invoke(config.config.commands["vlan"].commands["dhcp_relay"].commands["del"],
                                   ["1001", "192.0.0.1"])
            print(result.exit_code)
            print(result.output)
            # traceback.print_tb(result.exc_info[2])
            assert result.exit_code != 0
            assert "Error: Vlan1001 doesn't exist" in result.output
            assert mock_run_command.call_count == 0

    def test_config_vlan_proxy_arp_with_nonexist_vlan_intf_table(self):
        modes = ["enabled", "disabled"]
        runner = CliRunner()
        db = Db()
        db.cfgdb.delete_table("VLAN_INTERFACE")

        for mode in modes:
            result = runner.invoke(config.config.commands["vlan"].commands["proxy_arp"], ["1000", mode], obj=db)

            print(result.exit_code)
            print(result.output)

            assert result.exit_code != 0
            assert "Interface Vlan1000 does not exist" in result.output

    def test_config_vlan_proxy_arp_with_nonexist_vlan_intf(self):
        modes = ["enabled", "disabled"]
        runner = CliRunner()
        db = Db()

        for mode in modes:
            result = runner.invoke(config.config.commands["vlan"].commands["proxy_arp"], ["1001", mode], obj=db)

            print(result.exit_code)
            print(result.output)

            assert result.exit_code != 0
            assert "Interface Vlan1001 does not exist" in result.output

    def test_config_vlan_proxy_arp_enable(self):
        runner = CliRunner()
        db = Db()

        result = runner.invoke(config.config.commands["vlan"].commands["proxy_arp"], ["1000", "enabled"], obj=db)

        print(result.exit_code)
        print(result.output)

        assert result.exit_code == 0 
        assert db.cfgdb.get_entry("VLAN_INTERFACE", "Vlan1000") == {"proxy_arp": "enabled"}

    def test_config_vlan_proxy_arp_disable(self):
        runner = CliRunner()
        db = Db()

        result = runner.invoke(config.config.commands["vlan"].commands["proxy_arp"], ["2000", "disabled"], obj=db)

        print(result.exit_code)
        print(result.output)

        assert result.exit_code == 0
        assert db.cfgdb.get_entry("VLAN_INTERFACE", "Vlan2000") == {"proxy_arp": "disabled"}
        
    def test_config_2_untagged_vlan_on_same_interface(self):
        runner = CliRunner()
        db = Db()
        
        # add Ethernet4 to vlan 2000 as untagged - should fail as ethrnet4 is already untagged member in 1000
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"],
                ["2000", "Ethernet4", "--untagged"], obj=db)
        print(result.exit_code)
        assert result.exit_code != 0
        
        # add Ethernet4 to vlan 2000 as tagged - should succeed
        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"],
                ["2000", "Ethernet4" ], obj=db)
        print(result.exit_code)
        assert result.exit_code == 0
        
    def test_config_set_router_port_on_member_interface(self):        
        db = Db()
        runner = CliRunner()
        obj = {'config_db':db.cfgdb}
        
        # intf enable
        result = runner.invoke(config.config.commands["interface"].commands["ip"].commands["add"],
                               ["Ethernet4", "10.10.10.1/24"], obj=obj)        
        print(result.exit_code, result.output)
        assert result.exit_code == 0
        assert 'Interface Ethernet4 is a member of vlan' in result.output
        
    def test_config_vlan_add_member_of_portchannel(self):
        runner = CliRunner()
        db = Db()

        result = runner.invoke(config.config.commands["vlan"].commands["member"].commands["add"], \
				["1000", "Ethernet32", "--untagged"], obj=db)
        print(result.exit_code)
        print(result.output)
        assert result.exit_code != 0
        assert "Error: Ethernet32 is part of portchannel!" in result.output

    @classmethod
    def teardown_class(cls):
        os.environ['UTILITIES_UNIT_TESTING'] = "0"
        print("TEARDOWN")