Skip to content

Commit

Permalink
Added cisco config platform commands
Browse files Browse the repository at this point in the history
Signed-off-by: Yucai Gu <yucgu@cisco.com>
  • Loading branch information
yucgu committed Aug 4, 2022
1 parent 6995cba commit bafc9e3
Show file tree
Hide file tree
Showing 2 changed files with 46 additions and 0 deletions.
28 changes: 28 additions & 0 deletions tests/config_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -1585,3 +1585,31 @@ def test_config_rate(self, get_cmd_module, setup_single_broadcom_asic):
def teardown_class(cls):
print("TEARDOWN")
os.environ['UTILITIES_UNIT_TESTING'] = "0"

class TestConfigPlatorm(object):
@classmethod
def setup_class(cls):
os.environ['UTILITIES_UNIT_TESTING'] = "1"
print("SETUP")

import config.main
importlib.reload(config.main)

def test_config_platform_cisco(self, get_cmd_module, setup_single_cisco_asic):
with mock.patch("utilities_common.cli.run_command", mock.MagicMock(side_effect=mock_run_command_side_effect)) as mock_run_command:
(config, show) = get_cmd_module

runner = CliRunner()
result = runner.invoke(config.config.commands["platform"], ["cisco"])

print(result.exit_code)
print(result.output)
traceback.print_tb(result.exc_info[2])

assert result.exit_code == 0
assert result.output == ""

@classmethod
def teardown_class(cls):
print("TEARDOWN")
os.environ['UTILITIES_UNIT_TESTING'] = "0"
18 changes: 18 additions & 0 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,24 @@ def setup_qos_mock_apis():
return_value=os.path.join(cwd, "qos_config_input/sonic_version.yml")
)

@pytest.fixture
def set_cisco_mock_apis():
import config.main as config
cwd = os.path.dirname(os.path.realpath(__file__))
config.asic_type = mock.MagicMock(return_value="cisco-8000")
config._get_device_type = mock.MagicMock(return_value="ToRRouter")

@pytest.fixture
def setup_single_cisco_asic():
import config.main as config
import show.main as show

set_cisco_mock_apis()
device_info.get_num_npus = mock.MagicMock(return_value=1)
config._get_sonic_generated_services = \
mock.MagicMock(return_value=(generated_services_list, []))


@pytest.fixture
def setup_single_broadcom_asic():
import config.main as config
Expand Down

0 comments on commit bafc9e3

Please sign in to comment.