mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
ee34a1f278
* Fixed isort * Fixed issues * FIxed linters --------- Co-authored-by: James Maslek <jmaslek11@gmail.com>
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
from contextlib import contextmanager
|
|
|
|
import pytest
|
|
|
|
from openbb_terminal import terminal_controller
|
|
|
|
|
|
@pytest.mark.skip
|
|
@pytest.mark.block_network
|
|
@pytest.mark.vcr(record_mode="none")
|
|
@pytest.mark.record_stdout
|
|
def test_terminal_quick_exit(mocker, monkeypatch):
|
|
monkeypatch.setattr(terminal_controller.obbff, "ENABLE_QUICK_EXIT", True)
|
|
monkeypatch.setattr(terminal_controller.obbff, "USE_ION", False)
|
|
monkeypatch.setattr(terminal_controller.obbff, "USE_PROMPT_TOOLKIT", True)
|
|
|
|
mocker.patch("sys.stdin")
|
|
|
|
terminal_controller.terminal()
|
|
|
|
|
|
@pytest.mark.skip
|
|
@pytest.mark.vcr(record_mode="none")
|
|
@pytest.mark.record_stdout
|
|
def test_terminal_quit(mocker, monkeypatch):
|
|
monkeypatch.setattr(terminal_controller.obbff, "ENABLE_QUICK_EXIT", False)
|
|
monkeypatch.setattr(terminal_controller.obbff, "USE_ION", False)
|
|
monkeypatch.setattr(terminal_controller.obbff, "USE_PROMPT_TOOLKIT", True)
|
|
|
|
mocker.patch("sys.stdin")
|
|
mocker.patch("builtins.input", return_value="e")
|
|
mocker.patch("terminal_controller.session")
|
|
mocker.patch("terminal_controller.session.prompt", return_value="e")
|
|
mocker.patch("terminal_controller.print_goodbye")
|
|
spy_print_goodbye = mocker.spy(terminal_controller, "print_goodbye")
|
|
|
|
terminal_controller.terminal()
|
|
|
|
spy_print_goodbye.assert_called_once()
|
|
|
|
|
|
@contextmanager
|
|
def no_suppress():
|
|
yield
|
|
|
|
|
|
@pytest.mark.skip
|
|
@pytest.mark.vcr(record_mode="none")
|
|
@pytest.mark.record_stdout
|
|
@pytest.mark.parametrize(
|
|
"debug, path",
|
|
[(True, None), (False, ["scripts/test_alt_covid.openbb"])],
|
|
)
|
|
def test_menu(mocker, debug, path):
|
|
mocker.patch(target="terminal_controller.terminal")
|
|
mocker.patch(target="terminal_controller.suppress_stdout", side_effect=no_suppress)
|
|
terminal_controller.main(debug, path)
|