| Server IP : 82.148.16.210 / Your IP : 216.73.216.19 Web Server : nginx/1.29.5 System : Linux mail.sarafai.ru 6.1.0-43-amd64 #1 SMP PREEMPT_DYNAMIC Debian 6.1.162-1 (2026-02-08) x86_64 User : root ( 0) PHP Version : 7.4.33 Disable Function : pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wifcontinued,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_get_handler,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,pcntl_async_signals,pcntl_unshare, MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : OFF Directory : /lib/cloudinit/tests/ |
Upload File : |
# This file is part of cloud-init. See LICENSE file for license information.
"""Tests for cloudinit.temp_utils"""
from cloudinit.temp_utils import mkdtemp, mkstemp, tempdir
from cloudinit.tests.helpers import CiTestCase, wrap_and_call
import os
class TestTempUtils(CiTestCase):
def test_mkdtemp_default_non_root(self):
"""mkdtemp creates a dir under /tmp for the unprivileged."""
calls = []
def fake_mkdtemp(*args, **kwargs):
calls.append(kwargs)
return '/fake/return/path'
retval = wrap_and_call(
'cloudinit.temp_utils',
{'os.getuid': 1000,
'tempfile.mkdtemp': {'side_effect': fake_mkdtemp},
'_TMPDIR': {'new': None},
'os.path.isdir': True},
mkdtemp)
self.assertEqual('/fake/return/path', retval)
self.assertEqual([{'dir': '/tmp'}], calls)
def test_mkdtemp_default_non_root_needs_exe(self):
"""mkdtemp creates a dir under /var/tmp/cloud-init when needs_exe."""
calls = []
def fake_mkdtemp(*args, **kwargs):
calls.append(kwargs)
return '/fake/return/path'
retval = wrap_and_call(
'cloudinit.temp_utils',
{'os.getuid': 1000,
'tempfile.mkdtemp': {'side_effect': fake_mkdtemp},
'_TMPDIR': {'new': None},
'os.path.isdir': True},
mkdtemp, needs_exe=True)
self.assertEqual('/fake/return/path', retval)
self.assertEqual([{'dir': '/var/tmp/cloud-init'}], calls)
def test_mkdtemp_default_root(self):
"""mkdtemp creates a dir under /run/cloud-init for the privileged."""
calls = []
def fake_mkdtemp(*args, **kwargs):
calls.append(kwargs)
return '/fake/return/path'
retval = wrap_and_call(
'cloudinit.temp_utils',
{'os.getuid': 0,
'tempfile.mkdtemp': {'side_effect': fake_mkdtemp},
'_TMPDIR': {'new': None},
'os.path.isdir': True},
mkdtemp)
self.assertEqual('/fake/return/path', retval)
self.assertEqual([{'dir': '/run/cloud-init/tmp'}], calls)
def test_mkstemp_default_non_root(self):
"""mkstemp creates secure tempfile under /tmp for the unprivileged."""
calls = []
def fake_mkstemp(*args, **kwargs):
calls.append(kwargs)
return '/fake/return/path'
retval = wrap_and_call(
'cloudinit.temp_utils',
{'os.getuid': 1000,
'tempfile.mkstemp': {'side_effect': fake_mkstemp},
'_TMPDIR': {'new': None},
'os.path.isdir': True},
mkstemp)
self.assertEqual('/fake/return/path', retval)
self.assertEqual([{'dir': '/tmp'}], calls)
def test_mkstemp_default_root(self):
"""mkstemp creates a secure tempfile in /run/cloud-init for root."""
calls = []
def fake_mkstemp(*args, **kwargs):
calls.append(kwargs)
return '/fake/return/path'
retval = wrap_and_call(
'cloudinit.temp_utils',
{'os.getuid': 0,
'tempfile.mkstemp': {'side_effect': fake_mkstemp},
'_TMPDIR': {'new': None},
'os.path.isdir': True},
mkstemp)
self.assertEqual('/fake/return/path', retval)
self.assertEqual([{'dir': '/run/cloud-init/tmp'}], calls)
def test_tempdir_error_suppression(self):
"""test tempdir suppresses errors during directory removal."""
with self.assertRaises(OSError):
with tempdir(prefix='cloud-init-dhcp-') as tdir:
os.rmdir(tdir)
# As a result, the directory is already gone,
# so shutil.rmtree should raise OSError
with tempdir(rmtree_ignore_errors=True,
prefix='cloud-init-dhcp-') as tdir:
os.rmdir(tdir)
# Since the directory is already gone, shutil.rmtree would raise
# OSError, but we suppress that
# vi: ts=4 expandtab