| """Test the CLI module.""" |
|
|
| from unittest.mock import patch |
|
|
| from openbb_cli.cli import main |
|
|
|
|
| @patch("openbb_cli.config.setup.bootstrap") |
| @patch("openbb_cli.controllers.cli_controller.launch") |
| @patch("sys.argv", ["openbb", "--dev", "--debug"]) |
| def test_main_with_dev_and_debug(mock_launch, mock_bootstrap): |
| """Test the main function with dev and debug flags.""" |
| main() |
| mock_bootstrap.assert_called_once() |
| mock_launch.assert_called_once_with(True, True) |
|
|
|
|
| @patch("openbb_cli.config.setup.bootstrap") |
| @patch("openbb_cli.controllers.cli_controller.launch") |
| @patch("sys.argv", ["openbb"]) |
| def test_main_without_arguments(mock_launch, mock_bootstrap): |
| """Test the main function without arguments.""" |
| main() |
| mock_bootstrap.assert_called_once() |
| mock_launch.assert_called_once_with(False, False) |
|
|
|
|
| @patch("openbb_cli.config.setup.bootstrap") |
| @patch("openbb_cli.controllers.cli_controller.launch") |
| @patch("sys.argv", ["openbb", "--dev"]) |
| def test_main_with_dev_only(mock_launch, mock_bootstrap): |
| """Test the main function with dev flag only.""" |
| main() |
| mock_bootstrap.assert_called_once() |
| mock_launch.assert_called_once_with(True, False) |
|
|
|
|
| @patch("openbb_cli.config.setup.bootstrap") |
| @patch("openbb_cli.controllers.cli_controller.launch") |
| @patch("sys.argv", ["openbb", "--debug"]) |
| def test_main_with_debug_only(mock_launch, mock_bootstrap): |
| """Test the main function with debug flag only.""" |
| main() |
| mock_bootstrap.assert_called_once() |
| mock_launch.assert_called_once_with(False, True) |
|
|
