mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-05 08:51:20 +00:00
869cfed125
* merge branch develop with main * bump desktop version * fix the merge * sync Cargo.lock to desktop 2.0.0 * fix cli conftest.py --------- Co-authored-by: deeleeramone <>
29 lines
963 B
Python
29 lines
963 B
Python
"""Unit tests for ``openbb_sec.models.equity_ftd``."""
|
|
|
|
import asyncio
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from openbb_core.provider.utils.errors import EmptyDataError
|
|
|
|
from openbb_sec.models.equity_ftd import SecEquityFtdFetcher, SecEquityFtdQueryParams
|
|
|
|
|
|
def test_equity_ftd_aextract_no_results():
|
|
"""equity_ftd.py:93 -> raise EmptyDataError when no records collected."""
|
|
|
|
async def _urls():
|
|
return {"2024-01": "https://example.com/a.zip"}
|
|
|
|
async def _download(url, symbol, use_cache): # noqa: ARG001
|
|
return []
|
|
|
|
query = SecEquityFtdQueryParams(symbol="AAPL", limit=1, use_cache=False)
|
|
with (
|
|
patch("openbb_sec.utils.helpers.get_ftd_urls", _urls),
|
|
patch("openbb_sec.utils.helpers.download_zip_file", _download),
|
|
):
|
|
with pytest.raises(EmptyDataError) as exc:
|
|
asyncio.run(SecEquityFtdFetcher.aextract_data(query, None))
|
|
assert "no results were returned" in str(exc.value)
|