From a5503cc64a1ef44048c8aa10e36344508076201c Mon Sep 17 00:00:00 2001
From: Lidi Zheng <lidiz@google.com>
Date: Mon, 3 Feb 2020 16:46:48 -0800
Subject: [PATCH] Add unit test for channel ready

---
 .../tests_aio/unit/channel_ready_test.py      | 68 +++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 src/python/grpcio_tests/tests_aio/unit/channel_ready_test.py

diff --git a/src/python/grpcio_tests/tests_aio/unit/channel_ready_test.py b/src/python/grpcio_tests/tests_aio/unit/channel_ready_test.py
new file mode 100644
index 00000000000..cbdf2deda9f
--- /dev/null
+++ b/src/python/grpcio_tests/tests_aio/unit/channel_ready_test.py
@@ -0,0 +1,68 @@
+# Copyright 2020 The gRPC Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Testing the channel_ready function."""
+
+import asyncio
+import gc
+import logging
+import time
+import unittest
+
+import grpc
+from grpc.experimental import aio
+
+from tests.unit.framework.common import get_socket, test_constants
+from tests_aio.unit import _common
+from tests_aio.unit._test_base import AioTestBase
+from tests_aio.unit._test_server import start_test_server
+
+
+class TestChannelReady(AioTestBase):
+
+    async def setUp(self):
+        address, self._port, self._socket = get_socket(listen=False)
+        self._channel = aio.insecure_channel(f"{address}:{self._port}")
+        self._socket.close()
+
+    async def tearDown(self):
+        await self._channel.close()
+
+    async def test_channel_ready_success(self):
+        # Start `channel_ready` as another Task
+        channel_ready_task = self.loop.create_task(
+            aio.channel_ready(self._channel))
+
+        # Wait for TRANSIENT_FAILURE
+        await _common.block_until_certain_state(
+            self._channel, grpc.ChannelConnectivity.TRANSIENT_FAILURE)
+
+        try:
+            # Start the server
+            _, server = await start_test_server(port=self._port)
+
+            # The RPC should recover itself
+            await channel_ready_task
+        finally:
+            if server is not None:
+                await server.stop(None)
+
+    async def test_channel_ready_blocked(self):
+        with self.assertRaises(asyncio.TimeoutError):
+            await asyncio.wait_for(aio.channel_ready(self._channel),
+                                   test_constants.SHORT_TIMEOUT)
+
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG)
+    unittest.main(verbosity=2)