Added tests

tests/test_dispatcher/test_dispatcher.py

@@ -14,7 +14,7 @@ from aiogram import Bot
 from aiogram.dispatcher.dispatcher import Dispatcher
 from aiogram.dispatcher.event.bases import UNHANDLED, SkipHandler
 from aiogram.dispatcher.router import Router
-from aiogram.methods import GetMe, GetUpdates, Request, SendMessage, TelegramMethod
+from aiogram.methods import GetMe, GetUpdates, SendMessage, TelegramMethod
 from aiogram.types import (
     CallbackQuery,
     Chat,
@@ -462,9 +462,9 @@ class TestDispatcher:
         async def my_handler(event: Any, **kwargs: Any):
             assert event == getattr(update, event_type)
             if has_chat:
-                assert Chat.get_current(False)
+                assert kwargs["event_chat"]
             if has_user:
-                assert User.get_current(False)
+                assert kwargs["event_from_user"]
             return kwargs
 
         result = await router.feed_update(bot, update, test="PASS")

tests/test_dispatcher/test_middlewares/test_user_context.py

@@ -1,6 +1,9 @@
+from unittest.mock import patch
+
 import pytest
 
 from aiogram.dispatcher.middlewares.user_context import UserContextMiddleware
+from aiogram.types import Update
 
 
 async def next_handler(*args, **kwargs):
@@ -11,3 +14,13 @@ class TestUserContextMiddleware:
     async def test_unexpected_event_type(self):
         with pytest.raises(RuntimeError):
             await UserContextMiddleware()(next_handler, object(), {})
+
+    async def test_call(self):
+        middleware = UserContextMiddleware()
+        data = {}
+        with patch.object(UserContextMiddleware, "resolve_event_context", return_value=[1, 2, 3]):
+            await middleware(next_handler, Update(update_id=42), data)
+
+        assert data["event_chat"] == 1
+        assert data["event_from_user"] == 2
+        assert data["event_thread_id"] == 3

tests/test_fsm/storage/test_redis.py

@@ -11,6 +11,7 @@ PREFIX = "test"
 BOT_ID = 42
 CHAT_ID = -1
 USER_ID = 2
+THREAD_ID = 3
 FIELD = "data"
 
 
@@ -46,6 +47,19 @@ class TestRedisDefaultKeyBuilder:
         with pytest.raises(ValueError):
             key_builder.build(key, FIELD)
 
+    def test_thread_id(self):
+        key_builder = DefaultKeyBuilder(
+            prefix=PREFIX,
+        )
+        key = StorageKey(
+            chat_id=CHAT_ID,
+            user_id=USER_ID,
+            bot_id=BOT_ID,
+            thread_id=THREAD_ID,
+            destiny=DEFAULT_DESTINY,
+        )
+        assert key_builder.build(key, FIELD) == f"{PREFIX}:{CHAT_ID}:{THREAD_ID}:{USER_ID}:{FIELD}"
+
     def test_create_isolation(self):
         fake_redis = object()
         storage = RedisStorage(redis=fake_redis)