def test__start_trading_should_call_next_setup_message_correctly_when_step_3(
self):
tcb = TelegramController()
tcb.in_progress_step = 3
tcb._send_text_message = MagicMock()
tcb.currency = "BTC"
tcb._start_trading("upbit")
tcb._send_text_message.assert_called_once_with(
tcb.setup_list[3]["guide"], tcb.setup_list[3]["keyboard"])
self.assertIsNotNone(tcb.trader)
self.assertIsNotNone(tcb.data_provider)
self.assertEqual(tcb.in_progress, tcb._start_trading)
self.assertEqual(tcb.in_progress_step, 4)
def test__start_trading_should_call_next_setup_message_correctly_when_step_4(
self):
tcb = TelegramController()
tcb.in_progress_step = 4
tcb._send_text_message = MagicMock()
tcb.trader = MagicMock()
tcb.currency = "mango"
tcb.trader.NAME = "mango trader"
tcb.strategy = MagicMock()
tcb.strategy.NAME = "mango strategy"
tcb.budget = 500
tcb._start_trading("SMA")
tcb._send_text_message.assert_called_once_with(
f"화폐: mango\n전략: {StrategySma0.NAME}\n거래소: mango trader\n예산: 500\n자동 거래를 시작할까요?",
tcb.setup_list[4]["keyboard"],
)
self.assertIsNotNone(tcb.strategy)
self.assertEqual(tcb.in_progress, tcb._start_trading)
self.assertEqual(tcb.in_progress_step, 5)