update

Author: Tom Softreck

tests/integration/test_https.py

@@ -22,20 +22,35 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
         if self.session:
             await self.session.close()
 
-    async def receive(self):
-        """Receive data from the HTTP endpoint"""
+    async def _fetch_data(self):
+        """Fetch data from the HTTP endpoint"""
         if not self.session:
             self.session = aiohttp.ClientSession()
 
         async with self.session.get(self.url) as response:
             if response.status == 200:
                 data = await response.json()
-                return [{'data': data, 'metadata': {'url': self.url, 'status': response.status}}]
+                return {'data': data, 'metadata': {'url': self.url, 'status': response.status}}
             else:
-                return [{'error': f"HTTP {response.status}", 'metadata': {'url': self.url, 'status': response.status}}]
+                return {'error': f"HTTP {response.status}", 'metadata': {'url': self.url, 'status': response.status}}
+    
+    async def receive(self):
+        """Return an async iterator that yields messages"""
+        class HTTPSourceIterator:
+            def __init__(self, source):
+                self.source = source
+                self._data_received = False
+            
+            def __aiter__(self):
+                return self
+            
+            async def __anext__(self):
+                if self._data_received:
+                    raise StopAsyncIteration
+                self._data_received = True
+                return await self.source._fetch_data()
 
-        # Return an empty list if no data was received
-        return []
+        return HTTPSourceIterator(self)
 
 # Patch the engine to use our test HTTPSource
 @pytest.fixture(autouse=True)

tests/integration/test_mock_server.py

@@ -11,6 +11,7 @@
 
 # Import our test HTTPSource
 from .test_https import HTTPSource
+from dialogchain.engine import CamelRouterEngine
 
 # Get the directory of the current file
 TEST_DIR = Path(__file__).parent
@@ -109,10 +110,18 @@ async def test_mock_server_endpoints(mock_server, config):
             assert "event" in events[0], f"Event missing 'event' key: {events[0]}"
             assert "value" in events[0], f"Event missing 'value' key: {events[0]}"
 
+class TestCamelRouterEngine(CamelRouterEngine):
+    """Test engine that can handle HTTP sources"""
+    
+    async def create_source(self, uri: str):
+        """Create source connector from URI"""
+        if uri.startswith('http'):
+            return HTTPSource(uri)
+        return await super().create_source(uri)
+
 @pytest.mark.asyncio
 async def test_dialogchain_with_mock_server(mock_server, config):
     """Test DialogChain with the mock server"""
-    from dialogchain.engine import CamelRouterEngine
     from dialogchain.connectors import HTTPDestination
 
     # Get the port from config
@@ -135,18 +144,8 @@ async def test_dialogchain_with_mock_server(mock_server, config):
         ]
     }
 
-    # Initialize the engine with test config
-    engine = CamelRouterEngine(test_config, verbose=True)
-    
-    # Patch the create_source method to use our HTTPSource
-    original_create_source = engine.create_source
-    
-    def patched_create_source(uri):
-        if uri.startswith('http'):
-            return HTTPSource(uri)
-        return original_create_source(uri)
-    
-    engine.create_source = patched_create_source
+    # Initialize our test engine with the config
+    engine = TestCamelRouterEngine(test_config, verbose=True)
 
     # Test the route processing
     async with aiohttp.ClientSession() as session: