Merge pull request #333 from nats-io/sub-async-iter

Fix async iterator not decreasing pending data size

nats/aio/client.py

@@ -56,7 +56,7 @@
     Subscription,
 )
 
-__version__ = '2.1.5'
+__version__ = '2.1.6'
 __lang__ = 'python3'
 _logger = logging.getLogger(__name__)
 PROTOCOL = 1

nats/aio/subscription.py

@@ -334,6 +334,7 @@ async def __anext__(self) -> Msg:
         if get_task in finished:
             self._queue.task_done()
             msg = get_task.result()
+            self._sub._pending_size -= len(msg.data)
 
             # Unblock the iterator in case it has already received enough messages.
             if sub._max_msgs > 0 and sub._received >= sub._max_msgs:

tests/test_client.py

@@ -445,6 +445,7 @@ async def iterator_func(sub):
         self.assertEqual(5, len(msgs))
         self.assertEqual("tests.1", msgs[1].subject)
         self.assertEqual("tests.3", msgs[3].subject)
+        self.assertEqual(0, sub.pending_bytes)
         await nc.close()
 
     @async_test