diff --git a/openbb_terminal/econometrics/econometrics_controller.py b/openbb_terminal/econometrics/econometrics_controller.py index a429a7230ff..5229e030e5a 100644 --- a/openbb_terminal/econometrics/econometrics_controller.py +++ b/openbb_terminal/econometrics/econometrics_controller.py @@ -189,6 +189,7 @@ class EconometricsController(BaseController): "corr", "lag", "vif", + "panel", ]: choices[feature] = dict() diff --git a/tests/openbb_terminal/session/test_hub_model.py b/tests/openbb_terminal/session/test_hub_model.py index 489ccae660f..008325c9300 100644 --- a/tests/openbb_terminal/session/test_hub_model.py +++ b/tests/openbb_terminal/session/test_hub_model.py @@ -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")]) diff --git a/tests/openbb_terminal/session/test_routines_handler.py b/tests/openbb_terminal/session/test_routines_handler.py index 566c275fd10..e62c0fc1b0e 100644 --- a/tests/openbb_terminal/session/test_routines_handler.py +++ b/tests/openbb_terminal/session/test_routines_handler.py @@ -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, ) diff --git a/tests/openbb_terminal/session/test_session_model.py b/tests/openbb_terminal/session/test_session_model.py index 25861d3c6bf..138c9af8a6f 100644 --- a/tests/openbb_terminal/session/test_session_model.py +++ b/tests/openbb_terminal/session/test_session_model.py @@ -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(