mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-31 22:41:19 +00:00
068a0788f9
* add ETF holdings from SEC provider * black * helper function params * edge cases * ruff * add validator in company filings * remove xml field * update import * cleanup parsing * missing datetime import in company_filings * black * unused import * add missing py.typed files
115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
import pytest
|
|
from openbb_core.app.service.user_service import UserService
|
|
from openbb_sec.models.cik_map import SecCikMapFetcher
|
|
from openbb_sec.models.company_filings import SecCompanyFilingsFetcher
|
|
from openbb_sec.models.equity_ftd import SecEquityFtdFetcher
|
|
from openbb_sec.models.equity_search import SecEquitySearchFetcher
|
|
from openbb_sec.models.etf_holdings import SecEtfHoldingsFetcher
|
|
from openbb_sec.models.institutions_search import SecInstitutionsSearchFetcher
|
|
from openbb_sec.models.rss_litigation import SecRssLitigationFetcher
|
|
from openbb_sec.models.schema_files import SecSchemaFilesFetcher
|
|
from openbb_sec.models.sic_search import SecSicSearchFetcher
|
|
from openbb_sec.models.symbol_map import SecSymbolMapFetcher
|
|
|
|
test_credentials = UserService().default_user_settings.credentials.dict()
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_headers": [("User-Agent", None)],
|
|
"filter_query_parameters": [
|
|
None,
|
|
],
|
|
}
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_etf_holdings_fetcher(credentials=test_credentials):
|
|
params = {"symbol": "TQQQ", "use_cache": False}
|
|
|
|
fetcher = SecEtfHoldingsFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_sic_search_fetcher(credentials=test_credentials):
|
|
params = {"query": "oil", "use_cache": False}
|
|
|
|
fetcher = SecSicSearchFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_symbol_map_fetcher(credentials=test_credentials):
|
|
params = {"query": "0000909832"}
|
|
|
|
fetcher = SecSymbolMapFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_equity_ftd_fetcher(credentials=test_credentials):
|
|
params = {"symbol": "AAPL", "limit": 1}
|
|
|
|
fetcher = SecEquityFtdFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_equity_search_fetcher(credentials=test_credentials):
|
|
params = {"query": "trust", "use_cache": False}
|
|
|
|
fetcher = SecEquitySearchFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_company_filings_fetcher(credentials=test_credentials):
|
|
params = {"symbol": "AAPL", "type": "10-K", "use_cache": False}
|
|
|
|
fetcher = SecCompanyFilingsFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_institutions_search_fetcher(credentials=test_credentials):
|
|
params = {"query": "Investment Trust", "use_cache": False}
|
|
|
|
fetcher = SecInstitutionsSearchFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_schema_files_fetcher(credentials=test_credentials):
|
|
params = {"query": "2022"}
|
|
|
|
fetcher = SecSchemaFilesFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_rss_litigation_fetcher(credentials=test_credentials):
|
|
params = {}
|
|
|
|
fetcher = SecRssLitigationFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.record_http
|
|
def test_sec_cik_map_fetcher(credentials=test_credentials):
|
|
params = {"symbol": "OXY"}
|
|
|
|
fetcher = SecCikMapFetcher()
|
|
result = fetcher.test(params, credentials)
|
|
assert result is None
|