test: UNIX socket connect

Add test for `tarantool.connect` with UNIX socket. Add TarantoolServer
option to start a Tarantool instance which listens to UNIX socket.

Author: DifferentialOrange

test/suites/lib/tarantool_server.py

@@ -36,6 +36,26 @@ def find_port(port=None):
         port += 1
     return find_port(3300)
 
+def check_socket(bind_addr):
+    if os.path.exists(bind_addr):
+        return False
+
+    sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+    sock.bind(bind_addr)
+    sock.close()
+
+    return True
+
+def find_socket(bind_num=None):
+    if bind_num is None:
+        bind_num = random.randrange(0, 99)
+    while bind_num < 99:
+        bind_addr = f'/tmp/tarantool_{bind_num}.sock'
+        if check_socket(bind_addr):
+            return bind_addr
+        bind_num += 1
+    return find_socket(0)
+
 
 class RunnerException(object):
     pass
@@ -122,7 +142,8 @@ def __new__(cls,
                 ssl_key_file=None,
                 ssl_cert_file=None,
                 ssl_ca_file=None,
-                ssl_ciphers=None):
+                ssl_ciphers=None,
+                create_unix_socket=False):
         if os.name == 'nt':
             from .remote_tarantool_server import RemoteTarantoolServer
             return RemoteTarantoolServer()
@@ -133,12 +154,21 @@ def __init__(self,
                  ssl_key_file=None,
                  ssl_cert_file=None,
                  ssl_ca_file=None,
-                 ssl_ciphers=None):
+                 ssl_ciphers=None,
+                 create_unix_socket=False):
         os.popen('ulimit -c unlimited')
-        self.host = 'localhost'
-        self.args = {}
-        self.args['primary'] = find_port()
-        self.args['admin'] = find_port(self.args['primary'] + 1)
+
+        if create_unix_socket:
+            self.host = None
+            self.args = {}
+            self.args['primary'] = find_socket()
+            self.args['admin'] = find_port()
+        else:
+            self.host = 'localhost'
+            self.args = {}
+            self.args['primary'] = find_port()
+            self.args['admin'] = find_port(self.args['primary'] + 1)
+
         self._admin = self.args['admin']
         self.vardir = tempfile.mkdtemp(prefix='var_', dir=os.getcwd())
         self.find_exe()
@@ -244,6 +274,9 @@ def clean(self):
         if os.path.isdir(self.vardir):
             shutil.rmtree(self.vardir)
 
+        if os.path.exists(self.args['primary']):
+            os.remove(self.args['primary'])
+
     def __del__(self):
         self.stop()
         self.clean()

test/suites/test_dml.py

@@ -298,8 +298,26 @@ def test_12_update_fields(self):
             [[2, 'help', 7]]
         )
 
+    def test_13_unix_socket_connect(self):
+        if sys.platform.startswith("win"):
+            self.skipTest("Skip UNIX socket tests on Windows since it uses remote server")
+
+        self.sock_srv = TarantoolServer(create_unix_socket=True)
+        self.sock_srv.script = 'test/suites/box.lua'
+        self.sock_srv.start()
+
+        self.sock_con = tarantool.connect(self.sock_srv.host, self.sock_srv.args['primary'])
+        self.assertEqual(self.sock_con.ping(notime=True), "Success")
+
     @classmethod
     def tearDownClass(self):
         self.con.close()
         self.srv.stop()
         self.srv.clean()
+
+        if hasattr(self, 'sock_srv'):
+            self.sock_srv.stop()
+            self.sock_srv.clean()
+
+        if hasattr(self, 'sock_con'):
+            self.sock_con.close()