OpenBB/tests/openbb_terminal/stocks/fundamental_analysis/test_finnhub_view.py

# IMPORTATION STANDARD

# IMPORTATION THIRDPARTY
import pytest
from pandas import DataFrame

# IMPORTATION INTERNAL
from openbb_terminal.stocks.fundamental_analysis import finnhub_view


@pytest.fixture(scope="module")
def vcr_config():
    return {
        "filter_query_parameters": [
            ("token", "MOCK_TOKEN"),
        ]
    }


@pytest.mark.default_cassette("test_rating_over_time_TSLA")
@pytest.mark.vcr
@pytest.mark.record_stdout
def test_rating_over_time():
    finnhub_view.rating_over_time(
        symbol="TSLA",
        limit=10,
        raw=True,
        export=None,
    )


@pytest.mark.default_cassette("test_rating_over_time_TSLA")
@pytest.mark.vcr(mode="none")
def test_rating_over_time_plt(capsys):
    finnhub_view.rating_over_time(
        symbol="TSLA",
        limit=10,
        raw=False,
        export=None,
    )

    capsys.readouterr()


@pytest.mark.vcr
@pytest.mark.record_stdout
def test_rating_over_time_invalid_ticker():
    finnhub_view.rating_over_time(
        symbol="INVALID_TICKER",
        limit=10,
        raw=False,
        export=None,
    )


@pytest.mark.vcr
@pytest.mark.record_stdout
def test_rating_over_time_invalid_token():
    finnhub_view.rating_over_time(
        symbol="TSLA",
        limit=10,
        raw=False,
        export=None,
    )


def test_plot_rating_over_time():
    finnhub_view.plot_rating_over_time(
        data=DataFrame(
            {
                "period": ["2021-01-01", "2021-02-01"],
                "strongSell": [1, 2],
                "sell": [3, 4],
                "hold": [5, 6],
                "buy": [7, 8],
                "strongBuy": [9, 10],
            }
        ),
        symbol="TSLA",
        external_axes=False,
    )


@pytest.mark.http_record
@pytest.mark.record_verify_screen
@pytest.mark.skip
def test_rating_over_time_raw():
    finnhub_view.rating_over_time(
        symbol="TSLA",
        limit=10,
        raw=True,
        export=None,
    )