Add tests for _GeneratorRequest initialization and iteration

Author: grongierisc

src/iop/_generator_request.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional, Union
 
 from . import _iris
-from ._dispatch import dispatch_serializer, dispatch_deserializer
+from ._dispatch import dispatch_serializer
 
 class _GeneratorRequest:
     """Generator class to interetate over responses from a request.

src/tests/test_business_host.py

@@ -46,6 +46,58 @@ def test_send_multi_request_sync(self, business_host):
 
         assert result == [('test', SimpleMessage(integer=1, string='test'), MyResponse(value='test'), 1)]
 
+class TestGeneratorRequest:
+    def test_generator_request_initialization(self, business_host):
+        business_host.iris_handle.dispatchSendRequestSync = MagicMock()
+        business_host.iris_handle.dispatchSendRequestSync.return_value = iris.cls("IOP.Generator.Message.Ack")._New()
+
+        from iop._generator_request import _GeneratorRequest
+        generator = _GeneratorRequest(business_host, "test_target", SimpleMessage(integer=1, string='test'))
+
+        assert generator.host == business_host
+        assert generator.target == "test_target"
+        assert generator.request == SimpleMessage(integer=1, string='test')
+
+    @patch('iop._business_host._iris.get_iris')
+    @patch('iop._business_host.dispach_message')
+    def test_dispatch_generator_started(self, mock_dispatch, mock_iris, business_host):
+        mock_generator = iter([1, 2, 3])
+        mock_dispatch.return_value = mock_generator
+        mock_ack = MagicMock()
+        mock_iris.return_value.IOP.Generator.Message.Ack._New.return_value = mock_ack
+        
+        result = business_host._dispatch_generator_started("request")
+        
+        assert result == mock_ack
+        assert hasattr(business_host, '_gen')
+
+    @patch('iop._business_host.dispach_message')
+    def test_dispatch_generator_started_not_iterable(self, mock_dispatch, business_host):
+        mock_dispatch.return_value = SimpleMessage(integer=1, string='test')
+        
+        with pytest.raises(TypeError):
+            business_host._dispatch_generator_started("request")
+
+    def test_dispatch_generator_poll_next(self, business_host):
+        # Configure the mock to return proper buffer size
+        mock_msg = SimpleMessage(integer=1, string='test')
+        
+        business_host._gen = iter([SimpleMessage(integer=1, string='test'), SimpleMessage(integer=2, string='test2')])
+        
+        result = business_host._dispatch_generator_poll()
+
+        # deserialize the message
+        result = deserialize_message(result)
+        
+        assert result == SimpleMessage(integer=1, string='test')
+
+    def test_dispatch_generator_poll_stop(self, business_host):
+        business_host._gen = iter([])
+        
+        result = business_host._dispatch_generator_poll()
+        
+        assert result._IsA("IOP.Generator.Message.Stop")
+
 class TestMessageSerialization:
     def test_dispatch_serializer_valid(self, business_host):
         message = SimpleMessage(integer=1, string='test')

src/tests/test_generator_request.py

@@ -0,0 +1,79 @@
+import pytest
+from iop._generator_request import _GeneratorRequest
+from iop import PydanticMessage
+from unittest.mock import Mock, patch
+
+def test_generator_request_initialization():
+    # Mock host and request
+    mock_host = Mock()
+    mock_request = PydanticMessage()
+    
+    # Set up mock response for send_request_sync
+    mock_ack = Mock()
+    mock_ack._IsA.return_value = True
+    mock_host.send_request_sync.return_value = mock_ack
+    
+    # Test successful initialization
+    generator = _GeneratorRequest(mock_host, "test_target", mock_request)
+    assert generator.host == mock_host
+    assert generator.target == "test_target"
+    assert generator.request == mock_request
+    
+    # Verify send_request_sync was called correctly
+    mock_host.send_request_sync.assert_called_once()
+
+def test_generator_request_failed_initialization():
+    # Mock host and request
+    mock_host = Mock()
+    mock_request = PydanticMessage()
+    
+    # Set up mock response for send_request_sync to fail
+    mock_host.send_request_sync.return_value = None
+    
+    # Test failed initialization
+    with pytest.raises(RuntimeError, match="Failed to send request, no acknowledgment received."):
+        _GeneratorRequest(mock_host, "test_target", mock_request)
+
+def test_generator_request_iteration():
+    # Mock host and request
+    mock_host = Mock()
+    mock_request = PydanticMessage()
+    
+    # Set up mock responses for all calls: initialization, iteration, and stop
+    mock_ack = Mock()
+    mock_ack._IsA.return_value = True
+    mock_response = Mock()
+    mock_response._IsA.return_value = False
+    mock_stop = Mock()
+    mock_stop._IsA.return_value = True
+    
+    # Configure mock to return ack for init, regular response, then stop message
+    mock_host.send_request_sync.side_effect = [mock_ack, mock_response, mock_stop]
+    
+    # Create generator
+    generator = _GeneratorRequest(mock_host, "test_target", mock_request)
+    
+    # Test iteration
+    responses = list(generator)
+    assert len(responses) == 1
+    assert responses[0] == mock_response
+
+def test_generator_request_none_response():
+    # Mock host and request
+    mock_host = Mock()
+    mock_request = PydanticMessage()
+    
+    # Set up mock response for initialization
+    mock_ack = Mock()
+    mock_ack._IsA.return_value = True
+    mock_host.send_request_sync.return_value = mock_ack
+    
+    # Create generator
+    generator = _GeneratorRequest(mock_host, "test_target", mock_request)
+    
+    # Set up mock to return None for iteration
+    mock_host.send_request_sync.side_effect = [None]
+    
+    # Test iteration should stop on None response
+    responses = list(generator)
+    assert len(responses) == 0