mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-09 02:42:36 +00:00
b71abcfbf4
* 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>
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import pytest
|
|
import requests
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.stocks.comparison_analysis import polygon_model
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_query_parameters": [
|
|
("apiKey", "MOCK_APIKEY"),
|
|
],
|
|
}
|
|
|
|
|
|
@pytest.mark.vcr
|
|
def test_get_financial_comparisons(recorder):
|
|
result_tuple = polygon_model.get_similar_companies(
|
|
ticker="TSLA",
|
|
us_only=True,
|
|
)
|
|
|
|
recorder.capture(result_tuple)
|
|
|
|
|
|
@pytest.mark.vcr(record_mode="none")
|
|
def test_get_similar_companies_invalid_status(mocker, recorder):
|
|
mock_response = requests.Response()
|
|
mock_response.status_code = 400
|
|
# pylint: disable=protected-access
|
|
mock_response._content = b"""{"error":"MOCK_ERROR"}"""
|
|
mocker.patch(target="requests.get", new=mocker.Mock(return_value=mock_response))
|
|
result_tuple = polygon_model.get_similar_companies(
|
|
ticker="TSLA",
|
|
us_only=True,
|
|
)
|
|
|
|
recorder.capture(result_tuple)
|