Fix port number when use NodePort (dask#806)

dask_kubernetes/common/networking.py

@@ -56,18 +56,16 @@ async def get_external_address_for_scheduler_service(
     local_port=None,
 ):
     """Take a service object and return the scheduler address."""
-    [port] = [
-        port.port
-        for port in service.spec.ports
-        if port.name == service.metadata.name or port.name == port_name
-    ]
     if service.spec.type == "LoadBalancer":
+        port = _get_port(service)
         lb = service.status.load_balancer.ingress[0]
         host = lb.hostname or lb.ip
     elif service.spec.type == "NodePort":
+        port = _get_port(service, is_node_port=True)
         nodes = await core_api.list_node()
         host = nodes.items[0].status.addresses[0].address
     elif service.spec.type == "ClusterIP":
+        port = _get_port(service)
         if not port_forward_cluster_ip:
             with suppress(socket.gaierror):
                 # Try to resolve the service name. If we are inside the cluster this should succeed.
@@ -86,6 +84,16 @@ async def get_external_address_for_scheduler_service(
     return f"tcp://{host}:{port}"
 
 
+def _get_port(service, port_name, is_node_port=False):
+    """NodePort is a special case when we have to use node_port instead of node"""
+    [port] = [
+        port.port if not is_node_port else port.node_port
+        for port in service.spec.ports
+        if port.name == service.metadata.name or port.name == port_name
+    ]
+    return port
+
+
 async def _is_service_available(host, port, retries=20):
     for i in range(retries):
         try: