diff --git a/tests/brokers/base/middlewares.py b/tests/brokers/base/middlewares.py index 3f8b3e8f74..d16c4861ea 100644 --- a/tests/brokers/base/middlewares.py +++ b/tests/brokers/base/middlewares.py @@ -111,9 +111,7 @@ async def handler(msg): assert [c.args[0] for c in mock.call_args_list] == ["outer", "middle", "inner"] - async def test_publisher_with_router_middleware_order( - self, queue: str, mock: Mock - ): + async def test_publisher_with_router_middleware_order(self, queue: str, mock: Mock): class InnerMiddleware(BaseMiddleware): async def publish_scope(self, call_next, msg, *args, **kwargs): mock.publish_inner() @@ -238,6 +236,7 @@ async def handler(msg): assert [c.args[0] for c in mock.call_args_list] == ["outer", "middle", "inner"] + @pytest.mark.asyncio class LocalMiddlewareTestcase(BaseTestcaseConfig): broker_class: Type[BrokerUsecase]