Move query_available_service function outside

Author: ggqlq

Lib/test/test_socket.py

@@ -201,6 +201,36 @@ def _have_socket_hyperv():
         s.close()
     return True
 
+def _query_available_service(expected_protocols, services_file = '/etc/services'):
+    if not os.path.exists(services_file):
+        return None
+    services_found = dict()
+    with open(services_file, 'r') as f:
+        for line in f:
+            line = line.strip()
+            if line.startswith('#'):
+                # Skip comment line.
+                continue
+            tokens = line.split()
+            if len(tokens) < 2:
+                continue
+            if '/' not in tokens[1]:
+                continue
+            try:
+                _, entry_protocol = tokens[1].split('/')
+            except:
+                continue
+            entry_name = tokens[0]
+            if entry_name not in services_found:
+                services_found[entry_name] = [entry_protocol]
+            else:
+                services_found[entry_name].append(entry_protocol)
+            for protocol in expected_protocols:
+                if protocol not in services_found[entry_name]:
+                    break
+            else:
+                return entry_name
+    return None
 
 @contextlib.contextmanager
 def socket_setdefaulttimeout(timeout):
@@ -1259,38 +1289,6 @@ def testGetServBy(self):
         # Find one service that exists, then check all the related interfaces.
         # I've ordered this by protocols that have both a tcp and udp
         # protocol, at least for modern Linuxes.
-        def query_available_service(*protocols):
-            services_file = '/etc/services'
-            if not os.path.exists(services_file):
-                return None
-            services_found = dict()
-            with open(services_file, 'r') as f:
-                for line in f:
-                    line = line.strip()
-                    if line.startswith('#'):
-                        # Skip comment line.
-                        continue
-                    tokens = line.split()
-                    if len(tokens) < 2:
-                        continue
-                    if '/' not in tokens[1]:
-                        continue
-                    try:
-                        _, entry_protocol = tokens[1].split('/')
-                    except:
-                        continue
-                    entry_name = tokens[0]
-                    if entry_name not in services_found:
-                        services_found[entry_name] = [entry_protocol]
-                    else:
-                        services_found[entry_name].append(entry_protocol)
-                    for protocol in protocols:
-                        if protocol not in services_found[entry_name]:
-                            break
-                    else:
-                        return entry_name
-            return None
-
         if (
             sys.platform.startswith(
                 ('linux', 'android', 'freebsd', 'netbsd', 'gnukfreebsd'))
@@ -1308,7 +1306,7 @@ def query_available_service(*protocols):
             except OSError:
                 pass
         else:
-            service = query_available_service('tcp')
+            service = _query_available_service(('tcp'))
             if service is None:
                 self.skipTest('No available service found.')
             port = socket.getservbyname(service, 'tcp')