ISSUE #176

* Fix bug related with test's mock.

Author: Sergio García Prado

packages/core/minos-microservice-networks/tests/test_networks/test_brokers/test_collections/test_queues/test_database.py

@@ -75,9 +75,6 @@ async def test_aiter(self):
         ]
 
         queue = DatabaseBrokerQueue.from_config(self.config, operation_factory=self.operation_factory)
-        await queue.setup()
-        await queue.enqueue(messages[0])
-        await queue.enqueue(messages[1])
 
         with patch.object(
             MockedDatabaseClient,
@@ -90,6 +87,9 @@ async def test_aiter(self):
                 cycle([FakeAsyncIterator([(0,)])]),
             ),
         ):
+            await queue.setup()
+            await queue.enqueue(messages[0])
+            await queue.enqueue(messages[1])
 
             observed = list()
             async for message in queue:
@@ -110,11 +110,11 @@ async def test_dequeue_with_count(self):
             "fetch_all",
             return_value=FakeAsyncIterator([[1, messages[0].avro_bytes], [2, bytes()], [3, messages[1].avro_bytes]]),
         ):
-            async with DatabaseBrokerQueue.from_config(self.config, operation_factory=self.operation_factory) as queue:
-                queue._get_count = AsyncMock(side_effect=[3, 0])
+            queue = DatabaseBrokerQueue.from_config(self.config, operation_factory=self.operation_factory)
+            queue._get_count = AsyncMock(side_effect=[3, 0])
 
-                async with queue:
-                    observed = [await queue.dequeue(), await queue.dequeue()]
+            async with queue:
+                observed = [await queue.dequeue(), await queue.dequeue()]
 
         self.assertEqual(messages, observed)