Add function to create a temporary directory

Change-Id: I9c468e86ca66d10e43f258fc7e276f8cbf5e1181

tobiko/shell/sh/__init__.py

@@ -22,6 +22,7 @@ from tobiko.shell.sh import _execute
 from tobiko.shell.sh import _hostname
 from tobiko.shell.sh import _io
 from tobiko.shell.sh import _local
+from tobiko.shell.sh import _mktemp
 from tobiko.shell.sh import _nameservers
 from tobiko.shell.sh import _process
 from tobiko.shell.sh import _ps
@@ -63,6 +64,8 @@ local_process = _local.local_process
 LocalShellProcessFixture = _local.LocalShellProcessFixture
 LocalExecutePathFixture = _local.LocalExecutePathFixture
 
+make_temp_dir = _mktemp.make_temp_dir
+
 ListNameserversFixture = _nameservers.ListNameserversFixture
 list_nameservers = _nameservers.list_nameservers
 

tobiko/shell/sh/_mktemp.py

@@ -0,0 +1,34 @@
+# Copyright (c) 2021 Red Hat, Inc.
+#
+# All Rights Reserved.
+#
+#    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.
+from __future__ import absolute_import
+
+import tobiko
+from tobiko.shell import ssh
+from tobiko.shell.sh import _execute
+
+
+def make_temp_dir(ssh_client: ssh.SSHClientType = None,
+                  sudo: bool = None) \
+        -> str:
+    test_case = tobiko.get_test_case()
+    dir_name: str = _execute.execute('mktemp -d',
+                                     ssh_client=ssh_client,
+                                     sudo=sudo).stdout.strip()
+    test_case.addCleanup(_execute.execute,
+                         f'rm -fR "{dir_name}"',
+                         ssh_client=ssh_client,
+                         sudo=sudo)
+    return dir_name

tobiko/tests/functional/shell/test_mktemp.py

@@ -0,0 +1,28 @@
+# Copyright (c) 2021 Red Hat, Inc.
+#
+# All Rights Reserved.
+#
+#    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.
+from __future__ import absolute_import
+
+import testtools
+
+from tobiko.shell import sh
+
+
+class MktempTest(testtools.TestCase):
+
+    def test_make_temp_dir(self):
+        temp_dir = sh.make_temp_dir()
+        files = sh.execute(f'ls "{temp_dir}"').stdout.splitlines()
+        self.assertEqual([], files)