From 2f89ee09f7ebf41d08dd59460a0241a0ae19902a Mon Sep 17 00:00:00 2001 From: hjoaquim Date: Mon, 9 Oct 2023 10:42:21 +0100 Subject: [PATCH] solving naming issues on pytest --- .../platform/core/tests/__init__.py | 0 ...test_router.py => test_platform_router.py} | 0 .../platform/core/tests/test_query.py | 125 ------------------ 3 files changed, 125 deletions(-) delete mode 100644 openbb_platform/platform/core/tests/__init__.py rename openbb_platform/platform/core/tests/app/{test_router.py => test_platform_router.py} (100%) delete mode 100644 openbb_platform/platform/core/tests/test_query.py diff --git a/openbb_platform/platform/core/tests/__init__.py b/openbb_platform/platform/core/tests/__init__.py deleted file mode 100644 index e69de29bb2d..00000000000 diff --git a/openbb_platform/platform/core/tests/app/test_router.py b/openbb_platform/platform/core/tests/app/test_platform_router.py similarity index 100% rename from openbb_platform/platform/core/tests/app/test_router.py rename to openbb_platform/platform/core/tests/app/test_platform_router.py diff --git a/openbb_platform/platform/core/tests/test_query.py b/openbb_platform/platform/core/tests/test_query.py deleted file mode 100644 index d4df7740ce2..00000000000 --- a/openbb_platform/platform/core/tests/test_query.py +++ /dev/null @@ -1,125 +0,0 @@ -"""Test the Query class.""" -# pylint: disable=redefined-outer-name - -from dataclasses import dataclass -from unittest.mock import MagicMock, patch - -import pytest -from openbb_core.app.model.command_context import CommandContext -from openbb_core.app.provider_interface import ( - ExtraParams, - ProviderChoices, - StandardParams, -) -from openbb_core.app.query import Query -from pydantic import BaseModel, ConfigDict - - -class MockBaseModel(BaseModel): - """Mock BaseModel class.""" - - model_config = ConfigDict(extra="allow", populate_by_name=True) - - -def create_mock_query(): - """Mock query.""" - - class StockHistorical: - """Mock StockHistorical class.""" - - start_date = "2020-01-01" - end_date = "2020-01-05" - symbol = "AAPL" - - return StockHistorical() - - -def create_mock_extra_params(): - """Mock ExtraParams dataclass.""" - - @dataclass - class StockHistorical: - """Mock ExtraParams dataclass.""" - - sort: str = "desc" - - return StockHistorical() - - -@pytest.fixture(scope="module") -def query(): - """Set up query.""" - return Query( - cc=CommandContext(), - provider_choices=ProviderChoices(provider="fmp"), - standard_params=StandardParams(), - extra_params=ExtraParams(), - ) - - -def test_init(query): - """Test init.""" - assert query - - -def test_filter_extra_params(query): - """Test filter_extra_params.""" - extra_params = create_mock_extra_params() - extra_params = query.filter_extra_params(extra_params, "fmp") - - assert isinstance(extra_params, dict) - assert len(extra_params) == 0 - - -def test_filter_extra_params_wrong_param(query): - """Test filter_extra_params.""" - - @dataclass - class StockHistorical: - """Mock ExtraParams dataclass.""" - - sort: str = "desc" - limit: int = 4 - - extra_params = StockHistorical() - - assert not query.filter_extra_params(extra_params, "fmp") - - -@pytest.fixture -def mock_registry(): - """Mock registry.""" - with patch( - "openbb_core.app.provider_interface.ProviderInterface" - ) as mock_get_provider_interface: - mock_registry = MagicMock() - mock_get_provider_interface.return_value.build_registry.return_value = ( - mock_registry - ) - yield mock_registry - - -@pytest.fixture -def query_instance(): - """Set up query.""" - standard_params = create_mock_query() - extra_params = create_mock_extra_params() - - cc = CommandContext() - setattr(cc.user_settings.credentials, "fmp_api_key", "1234") - - return Query( - cc=cc, - provider_choices=ProviderChoices(provider="fmp"), - standard_params=standard_params, - extra_params=extra_params, - ) - - -def test_execute_method_fake_credentials(query_instance, mock_registry): - """Test execute method without setting credentials.""" - mock_fetch_result = MockBaseModel() - mock_registry.fetch.return_value = mock_fetch_result - - with pytest.raises(Exception): - query_instance.execute()