mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-02 07:23:25 +00:00
eb2c11060d
* bump * Initial function write * Delete this line from yaml * Changed all the main things - Need help on tests * pylint errors * Tests v0 * Tests v1 * Tests v2 --------- Co-authored-by: andrewkenreich <andrew.kenreich@gmail.com>
44 lines
1013 B
Python
44 lines
1013 B
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import pytest
|
|
import requests
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.stocks.discovery import finnhub_model
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_query_parameters": [
|
|
("from", "MOCK_FROM"),
|
|
("to", "MOCK_TO"),
|
|
("token", "MOCK_TOKEN"),
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.mark.vcr
|
|
def test_get_ipo_calendar(recorder):
|
|
ipo_df = finnhub_model.get_ipo_calendar(
|
|
start_date="2021-12-01",
|
|
end_date="2021-12-02",
|
|
)
|
|
recorder.capture(ipo_df)
|
|
|
|
|
|
@pytest.mark.vcr(record_mode="none")
|
|
def test_get_ipo_calendar_400(mocker):
|
|
mock_response = requests.Response()
|
|
mock_response.status_code = 401
|
|
mocker.patch(
|
|
target="openbb_terminal.helper_funcs.requests.get",
|
|
new=mocker.Mock(return_value=mock_response),
|
|
)
|
|
ipo_df = finnhub_model.get_ipo_calendar(
|
|
start_date="2021-12-01",
|
|
end_date="2021-12-02",
|
|
)
|
|
assert ipo_df.empty
|