Add test and util functions for pending track

tchannel/tornado/connection.py

@@ -141,6 +141,10 @@ def __init__(self, connection, tchannel=None, direction=None):
 
         connection.set_close_callback(self._on_close)
 
+    @property
+    def num_out_pendings(self):
+        return len(self._out_pending_res) + len(self._out_pending_req)
+
     def set_close_callback(self, cb):
         """Specify a function to be called when this connection is closed.
 

tchannel/tornado/util.py

@@ -90,3 +90,12 @@ def go():
     go()
 
     return all_done_future
+
+
+def num_out_pendings(conns):
+    """Return the total number of out pending req/res among connections"""
+    num = 0
+    for con in conns:
+        num += con.num_out_pendings
+
+    return num

tests/tornado/test_connection.py

@@ -28,6 +28,7 @@
 from tchannel.messages import Types
 from tchannel import TChannel
 from tchannel.tornado.connection import StreamConnection
+from tchannel.tornado.util import num_out_pendings
 
 
 def dummy_headers():
@@ -81,3 +82,55 @@ def test_close_callback_is_called():
     conn.close()
 
     assert (yield cb_future)
+
+
+@pytest.mark.gen_test
+def test_pending_outgoing():
+    server = TChannel('server')
+    server.listen()
+
+    @server.raw.register
+    def hello(request):
+        return 'hi'
+
+    client = TChannel('client')
+    yield client.raw(
+        hostport=server.hostport,
+        body='work',
+        endpoint='hello',
+        service='server'
+    )
+
+    assert num_out_pendings(
+        client._dep_tchannel.peers.get(server.hostport).connections
+    ) == 0
+
+    assert num_out_pendings(
+        server._dep_tchannel.peers.peers[0].connections
+    ) == 0
+
+
+@pytest.mark.gen_test
+def test_pending_outgoing_mock():
+    server = TChannel('server')
+    server.listen()
+
+    @server.raw.register
+    def hello(request):
+        return 'hi'
+
+    client = TChannel('client')
+    yield client.raw(
+        hostport=server.hostport,
+        body='work',
+        endpoint='hello',
+        service='server'
+    )
+
+    assert num_out_pendings(
+        client._dep_tchannel.peers.get(server.hostport).connections
+    ) == 0
+
+    assert num_out_pendings(
+        server._dep_tchannel.peers.peers[0].connections
+    ) == 0