fix tests

rasa/core/channels/console.py

@@ -98,7 +98,7 @@ async def send_message_receive_stream(server_url, auth_token, sender_id, message
 
             async for line in resp.content:
                 if line:
-                    yield await json.loads(line.decode("utf-8"))
+                    yield json.loads(line.decode("utf-8"))
 
 
 async def record_messages(

rasa/core/lock_store.py

@@ -2,7 +2,7 @@
 import json
 import logging
 import os
-from typing import Text, Optional, Union
+from typing import Text, Optional, Union, AsyncGenerator, Coroutine
 
 from async_generator import asynccontextmanager
 
@@ -129,7 +129,7 @@ async def lock(
         conversation_id: Text,
         lock_lifetime: int = LOCK_LIFETIME,
         wait_time_in_seconds: Union[int, float] = 1,
-    ) -> None:
+    ) -> AsyncGenerator[TicketLock, None]:
         """Acquire lock with lifetime `lock_lifetime`for `conversation_id`.
 
         Try acquiring lock with a wait time of `wait_time_in_seconds` seconds

tests/core/test_agent.py

@@ -53,7 +53,7 @@ async def model_server(test_server, trained_moodbot_path):
     server = await test_server(
         model_server_app(trained_moodbot_path, model_hash="somehash")
     )
-    yield await server
+    yield server
     await server.close()