fix(asyncio): use asyncio StateContext in StateMiddleware

Author: juancabe

telebot/states/asyncio/middleware.py

@@ -1,6 +1,6 @@
 from telebot.asyncio_handler_backends import BaseMiddleware
 from telebot.async_telebot import AsyncTeleBot
-from telebot.states.sync.context import StateContext
+from telebot.states.asyncio.context import StateContext
 from telebot.util import update_types
 from telebot import types
 

tests/test_async_telebot.py

@@ -238,3 +238,11 @@ async def fake_sleep(_seconds):
     assert calls["n"] == budget, (
         f"Expected {budget} total attempts, got {calls['n']}"
     )
+
+
+def test_async_state_middleware_uses_async_context():
+    from telebot.states.asyncio.middleware import StateMiddleware, StateContext
+    from telebot.states.asyncio.context import StateContext as AsyncStateContext
+
+    assert StateContext is AsyncStateContext, "StateMiddleware must use asyncio.context.StateContext"
+