test: telegram.py::init
This commit is contained in:
parent
e9e644ffaf
commit
c47b73be8c
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user