fix: return if return route but no response

Signed-off-by: Timo Glastra <timo@animo.id>

Author: TimoGlastra

aries_cloudagent/transport/inbound/http.py

@@ -96,7 +96,12 @@ async def inbound_message_handler(self, request: web.BaseRequest):
                 raise web.HTTPBadRequest()
 
             if inbound.receipt.direct_response_requested:
-                response = await session.wait_response()
+                # Wait for the message to be processed. Only send a response if a response
+                # buffer is present.
+                await inbound.wait_processing_complete()
+                response = (
+                    await session.wait_response() if session.response_buffer else None
+                )
 
                 # no more responses
                 session.can_respond = False

aries_cloudagent/transport/inbound/manager.py

@@ -181,6 +181,8 @@ def dispatch_complete(self, message: InboundMessage, completed: CompletedTask):
         if session and session.accept_undelivered and not session.response_buffered:
             self.process_undelivered(session)
 
+        message.dispatch_processing_complete()
+
     def closed_session(self, session: InboundSession):
         """
         Clean up a closed session.

aries_cloudagent/transport/inbound/message.py

@@ -1,5 +1,6 @@
 """Classes representing inbound messages."""
 
+import asyncio
 from typing import Union
 
 from .receipt import MessageReceipt
@@ -23,3 +24,12 @@ def __init__(
         self.receipt = receipt
         self.session_id = session_id
         self.transport_type = transport_type
+        self.processing_complete_event = asyncio.Event()
+
+    def dispatch_processing_complete(self):
+        """Dispatch processing complete."""
+        self.processing_complete_event.set()
+
+    async def wait_processing_complete(self):
+        """Wait for processing to complete."""
+        await self.processing_complete_event.wait()

aries_cloudagent/transport/inbound/tests/test_message.py

@@ -0,0 +1,30 @@
+import asyncio
+
+from asynctest import TestCase
+
+from ..message import InboundMessage
+from ..receipt import MessageReceipt
+
+
+class TestInboundMessage(TestCase):
+    async def test_wait_response(self):
+        message = InboundMessage(
+            payload="test",
+            connection_id="conn_id",
+            receipt=MessageReceipt(),
+            session_id="session_id",
+        )
+        assert not message.processing_complete_event.is_set()
+        message.dispatch_processing_complete()
+        assert message.processing_complete_event.is_set()
+
+        message = InboundMessage(
+            payload="test",
+            connection_id="conn_id",
+            receipt=MessageReceipt(),
+            session_id="session_id",
+        )
+        assert not message.processing_complete_event.is_set()
+        task = message.wait_processing_complete()
+        message.dispatch_processing_complete()
+        await asyncio.wait_for(task, 1)