mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-01 15:01:18 +00:00
937d033160
* adding BaseModel * initial version of the system model * creating setup functions and removing constants from config_terminal * removing log_supress const * fix test * merge * system handler * fix import * centralized configs setup * repr to use the BaseModel * updating i18n dict location * ignoring warning on wrong type-var * ruff * Update openbb_terminal/core/models/system_model.py Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> * Update openbb_terminal/core/models/system_model.py Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> * import literal * loading env files before the plot backend is called * fix tests/openbb_terminal/test_terminal_helper.py * tests need to have i18n initialized * fix seeking_alpha tests --------- Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com>
93 lines
2.4 KiB
Python
93 lines
2.4 KiB
Python
import sys
|
|
import unittest
|
|
from contextlib import contextmanager
|
|
|
|
import pytest
|
|
|
|
from openbb_terminal import terminal_helper
|
|
from tests.helpers.helpers import check_print
|
|
|
|
|
|
def return_val(x, shell, check):
|
|
# pylint: disable=unused-argument
|
|
# pylint: disable=R0903
|
|
class ReturnVal:
|
|
def __init__(self, code):
|
|
self.returncode = code
|
|
|
|
return ReturnVal(2)
|
|
|
|
|
|
@contextmanager
|
|
def replace_stdin(target):
|
|
orig = sys.stdin
|
|
sys.stdin = target
|
|
yield
|
|
sys.stdin = orig
|
|
|
|
|
|
class TestMainHelper(unittest.TestCase):
|
|
@check_print(length=0)
|
|
def test_print_goodbye(self):
|
|
terminal_helper.print_goodbye()
|
|
|
|
@check_print(assert_in="Welcome to OpenBB Terminal")
|
|
def test_welcome_message(self):
|
|
terminal_helper.welcome_message()
|
|
|
|
# @check_print(assert_in="Unfortunately, resetting wasn't")
|
|
# @patch("subprocess.run", side_effect=return_val)
|
|
# def test_reset(self, mock):
|
|
# # pylint: disable=unused-argument
|
|
# terminal_helper.reset()
|
|
|
|
|
|
@pytest.mark.vcr(record_mode="none")
|
|
@pytest.mark.record_stdout
|
|
@pytest.mark.parametrize(
|
|
"last_release, current_release",
|
|
[
|
|
("v0.0.1", "v0.0.1"),
|
|
("v0.0.2", "v0.0.1"),
|
|
("v0.0.1", "v0.0.2"),
|
|
("v1.9.0", "v1.10.0"),
|
|
("v1.10.0", "v1.9.0"),
|
|
("2.0.0", "2.0.0rc1"),
|
|
("2.0.0rc1", "2.0.0"),
|
|
("2.0.0rc1", "2.0.0rc2"),
|
|
("2.0.0rc2", "2.0.0rc1"),
|
|
("3.41.99", "88.123.456"),
|
|
("88.123.456", "3.41.99"),
|
|
("3.41.99", "3.41.99"),
|
|
("3.41.99", "3.41.99rc1"),
|
|
("8.10.01", "8.1.01"),
|
|
],
|
|
)
|
|
def test_check_for_updates(mocker, last_release, current_release):
|
|
# MOCK GET
|
|
attrs = {
|
|
"status_code": 200,
|
|
"json.return_value": {
|
|
"url": "mock_url",
|
|
"assets_url": "mock_url",
|
|
"upload_url": "mock_url",
|
|
"html_url": "mock_url",
|
|
"node_id": "mock_node_id",
|
|
"tag_name": last_release,
|
|
"name": "mock_terminal_name",
|
|
},
|
|
}
|
|
mock_response = mocker.Mock(**attrs)
|
|
mocker.patch(
|
|
target="openbb_terminal.helper_funcs.request",
|
|
new=mocker.Mock(return_value=mock_response),
|
|
)
|
|
|
|
# MOCK FF VERSION
|
|
mocker.patch(
|
|
target="openbb_terminal.core.models.system_model.SystemModel.VERSION",
|
|
new=current_release,
|
|
)
|
|
|
|
terminal_helper.check_for_updates()
|