56533ceb9e
#### Why I did it - Adapt config/show CLI commands to support DHCPv6 relay - Support multiple dhcp servers assignment in one command - Fix IP validation - Adapt UT and add new UT cases #### How I did it - Modify config/show dhcp relay files - Modify config/show UT files #### How to verify it This PR has a dependency on PR https://github.com/Azure/sonic-utilities/pull/1717 Build an image with the dependent PR and this PR Use config/show DHCPv6 relay commands.
29 lines
794 B
Python
29 lines
794 B
Python
import os
|
|
import sys
|
|
import traceback
|
|
from unittest import mock
|
|
|
|
from click.testing import CliRunner
|
|
|
|
import show.vlan as vlan
|
|
from utilities_common.db import Db
|
|
|
|
sys.path.insert(0, '../cli/show/plugins/')
|
|
import show_dhcp_relay
|
|
|
|
|
|
class TestVlanDhcpRelay(object):
|
|
def test_plugin_registration(self):
|
|
cli = mock.MagicMock()
|
|
show_dhcp_relay.register(cli)
|
|
assert 'DHCP Helper Address' in dict(vlan.VlanBrief.COLUMNS)
|
|
|
|
def test_dhcp_relay_column_output(self):
|
|
ctx = (
|
|
({'Vlan100': {'dhcp_servers': ['192.0.0.1', '192.168.0.2'], 'dhcpv6_servers': ['fc02:2000::1', 'fc02:2000::2']}}, {}, {}),
|
|
(),
|
|
)
|
|
assert show_dhcp_relay.get_dhcp_helper_address(ctx, 'Vlan100') == '192.0.0.1\n192.168.0.2\nfc02:2000::1\nfc02:2000::2'
|
|
|
|
|