mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: simplify test_worker_lifecycle

Browse Source
This commit is contained in:
Matthias
2025-03-26 06:46:06 +01:00
parent 948487518d
commit bb08880c4a
1 changed files with 11 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/freqtradebot/test_worker.py
+11 -30
View File
@@ -71,36 +71,18 @@ def test_worker_stopped(mocker, default_conf, caplog) -> None:
@pytest.mark.parametrize( @pytest.mark.parametrize(
"old_state, target_state, expected_state,startup_call, log_fragment", "old_state,target_state,startup_call,log_fragment",
[ [
(State.STOPPED, State.PAUSED, State.PAUSED, True, "Changing state from STOPPED to: PAUSED"), (State.STOPPED, State.PAUSED, True, "Changing state from STOPPED to: PAUSED"),
( (State.RUNNING, State.PAUSED, False, "Changing state from RUNNING to: PAUSED"),
State.RUNNING, (State.PAUSED, State.RUNNING, False, "Changing state from PAUSED to: RUNNING"),
State.PAUSED, (State.PAUSED, State.STOPPED, False, "Changing state from PAUSED to: STOPPED"),
State.PAUSED, (State.RELOAD_CONFIG, State.RUNNING, True, "Changing state from RELOAD_CONFIG to: RUNNING"),
False,
"Changing state from RUNNING to: PAUSED",
),
(
State.PAUSED,
State.RUNNING,
State.RUNNING,
False,
"Changing state from PAUSED to: RUNNING",
),
(
State.PAUSED,
State.STOPPED,
State.STOPPED,
False,
"Changing state from PAUSED to: STOPPED",
),
( (
State.RELOAD_CONFIG, State.RELOAD_CONFIG,
State.RUNNING, State.STOPPED,
State.RUNNING, False,
True, "Changing state from RELOAD_CONFIG to: STOPPED",
"Changing state from RELOAD_CONFIG to: RUNNING",
), ),
], ],
) )
@@ -110,7 +92,6 @@ def test_worker_lifecycle(
caplog, caplog,
old_state, old_state,
target_state, target_state,
expected_state,
startup_call, startup_call,
log_fragment, log_fragment,
): ):
@@ -124,13 +105,13 @@ def test_worker_lifecycle(
new_state = worker._worker(old_state=old_state) new_state = worker._worker(old_state=old_state)
assert new_state is expected_state assert new_state is target_state
assert log_has(log_fragment, caplog) assert log_has(log_fragment, caplog)
assert mock_throttle.call_count == 1 assert mock_throttle.call_count == 1
assert startup.call_count == (1 if startup_call else 0) assert startup.call_count == (1 if startup_call else 0)
# For any state where the strategy should be initialized # For any state where the strategy should be initialized
if expected_state in (State.RUNNING, State.PAUSED): if target_state in (State.RUNNING, State.PAUSED):
assert worker.freqtrade.strategy assert worker.freqtrade.strategy
assert isinstance(worker.freqtrade.strategy.dp, DataProvider) assert isinstance(worker.freqtrade.strategy.dp, DataProvider)
else: else: