Fix wait_for_ready example for ipv4-only environments like Kokoro

6 years ago · 1c65fd19e8
parent 4de4a7da7b
commit 1c65fd19e8
1 changed files with 6 additions and 2 deletions
--- a/examples/python/wait_for_ready/wait_for_ready_example.py
+++ b/examples/python/wait_for_ready/wait_for_ready_example.py
@ -33,8 +33,12 @@ _ONE_DAY_IN_SECONDS = 60 * 60 * 24

@contextmanager
 def get_free_loopback_tcp_port():
-    tcp_socket = socket.socket(socket.AF_INET6)
-    tcp_socket.bind(('', 0))
+    if socket.has_ipv6:
+        tcp_socket = socket.socket(socket.AF_INET6)
+        tcp_socket.bind(('', 0))
+    else:
+        tcp_socket = socket.socket(socket.AF_INET)
+        tcp_socket.bind(('', 0))
    address_tuple = tcp_socket.getsockname()
    yield "[::1]:%s" % (address_tuple[1])
    tcp_socket.close()