sonic-buildimage/dockers/docker-dhcp-relay/cli-plugin-tests/test_show_dhcp6relay_counters.py

import sys
import os
from unittest import mock
sys.path.append('../cli/show/plugins/')
import show_dhcp_relay as show

from click.testing import CliRunner

try:
    modules_path = os.path.join(os.path.dirname(__file__), "../../../src/sonic-utilities")
    test_path = os.path.join(modules_path, "tests")
    mock_table_path = os.path.join(test_path, "mock_tables")
    sys.path.insert(0, modules_path)
    sys.path.insert(0, test_path)
    sys.path.insert(0, mock_table_path)
    import dbconnector
except KeyError:
    pass

expected_counts = """\
       Message Type     Vlan1000
-------------------  -----------
            Unknown
            Solicit
          Advertise
            Request
            Confirm
              Renew
             Rebind
              Reply
            Release
            Decline
        Reconfigure
Information-Request
      Relay-Forward
        Relay-Reply
          Malformed

"""

class TestDhcp6RelayCounters(object):

    def test_show_counts(self):           
        runner = CliRunner()
        result = runner.invoke(show.dhcp6relay_counters.commands["counts"], ["-i Vlan1000"])
        assert result.output == expected_counts