diff options
author | Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com> | 2023-09-18 13:37:23 +0200 |
---|---|---|
committer | Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com> | 2023-09-18 13:37:23 +0200 |
commit | 2e95b43f7765ba42f8daa6950b3b833f2a93ccf6 (patch) | |
tree | b0ca442e3b35626ebad0d12badfe474cb5d08326 | |
parent | c2021fff49c353b6ce72325a05f057ab1c7639d0 (diff) |
Improvements
10 files changed, 22 insertions, 20 deletions
diff --git a/openbb_sdk/providers/alpha_vantage/tests/test_alpha_vantage_fetchers.py b/openbb_sdk/providers/alpha_vantage/tests/test_alpha_vantage_fetchers.py index 545b932119e..ba60aa7119d 100644 --- a/openbb_sdk/providers/alpha_vantage/tests/test_alpha_vantage_fetchers.py +++ b/openbb_sdk/providers/alpha_vantage/tests/test_alpha_vantage_fetchers.py @@ -1,8 +1,8 @@ import pytest -from openbb import obb from openbb_alpha_vantage.models.stock_historical import AVStockHistoricalFetcher +from openbb_core.app.service.user_service import UserService -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/benzinga/tests/test_benzinga_fetchers.py b/openbb_sdk/providers/benzinga/tests/test_benzinga_fetchers.py index 90d938f1070..ffa46217bfc 100644 --- a/openbb_sdk/providers/benzinga/tests/test_benzinga_fetchers.py +++ b/openbb_sdk/providers/benzinga/tests/test_benzinga_fetchers.py @@ -1,9 +1,9 @@ import pytest -from openbb import obb from openbb_benzinga.models.global_news import BenzingaGlobalNewsFetcher from openbb_benzinga.models.stock_news import BenzingaStockNewsFetcher +from openbb_core.app.service.user_service import UserService -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/cboe/tests/test_cboe_fetchers.py b/openbb_sdk/providers/cboe/tests/test_cboe_fetchers.py index a62f7867849..86422c1b5ee 100644 --- a/openbb_sdk/providers/cboe/tests/test_cboe_fetchers.py +++ b/openbb_sdk/providers/cboe/tests/test_cboe_fetchers.py @@ -1,5 +1,4 @@ import pytest -from openbb import obb from openbb_cboe.models.available_indices import CboeAvailableIndicesFetcher from openbb_cboe.models.european_index_constituents import ( CboeEuropeanIndexConstituentsFetcher, @@ -17,8 +16,9 @@ from openbb_cboe.models.options_chains import CboeOptionsChainsFetcher from openbb_cboe.models.stock_historical import CboeStockHistoricalFetcher from openbb_cboe.models.stock_info import CboeStockInfoFetcher from openbb_cboe.models.stock_search import CboeStockSearchFetcher +from openbb_core.app.service.user_service import UserService -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/fmp/tests/test_fmp_fetchers.py b/openbb_sdk/providers/fmp/tests/test_fmp_fetchers.py index ec1efb4af4e..d1aeb90a0ab 100644 --- a/openbb_sdk/providers/fmp/tests/test_fmp_fetchers.py +++ b/openbb_sdk/providers/fmp/tests/test_fmp_fetchers.py @@ -1,5 +1,5 @@ import pytest -from openbb import obb +from openbb_core.app.service.user_service import UserService from openbb_fmp.models.analyst_estimates import FMPAnalystEstimatesFetcher from openbb_fmp.models.available_indices import FMPAvailableIndicesFetcher from openbb_fmp.models.balance_sheet import FMPBalanceSheetFetcher @@ -45,7 +45,7 @@ from openbb_fmp.models.stock_quote import FMPStockQuoteFetcher from openbb_fmp.models.stock_splits import FMPStockSplitCalendarFetcher from openbb_fmp.models.treasury_rates import FMPTreasuryRatesFetcher -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/fred/tests/test_fred_fetchers.py b/openbb_sdk/providers/fred/tests/test_fred_fetchers.py index eba04b327ff..ed61470b697 100644 --- a/openbb_sdk/providers/fred/tests/test_fred_fetchers.py +++ b/openbb_sdk/providers/fred/tests/test_fred_fetchers.py @@ -1,5 +1,5 @@ import pytest -from openbb import obb +from openbb_core.app.service.user_service import UserService from openbb_fred.models.ameribor_rates import FREDAMERIBORFetcher from openbb_fred.models.cpi import FREDCPIFetcher from openbb_fred.models.estr_rates import FREDESTRFetcher @@ -10,7 +10,7 @@ from openbb_fred.models.sofr_rates import FREDSOFRFetcher from openbb_fred.models.sonia_rates import FREDSONIAFetcher from openbb_fred.models.us_yield_curve import FREDYieldCurveFetcher -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/intrinio/tests/test_intrinio_fetchers.py b/openbb_sdk/providers/intrinio/tests/test_intrinio_fetchers.py index 655f5f8e277..4c946e760cb 100644 --- a/openbb_sdk/providers/intrinio/tests/test_intrinio_fetchers.py +++ b/openbb_sdk/providers/intrinio/tests/test_intrinio_fetchers.py @@ -1,5 +1,5 @@ import pytest -from openbb import obb +from openbb_core.app.service.user_service import UserService from openbb_intrinio.models.balance_sheet import IntrinioBalanceSheetFetcher from openbb_intrinio.models.cash_flow import IntrinioCashFlowStatementFetcher from openbb_intrinio.models.forex_pairs import IntrinioForexPairsFetcher @@ -10,7 +10,7 @@ from openbb_intrinio.models.stock_historical import IntrinioStockHistoricalFetch from openbb_intrinio.models.stock_news import IntrinioStockNewsFetcher from openbb_intrinio.models.stock_quote import IntrinioStockQuoteFetcher -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/polygon/tests/test_polygon_fetchers.py b/openbb_sdk/providers/polygon/tests/test_polygon_fetchers.py index 091da98685d..c8e787e7058 100644 --- a/openbb_sdk/providers/polygon/tests/test_polygon_fetchers.py +++ b/openbb_sdk/providers/polygon/tests/test_polygon_fetchers.py @@ -1,5 +1,5 @@ import pytest -from openbb import obb +from openbb_core.app.service.user_service import UserService from openbb_polygon.models.balance_sheet import PolygonBalanceSheetFetcher from openbb_polygon.models.cash_flow import PolygonCashFlowStatementFetcher from openbb_polygon.models.crypto_historical import PolygonCryptoHistoricalFetcher @@ -12,7 +12,7 @@ from openbb_polygon.models.major_indices_historical import ( from openbb_polygon.models.stock_historical import PolygonStockHistoricalFetcher from openbb_polygon.models.stock_news import PolygonStockNewsFetcher -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/quandl/tests/test_quandl_fetchers.py b/openbb_sdk/providers/quandl/tests/test_quandl_fetchers.py index bdce5d46509..e8d1961a419 100644 --- a/openbb_sdk/providers/quandl/tests/test_quandl_fetchers.py +++ b/openbb_sdk/providers/quandl/tests/test_quandl_fetchers.py @@ -1,10 +1,10 @@ import pytest -from openbb import obb +from openbb_core.app.service.user_service import UserService from openbb_quandl.models.cot import QuandlCotFetcher from openbb_quandl.models.cot_search import QuandlCotSearchFetcher from openbb_quandl.models.sp500_multiples import QuandlSP500MultiplesFetcher -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/utils/unit_test_generator.py b/openbb_sdk/providers/utils/unit_test_generator.py index d7a2c0e23d7..4ce47ef07d1 100644 --- a/openbb_sdk/providers/utils/unit_test_generator.py +++ b/openbb_sdk/providers/utils/unit_test_generator.py @@ -31,7 +31,9 @@ def generate_fetcher_unit_tests(path: str) -> None: """Generate the fetcher unit tests in the provider test folders.""" if not os.path.exists(path): with open(path, "w") as f: - f.write("import pytest\nfrom openbb import obb\n") + f.write( + "import pytest\nfrom openbb_core.app.service.user_service import UserService\n" + ) def get_test_params(param_fields: Dict[str, ModelField]) -> Dict[str, Any]: @@ -71,7 +73,7 @@ def write_test_credentials(path: str, provider: str) -> None: credentials: Dict[str, str] = test_credentials.get(provider, {}) template = """ -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") diff --git a/openbb_sdk/providers/yfinance/tests/test_yfinance_fetchers.py b/openbb_sdk/providers/yfinance/tests/test_yfinance_fetchers.py index e98e3736eb5..9688d55ca07 100644 --- a/openbb_sdk/providers/yfinance/tests/test_yfinance_fetchers.py +++ b/openbb_sdk/providers/yfinance/tests/test_yfinance_fetchers.py @@ -1,5 +1,5 @@ import pytest -from openbb import obb +from openbb_core.app.service.user_service import UserService from openbb_yfinance.models.available_indices import YFinanceAvailableIndicesFetcher from openbb_yfinance.models.balance_sheet import YFinanceBalanceSheetFetcher from openbb_yfinance.models.cash_flow import YFinanceCashFlowStatementFetcher @@ -14,7 +14,7 @@ from openbb_yfinance.models.major_indices_historical import ( from openbb_yfinance.models.stock_historical import YFinanceStockHistoricalFetcher from openbb_yfinance.models.stock_news import YFinanceStockNewsFetcher -test_credentials = obb.user.credentials.__dict__ +test_credentials = UserService().default_user_settings.credentials.dict() @pytest.fixture(scope="module") |