From bafc9e30097e113dae92a29649d88cb97651915d Mon Sep 17 00:00:00 2001 From: Yucai Gu Date: Wed, 3 Aug 2022 17:16:36 -0700 Subject: [PATCH] Added cisco config platform commands Signed-off-by: Yucai Gu --- tests/config_test.py | 28 ++++++++++++++++++++++++++++ tests/conftest.py | 18 ++++++++++++++++++ 2 files changed, 46 insertions(+) diff --git a/tests/config_test.py b/tests/config_test.py index e9dbae4194..1405f8443b 100644 --- a/tests/config_test.py +++ b/tests/config_test.py @@ -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" diff --git a/tests/conftest.py b/tests/conftest.py index b48d0ef93f..191982f31a 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -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