test: telegram.py::init

This commit is contained in:
kryofly 2018-01-29 16:13:37 +01:00
parent e9e644ffaf
commit c47b73be8c

View File

@ -17,6 +17,8 @@ from freqtrade.rpc import telegram
from freqtrade.rpc.telegram import authorized_only, is_enabled, send_msg, _status, _status_table, \ from freqtrade.rpc.telegram import authorized_only, is_enabled, send_msg, _status, _status_table, \
_profit, _forcesell, _performance, _daily, _count, _start, _stop, _balance, _version, _help _profit, _forcesell, _performance, _daily, _count, _start, _stop, _balance, _version, _help
import freqtrade.rpc.telegram as tg
def test_is_enabled(default_conf, mocker): def test_is_enabled(default_conf, mocker):
mocker.patch.dict('freqtrade.rpc.telegram._CONF', default_conf) mocker.patch.dict('freqtrade.rpc.telegram._CONF', default_conf)
@ -741,15 +743,20 @@ def test_send_msg_network_error(default_conf, mocker):
def test_init(default_conf, update, ticker, limit_buy_order, limit_sell_order, mocker): def test_init(default_conf, update, ticker, limit_buy_order, limit_sell_order, mocker):
mocker.patch.dict('freqtrade.main._CONF', default_conf) start_polling = MagicMock()
mocker.patch('freqtrade.main.get_signal', side_effect=lambda s, t: (True, False))
msg_mock = MagicMock()
mocker.patch('freqtrade.main.rpc.send_msg', MagicMock())
mocker.patch.multiple('freqtrade.rpc.telegram', mocker.patch.multiple('freqtrade.rpc.telegram',
_CONF=default_conf, _CONF=default_conf,
init=MagicMock(), # mock telegram.ext.Updater
send_msg=msg_mock) Updater=MagicMock(return_value=start_polling))
mocker.patch.multiple('freqtrade.main.exchange', # not enabled
validate_pairs=MagicMock(), tg.init(default_conf)
get_ticker=ticker) assert start_polling.call_count == 0
init(default_conf, create_engine('sqlite://')) # number of handles registered
assert start_polling.dispatcher.add_handler.call_count == 11
assert start_polling.start_polling.call_count == 1
# enabled
default_conf['telegram'] = {}
default_conf['telegram']['enabled'] = True
default_conf['telegram']['token'] = ''
tg.init(default_conf)