got tests passing

Author: daniel-sanche

tests/unit/data/_async/test_metrics_interceptor.py

@@ -34,6 +34,7 @@
 __CROSS_SYNC_OUTPUT__ = "tests.unit.data._sync_autogen.test_metrics_interceptor"
 
 
+@CrossSync.drop
 class _AsyncIterator:
     """Helper class to wrap an iterator or async generator in an async iterator"""
 
@@ -271,7 +272,10 @@ async def test_unary_stream_interceptor_success(self):
 
         continuation = CrossSync.Mock()
         call = continuation.return_value
-        call.__aiter__ = mock.Mock(return_value=_AsyncIterator([1, 2]))
+        if CrossSync.is_async:
+            call.__aiter__ = mock.Mock(return_value=_AsyncIterator([1, 2]))
+        else:
+            call.__iter__ = mock.Mock(return_value=iter([1, 2]))
         call.trailing_metadata = CrossSync.Mock(return_value=[("a", "b")])
         call.initial_metadata = CrossSync.Mock(return_value=[("c", "d")])
         details = mock.Mock()
@@ -303,10 +307,16 @@ async def test_unary_stream_interceptor_failure_mid_stream(self):
 
         continuation = CrossSync.Mock()
         call = continuation.return_value
-        async def mock_generator():
-            yield 1
-            raise exc
-        call.__aiter__ = mock.Mock(return_value=_AsyncIterator(mock_generator()))
+        if CrossSync.is_async:
+            async def mock_generator():
+                yield 1
+                raise exc
+            call.__aiter__ = mock.Mock(return_value=_AsyncIterator(mock_generator()))
+        else:
+            def mock_generator():
+                yield 1
+                raise exc
+            call.__iter__ = mock.Mock(return_value=mock_generator())
         call.trailing_metadata = CrossSync.Mock(return_value=[("a", "b")])
         call.initial_metadata = CrossSync.Mock(return_value=[("c", "d")])
         details = mock.Mock()
@@ -427,6 +437,8 @@ async def test_unary_unary_interceptor_start_operation(self, initial_state):
         instance.operation_map[op.uuid] = op
         continuation = CrossSync.Mock()
         call = continuation.return_value
+        call.trailing_metadata = CrossSync.Mock(return_value=[])
+        call.initial_metadata = CrossSync.Mock(return_value=[])
         details = mock.Mock()
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()
@@ -451,7 +463,12 @@ async def test_unary_stream_interceptor_start_operation(self, initial_state):
 
         continuation = CrossSync.Mock()
         call = continuation.return_value
-        call.__aiter__ = mock.Mock(return_value=_AsyncIterator([1, 2]))
+        if CrossSync.is_async:
+            call.__aiter__ = mock.Mock(return_value=_AsyncIterator([1, 2]))
+        else:
+            call.__iter__ = mock.Mock(return_value=iter([1, 2]))
+        call.trailing_metadata = CrossSync.Mock(return_value=[])
+        call.initial_metadata = CrossSync.Mock(return_value=[])
         details = mock.Mock()
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()

tests/unit/data/_sync_autogen/test_metrics_interceptor.py

@@ -29,27 +29,6 @@
 )
 
 
-class _AsyncIterator:
-    """Helper class to wrap an iterator or async generator in an async iterator"""
-
-    def __init__(self, iterable):
-        if hasattr(iterable, "__anext__"):
-            self._iterator = iterable
-        else:
-            self._iterator = iter(iterable)
-
-    def __aiter__(self):
-        return self
-
-    async def __anext__(self):
-        if hasattr(self._iterator, "__anext__"):
-            return await self._iterator.__anext__()
-        try:
-            return next(self._iterator)
-        except StopIteration:
-            raise StopAsyncIteration
-
-
 class TestMetricsInterceptor:
     @staticmethod
     def _get_target_class():
@@ -247,7 +226,7 @@ def test_unary_stream_interceptor_success(self):
         instance.operation_map[op.uuid] = op
         continuation = CrossSync._Sync_Impl.Mock()
         call = continuation.return_value
-        call.__aiter__ = mock.Mock(return_value=_AsyncIterator([1, 2]))
+        call.__iter__ = mock.Mock(return_value=iter([1, 2]))
         call.trailing_metadata = CrossSync._Sync_Impl.Mock(return_value=[("a", "b")])
         call.initial_metadata = CrossSync._Sync_Impl.Mock(return_value=[("c", "d")])
         details = mock.Mock()
@@ -282,7 +261,7 @@ def mock_generator():
             yield 1
             raise exc
 
-        call.__aiter__ = mock.Mock(return_value=_AsyncIterator(mock_generator()))
+        call.__iter__ = mock.Mock(return_value=mock_generator())
         call.trailing_metadata = CrossSync._Sync_Impl.Mock(return_value=[("a", "b")])
         call.initial_metadata = CrossSync._Sync_Impl.Mock(return_value=[("c", "d")])
         details = mock.Mock()
@@ -396,6 +375,8 @@ def test_unary_unary_interceptor_start_operation(self, initial_state):
         instance.operation_map[op.uuid] = op
         continuation = CrossSync._Sync_Impl.Mock()
         call = continuation.return_value
+        call.trailing_metadata = CrossSync._Sync_Impl.Mock(return_value=[])
+        call.initial_metadata = CrossSync._Sync_Impl.Mock(return_value=[])
         details = mock.Mock()
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()
@@ -418,7 +399,9 @@ def test_unary_stream_interceptor_start_operation(self, initial_state):
         instance.operation_map[op.uuid] = op
         continuation = CrossSync._Sync_Impl.Mock()
         call = continuation.return_value
-        call.__aiter__ = mock.Mock(return_value=_AsyncIterator([1, 2]))
+        call.__iter__ = mock.Mock(return_value=iter([1, 2]))
+        call.trailing_metadata = CrossSync._Sync_Impl.Mock(return_value=[])
+        call.initial_metadata = CrossSync._Sync_Impl.Mock(return_value=[])
         details = mock.Mock()
         details.metadata = [(OPERATION_INTERCEPTOR_METADATA_KEY, op.uuid)]
         request = mock.Mock()