mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 21:12:16 +00:00
8a70c6205d
* Update .gitignore * init * Update backend.py * Update main.tsx * Update no_import.py * Update session_controller.py * indents * adjust window size * update keys * re-add save session for pywry login * pop response keys on rename * walrus remember * use session username for profile * implement logout clearing of cookies * don't display welcome when logging in from `account` menu * Update session_model.py * Update session_model.py * customdata handler for charts * fix remember key * only slice data if chandlestick trace in data * Update profile_model.py * Update streamlit_run.py * Update plotly.html * test * Update loggers.py * updates * Update session_controller.py * Update session_controller.py * no longer need to call_hub for logout * tests * Update test_session_controller.py * Update profile_model.py * bump pywry, fix numbers passed as strings tables * update plotly frontend packages * formatting * Update backend.py * Update backend.py * change to prod * return on image export --------- Co-authored-by: James Maslek <jmaslek11@gmail.com>
122 lines
4.2 KiB
Python
122 lines
4.2 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from openbb_terminal import terminal_controller
|
|
from openbb_terminal.core.session import session_controller
|
|
|
|
|
|
@pytest.mark.record_stdout
|
|
def test_display_welcome_message():
|
|
session_controller.display_welcome_message()
|
|
|
|
|
|
@pytest.mark.record_stdout
|
|
@patch("openbb_terminal.core.session.session_controller.PromptSession")
|
|
def test_get_user_input(mock_prompt_session):
|
|
mock_prompt_session().prompt.side_effect = ["test@example.com", "password", "n"]
|
|
|
|
result = session_controller.get_user_input()
|
|
|
|
assert result == ("test@example.com", "password", False)
|
|
|
|
|
|
@pytest.mark.record_stdout
|
|
@patch("openbb_terminal.core.session.session_controller.PromptSession")
|
|
def test_get_user_input_with_save(mock_prompt_session):
|
|
mock_prompt_session().prompt.side_effect = ["test@example.com", "password", "y"]
|
|
|
|
result = session_controller.get_user_input()
|
|
|
|
assert result == ("test@example.com", "password", True)
|
|
|
|
|
|
def test_prompt():
|
|
with patch(
|
|
"openbb_terminal.core.session.session_controller.get_user_input",
|
|
return_value=("email", "password", False),
|
|
) as get_user_input_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.create_session",
|
|
return_value={"session": "session"},
|
|
) as create_session_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.login_and_launch",
|
|
return_value=True,
|
|
):
|
|
session_controller.prompt_cli()
|
|
assert get_user_input_mock.call_count == 1
|
|
create_session_mock.assert_called_once_with("email", "password", False)
|
|
assert create_session_mock.call_count == 1
|
|
|
|
|
|
def test_launch_terminal(mocker):
|
|
parse_args_mock = mocker.patch.object(
|
|
target=terminal_controller,
|
|
attribute="parse_args_and_run",
|
|
new=MagicMock(),
|
|
)
|
|
session_controller.launch_terminal()
|
|
assert parse_args_mock.call_count == 1
|
|
|
|
|
|
def test_login_and_launch():
|
|
with patch(
|
|
"openbb_terminal.core.session.session_controller.login",
|
|
return_value=session_controller.LoginStatus.SUCCESS,
|
|
) as login_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.launch_terminal",
|
|
return_value=True,
|
|
) as launch_terminal_mock:
|
|
session_controller.login_and_launch(session={})
|
|
assert login_mock.call_count == 1
|
|
assert launch_terminal_mock.call_count == 1
|
|
|
|
|
|
def test_login_and_launch_failed():
|
|
with patch(
|
|
"openbb_terminal.core.session.session_controller.login",
|
|
return_value=session_controller.LoginStatus.FAILED,
|
|
) as login_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.prompt_cli", return_value=True
|
|
) as prompt_mock:
|
|
session_controller.login_and_launch(session={})
|
|
assert login_mock.call_count == 1
|
|
assert prompt_mock.call_count == 1
|
|
|
|
|
|
@pytest.mark.skip(reason="To review")
|
|
def test_login_and_launch_no_response():
|
|
with patch(
|
|
"openbb_terminal.core.session.session_controller.login",
|
|
return_value=session_controller.LoginStatus.NO_RESPONSE,
|
|
) as login_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.prompt_cli", return_value=True
|
|
) as prompt_mock:
|
|
session_controller.login_and_launch(session={})
|
|
assert login_mock.call_count == 1
|
|
assert prompt_mock.call_count == 1
|
|
|
|
|
|
def test_main_local_session():
|
|
with patch(
|
|
"openbb_terminal.core.session.session_controller.prompt_cli", return_value=True
|
|
) as prompt_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.Local.get_session",
|
|
return_value=False,
|
|
) as get_session_mock:
|
|
session_controller.main()
|
|
assert prompt_mock.call_count == 1
|
|
assert get_session_mock.call_count == 1
|
|
|
|
|
|
def test_main_no_local_session():
|
|
with patch(
|
|
"openbb_terminal.core.session.session_controller.login_and_launch",
|
|
return_value=True,
|
|
) as login_and_launch_mock, patch(
|
|
"openbb_terminal.core.session.session_controller.Local.get_session",
|
|
return_value=True,
|
|
) as get_session_mock:
|
|
session_controller.main()
|
|
assert login_and_launch_mock.call_count == 1
|
|
assert get_session_mock.call_count == 1
|