jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-27 20:42:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests

Browse Source
This commit is contained in:
James Maslek
2023-08-14 11:40:39 -04:00
parent 3ee60e0fc9
commit f4460fbf18
4 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
openbb_terminal/econometrics/econometrics_controller.py
View File

@@ -189,6 +189,7 @@ class EconometricsController(BaseController):
"corr",
"lag",
"vif",
"panel",
]:
choices[feature] = dict()

32
tests/openbb_terminal/session/test_hub_model.py
View File

@@ -233,7 +233,9 @@ def test_get_session_success():
) as create_session_mock:
result = hub_model.get_session("email", "password")
assert result == {"session": "info"}
create_session_mock.assert_called_once_with("email", "password")
create_session_mock.assert_called_once_with(
"email", "password", base_url="https://payments.openbb.co/"
)
def test_get_session_401():
@@ -246,7 +248,9 @@ def test_get_session_401():
) as create_session_mock:
result = hub_model.get_session("email", "password")
assert result == {}
create_session_mock.assert_called_once_with("email", "password")
create_session_mock.assert_called_once_with(
"email", "password", base_url="https://payments.openbb.co/"
)
def test_get_session_403():
@@ -259,7 +263,9 @@ def test_get_session_403():
) as create_session_mock:
result = hub_model.get_session("email", "password")
assert result == {}
create_session_mock.assert_called_once_with("email", "password")
create_session_mock.assert_called_once_with(
"email", "password", base_url="https://payments.openbb.co/"
)
def test_get_session_failed_to_request():
@@ -268,7 +274,9 @@ def test_get_session_failed_to_request():
) as create_session_mock:
result = hub_model.get_session("email", "password")
assert result == {}
create_session_mock.assert_called_once_with("email", "password")
create_session_mock.assert_called_once_with(
"email", "password", base_url="https://payments.openbb.co/"
)
def test_get_session_from_token():
@@ -282,7 +290,9 @@ def test_get_session_from_token():
) as create_session_mock:
result = hub_model.get_session_from_token("token")
assert result == {"session": "info"}
create_session_mock.assert_called_once_with("token")
create_session_mock.assert_called_once_with(
"token", base_url="https://payments.openbb.co/"
)
def test_get_session_from_token_401():
@@ -295,7 +305,9 @@ def test_get_session_from_token_401():
) as create_session_mock:
result = hub_model.get_session_from_token("token")
assert result == {}
create_session_mock.assert_called_once_with("token")
create_session_mock.assert_called_once_with(
"token", base_url="https://payments.openbb.co/"
)
def test_get_session_from_token_403():
@@ -308,7 +320,9 @@ def test_get_session_from_token_403():
) as create_session_mock:
result = hub_model.get_session_from_token("token")
assert result == {}
create_session_mock.assert_called_once_with("token")
create_session_mock.assert_called_once_with(
"token", base_url="https://payments.openbb.co/"
)
def test_get_session_from_token_failed_to_request():
@@ -318,7 +332,9 @@ def test_get_session_from_token_failed_to_request():
) as create_session_mock:
result = hub_model.get_session_from_token("token")
assert result == {}
create_session_mock.assert_called_once_with("token")
create_session_mock.assert_called_once_with(
"token", base_url="https://payments.openbb.co/"
)
@pytest.mark.parametrize("token_type, access_token", [("TokenType", "AccessToken")])

4
tests/openbb_terminal/session/test_routines_handler.py
View File

@@ -234,6 +234,7 @@ def test_download_routine(mocker, test_user, silent=False):
fields=["name", "script"],
page=1,
size=100,
base_url="https://payments.openbb.co/",
silent=silent,
)
@@ -272,6 +273,7 @@ def test_download_default_routine_exception(mocker, test_user, silent=False):
fields=["name", "script"],
page=1,
size=100,
base_url="https://payments.openbb.co/",
silent=silent,
)
@@ -310,6 +312,7 @@ def test_download_personal_routine_exception(mocker, test_user, silent=False):
fields=["name", "script"],
page=1,
size=100,
base_url="https://payments.openbb.co/",
silent=silent,
)
@@ -339,5 +342,6 @@ def test_download_default_and_personal_routine_exception(
fields=["name", "script"],
page=1,
size=100,
base_url="https://payments.openbb.co/",
silent=silent,
)

17
tests/openbb_terminal/session/test_session_model.py
View File

@@ -75,7 +75,9 @@ def test_login_no_response(mocker):
assert session_model.login(TEST_SESSION) == session_model.LoginStatus.NO_RESPONSE
mock_fetch_user_configs.assert_called_once_with(TEST_SESSION)
mock_fetch_user_configs.assert_called_once_with(
TEST_SESSION, "https://payments.openbb.co/"
)
mock_apply_configs.assert_not_called()
mock_update_flair.assert_not_called()
mock_get_updated_hub_sources.assert_not_called()
@@ -96,7 +98,9 @@ def test_login_fail_response(mocker):
assert session_model.login(TEST_SESSION) == session_model.LoginStatus.FAILED
mock_fetch_user_configs.assert_called_once_with(TEST_SESSION)
mock_fetch_user_configs.assert_called_once_with(
TEST_SESSION, "https://payments.openbb.co/"
)
mock_apply_configs.assert_not_called()
mock_update_flair.assert_not_called()
mock_get_updated_hub_sources.assert_not_called()
@@ -119,7 +123,9 @@ def test_login_success_response(mocker):
assert session_model.login(TEST_SESSION) == session_model.LoginStatus.SUCCESS
mock_fetch_user_configs.assert_called_once_with(TEST_SESSION)
mock_fetch_user_configs.assert_called_once_with(
TEST_SESSION, "https://payments.openbb.co/"
)
mock_apply_configs.assert_called_once()
mock_update_flair.assert_called_once()
@@ -142,10 +148,13 @@ def test_logout_user(mocker, guest):
auth_header = "Bearer test_token"
token = "test_token"
base_url = "https://payments.openbb.co/"
session_model.logout(auth_header, token)
if not guest:
mock_delete_session.assert_called_once_with(auth_header, token)
mock_delete_session.assert_called_once_with(
auth_header, token, base_url=base_url
)
assert mock_remove.call_args_list[0] == mocker.call(SESSION_FILE_PATH)
assert mock_remove.call_args_list[1] == mocker.call(HIST_FILE_PATH)
assert mock_remove.call_args_list[2] == mocker.call(