diff --git a/tests/test_fsm/test_strategy.py b/tests/test_fsm/test_strategy.py index 1938f7a0..1f6f2e93 100644 --- a/tests/test_fsm/test_strategy.py +++ b/tests/test_fsm/test_strategy.py @@ -27,9 +27,9 @@ class TestStrategy: [FSMStrategy.USER_IN_TOPIC, CHAT, CHAT], [FSMStrategy.USER_IN_TOPIC, PRIVATE, PRIVATE], [FSMStrategy.USER_IN_TOPIC, THREAD, THREAD], - [FSMStrategy.CHAT_TOPIC, CHAT, (CHAT_ID, None, None)], - [FSMStrategy.CHAT_TOPIC, PRIVATE, (USER_ID, None, None)], - [FSMStrategy.CHAT_TOPIC, THREAD, (CHAT_ID, None, THREAD_ID)], + [FSMStrategy.CHAT_TOPIC, CHAT, (CHAT_ID, CHAT_ID, None)], + [FSMStrategy.CHAT_TOPIC, PRIVATE, PRIVATE], + [FSMStrategy.CHAT_TOPIC, THREAD, (CHAT_ID, CHAT_ID, THREAD_ID)], ], ) def test_strategy(self, strategy, case, expected):