mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 12:40:35 +00:00
b7d30c22dd
* Updating dependencies and putting kaleido into a shibajail * Downgrading jinja2 to accomodate papermill Co-authored-by: Theodore Aptekarev <aptekarev@gmail.com>
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from contextlib import contextmanager
|
|
import pytest
|
|
import terminal
|
|
|
|
|
|
@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.obbff, "ENABLE_QUICK_EXIT", True)
|
|
monkeypatch.setattr(terminal.obbff, "USE_ION", False)
|
|
monkeypatch.setattr(terminal.obbff, "USE_PROMPT_TOOLKIT", True)
|
|
|
|
mocker.patch("sys.stdin")
|
|
|
|
terminal.terminal()
|
|
|
|
|
|
@pytest.mark.skip
|
|
@pytest.mark.vcr(record_mode="none")
|
|
@pytest.mark.record_stdout
|
|
def test_terminal_quit(mocker, monkeypatch):
|
|
monkeypatch.setattr(terminal.obbff, "ENABLE_QUICK_EXIT", False)
|
|
monkeypatch.setattr(terminal.obbff, "USE_ION", False)
|
|
monkeypatch.setattr(terminal.obbff, "USE_PROMPT_TOOLKIT", True)
|
|
|
|
mocker.patch("sys.stdin")
|
|
mocker.patch("builtins.input", return_value="e")
|
|
mocker.patch("terminal.session")
|
|
mocker.patch("terminal.session.prompt", return_value="e")
|
|
mocker.patch("terminal.print_goodbye")
|
|
spy_print_goodbye = mocker.spy(terminal, "print_goodbye")
|
|
|
|
terminal.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, test, verbose, filtert, path",
|
|
[
|
|
(True, False, False, None, None),
|
|
(False, False, False, None, ["scripts/test_alt_covid.openbb"]),
|
|
# (False, True, False, "alt_covid", ["scripts/"]),
|
|
# (False, True, True, "alt_covid", ["scripts/"]),
|
|
],
|
|
)
|
|
def test_menu(mocker, debug, test, filtert, path, verbose):
|
|
mocker.patch(target="terminal.terminal")
|
|
mocker.patch(target="terminal.suppress_stdout", side_effect=no_suppress)
|
|
terminal.main(debug, test, filtert, path, verbose)
|