Handle ack_sync and test methods as async

Author: wpn10

faststream/nats/message.py

@@ -17,10 +17,10 @@ async def ack(self) -> None:
             await self.raw_message.ack()
         await super().ack()
 
-    def ack_sync(self) -> None:
+    async def ack_sync(self) -> None:
         if not self.raw_message._ackd:
-            self.raw_message.ack_sync()
-        super().ack()
+            await self.raw_message.ack_sync()
+        await super().ack()
 
     async def nack(
         self,

tests/brokers/nats/test_consume.py

@@ -1,5 +1,5 @@
 import asyncio
-from unittest.mock import MagicMock, Mock, patch
+from unittest.mock import Mock, patch
 
 import pytest
 from nats.aio.msg import Msg
@@ -224,21 +224,21 @@ async def test_consume_ack_sync_manual(
 
         @consume_broker.subscriber(queue, stream=stream)
         async def handler(msg: NatsMessage):
-            msg.ack_sync()
+            await msg.ack_sync()
             event.set()
 
         async with self.patch_broker(consume_broker) as br:
             await br.start()
 
-            with patch.object(Msg, "ack_sync", new_callable=MagicMock) as m:
+            with patch.object(Msg, "ack_sync", spy_decorator(Msg.ack_sync)) as m:
                 await asyncio.wait(
                     (
                         asyncio.create_task(br.publish("hello", queue)),
                         asyncio.create_task(event.wait()),
                     ),
                     timeout=3,
                 )
-                m.assert_called_once()
+                m.mock.assert_called_once()
 
         assert event.is_set()