def call_fmp(self, _):
"""Process fmp command"""
ret = fmp_controller.menu(self.ticker, self.start, self.interval)
if ret is False:
self.print_help()
else:
return True
def test_menu_with_queue(expected, mocker, queue):
mocker.patch(
target=
("gamestonk_terminal.stocks.fundamental_analysis.financial_modeling_prep.fmp_controller."
"FinancialModelingPrepController.switch"),
return_value=["quit"],
)
result_menu = fmp_controller.menu(
ticker="TSLA",
start="10/25/2021",
interval="1440min",
queue=queue,
)
assert result_menu == expected
def test_menu_without_queue_sys_exit(mock_input, mocker):
# DISABLE AUTO-COMPLETION
mocker.patch.object(
target=fmp_controller.gtff,
attribute="USE_PROMPT_TOOLKIT",
new=False,
)
mocker.patch(
target=
"gamestonk_terminal.stocks.fundamental_analysis.financial_modeling_prep.fmp_controller.session",
return_value=None,
)
# MOCK USER INPUT
mocker.patch("builtins.input", return_value=mock_input)
# MOCK SWITCH
class SystemExitSideEffect:
def __init__(self):
self.first_call = True
def __call__(self, *args, **kwargs):
if self.first_call:
self.first_call = False
raise SystemExit()
return ["quit"]
mock_switch = mocker.Mock(side_effect=SystemExitSideEffect())
mocker.patch(
target=
("gamestonk_terminal.stocks.fundamental_analysis.financial_modeling_prep.fmp_controller."
"FinancialModelingPrepController.switch"),
new=mock_switch,
)
result_menu = fmp_controller.menu(ticker="TSLA",
start="10/25/2021",
interval="1440min",
queue=None)
assert result_menu == []
def test_menu_without_queue_completion(mocker):
# DISABLE AUTO-COMPLETION
mocker.patch.object(
target=fmp_controller.gtff,
attribute="USE_PROMPT_TOOLKIT",
new=True,
)
mocker.patch(
target=
"gamestonk_terminal.stocks.fundamental_analysis.financial_modeling_prep.fmp_controller.session",
)
mocker.patch(
target=
"gamestonk_terminal.stocks.fundamental_analysis.financial_modeling_prep.fmp_controller.session.prompt",
return_value="quit",
)
result_menu = fmp_controller.menu(ticker="TSLA",
start="10/25/2021",
interval="1440min",
queue=None)
assert result_menu == []
def call_fmp(self, _):
"""Process fmp command."""
self.queue = fmp_controller.menu(self.ticker, self.start,
self.interval, self.queue)