Files
OpenBB/tests/openbb_terminal/economy/test_finnhub_model.py
jmaslek b71abcfbf4 Updating some names (#1575)
* quick econ fix

* black

* keys and feature flags

* terminal name 👀

* some more replacements

* some more replacements

* edit pyproject

* gst -> openbb

* add example portfolios back to git

* Update api from gst

* sorry.  skipping some tests

* another round of names

* another round of test edits

* Missed some .gst refs and update timezone

* water mark stuff

* Fixing Names in terminal.spec and name of GTFF_DEFAULTS to OBBFF_DEFAULTS

* fix more GST to OpenBB Terminal

* Logging : merge conflicts with main

* Revert wrong files

Co-authored-by: Andrew <andrew.kenreich@gmail.com>
Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt>
Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
2022-03-29 09:46:26 -04:00

51 lines
1.2 KiB
Python

# IMPORTATION STANDARD
# IMPORTATION THIRDPARTY
import pandas as pd
import pytest
# IMPORTATION INTERNAL
from openbb_terminal.economy import finnhub_model
@pytest.fixture(scope="module")
def vcr_config():
return {
"filter_query_parameters": [("token", "MOCK_TOKEN")],
}
@pytest.mark.vcr(record_mode="none")
def test_get_economy_calendar_events(mocker):
# MOCK JSON
mock_json = pd.DataFrame()
mock_json["economicCalendar"] = ["MOCK_ROW_1", "MOCK_ROW_2"]
# MOCK GET
attrs = {
"status_code": 200,
"json.return_value": mock_json,
}
mock_response = mocker.Mock(**attrs)
mocker.patch(target="requests.get", new=mocker.Mock(return_value=mock_response))
result_df = finnhub_model.get_economy_calendar_events()
assert not result_df.empty
@pytest.mark.vcr(record_mode="none")
def test_get_economy_calendar_events_no_response(mocker):
# MOCK GET
attrs = {
"json.return_value": {"error": "mock error message"},
}
mock_response = mocker.Mock(**attrs)
mocker.patch(target="requests.get", new=mocker.Mock(return_value=mock_response))
result_df = finnhub_model.get_economy_calendar_events()
assert isinstance(result_df, pd.DataFrame)
assert result_df.empty