126 lines
4.1 KiB
Python
126 lines
4.1 KiB
Python
import logging
|
|
import time
|
|
from unittest.mock import MagicMock, PropertyMock
|
|
|
|
from freqtrade.data.dataprovider import DataProvider
|
|
from freqtrade.enums import State
|
|
from freqtrade.worker import Worker
|
|
from tests.conftest import get_patched_worker, log_has, log_has_re
|
|
|
|
|
|
def test_worker_state(mocker, default_conf, markets) -> None:
|
|
mocker.patch('freqtrade.exchange.Exchange.markets', PropertyMock(return_value=markets))
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
assert worker.freqtrade.state is State.RUNNING
|
|
|
|
default_conf.pop('initial_state')
|
|
worker = Worker(args=None, config=default_conf)
|
|
assert worker.freqtrade.state is State.STOPPED
|
|
|
|
|
|
def test_worker_running(mocker, default_conf, caplog) -> None:
|
|
mock_throttle = MagicMock()
|
|
mocker.patch('freqtrade.worker.Worker._throttle', mock_throttle)
|
|
mocker.patch('freqtrade.persistence.Trade.stoploss_reinitialization', MagicMock())
|
|
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
|
|
state = worker._worker(old_state=None)
|
|
assert state is State.RUNNING
|
|
assert log_has('Changing state to: RUNNING', caplog)
|
|
assert mock_throttle.call_count == 1
|
|
# Check strategy is loaded, and received a dataprovider object
|
|
assert worker.freqtrade.strategy
|
|
assert worker.freqtrade.strategy.dp
|
|
assert isinstance(worker.freqtrade.strategy.dp, DataProvider)
|
|
|
|
|
|
def test_worker_stopped(mocker, default_conf, caplog) -> None:
|
|
mock_throttle = MagicMock()
|
|
mocker.patch('freqtrade.worker.Worker._throttle', mock_throttle)
|
|
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
worker.freqtrade.state = State.STOPPED
|
|
state = worker._worker(old_state=State.RUNNING)
|
|
assert state is State.STOPPED
|
|
assert log_has('Changing state from RUNNING to: STOPPED', caplog)
|
|
assert mock_throttle.call_count == 1
|
|
|
|
|
|
def test_throttle(mocker, default_conf, caplog) -> None:
|
|
def throttled_func():
|
|
return 42
|
|
|
|
caplog.set_level(logging.DEBUG)
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
|
|
start = time.time()
|
|
result = worker._throttle(throttled_func, throttle_secs=0.1)
|
|
end = time.time()
|
|
|
|
assert result == 42
|
|
assert end - start > 0.1
|
|
assert log_has_re(r"Throttling with 'throttled_func\(\)': sleep for \d\.\d{2} s.*", caplog)
|
|
|
|
result = worker._throttle(throttled_func, throttle_secs=-1)
|
|
assert result == 42
|
|
|
|
|
|
def test_throttle_with_assets(mocker, default_conf) -> None:
|
|
def throttled_func(nb_assets=-1):
|
|
return nb_assets
|
|
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
|
|
result = worker._throttle(throttled_func, throttle_secs=0.1, nb_assets=666)
|
|
assert result == 666
|
|
|
|
result = worker._throttle(throttled_func, throttle_secs=0.1)
|
|
assert result == -1
|
|
|
|
|
|
def test_worker_heartbeat_running(default_conf, mocker, caplog):
|
|
message = r"Bot heartbeat\. PID=.*state='RUNNING'"
|
|
|
|
mock_throttle = MagicMock()
|
|
mocker.patch('freqtrade.worker.Worker._throttle', mock_throttle)
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
|
|
worker.freqtrade.state = State.RUNNING
|
|
worker._worker(old_state=State.STOPPED)
|
|
assert log_has_re(message, caplog)
|
|
|
|
caplog.clear()
|
|
# Message is not shown before interval is up
|
|
worker._worker(old_state=State.RUNNING)
|
|
assert not log_has_re(message, caplog)
|
|
|
|
caplog.clear()
|
|
# Set clock - 70 seconds
|
|
worker._heartbeat_msg -= 70
|
|
worker._worker(old_state=State.RUNNING)
|
|
assert log_has_re(message, caplog)
|
|
|
|
|
|
def test_worker_heartbeat_stopped(default_conf, mocker, caplog):
|
|
message = r"Bot heartbeat\. PID=.*state='STOPPED'"
|
|
|
|
mock_throttle = MagicMock()
|
|
mocker.patch('freqtrade.worker.Worker._throttle', mock_throttle)
|
|
worker = get_patched_worker(mocker, default_conf)
|
|
|
|
worker.freqtrade.state = State.STOPPED
|
|
worker._worker(old_state=State.RUNNING)
|
|
assert log_has_re(message, caplog)
|
|
|
|
caplog.clear()
|
|
# Message is not shown before interval is up
|
|
worker._worker(old_state=State.STOPPED)
|
|
assert not log_has_re(message, caplog)
|
|
|
|
caplog.clear()
|
|
# Set clock - 70 seconds
|
|
worker._heartbeat_msg -= 70
|
|
worker._worker(old_state=State.STOPPED)
|
|
assert log_has_re(message, caplog)
|