44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
|
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
|
||
|
|
-------------- -----------
|
||
|
|
Solicit
|
||
|
|
Advertise
|
||
|
|
Request
|
||
|
|
Confirm
|
||
|
|
Renew
|
||
|
|
Rebind
|
||
|
|
Reply
|
||
|
|
Release
|
||
|
|
Decline
|
||
|
|
Relay-Forward
|
||
|
|
Relay-Reply
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
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
|
||
|
|
|