Mini Shell
# Copyright 2009 Google Inc. All Rights Reserved.
# Copyright 2015-2017 John McGehee
#
# 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.
# Disable attribute errors - attributes not be found in mixin (shall be cleaned up...)
# pytype: disable=attribute-error
"""Common helper classes used in tests, or as test class base."""
import os
import platform
import shutil
import stat
import sys
import tempfile
import unittest
from contextlib import contextmanager
from unittest import mock
from pyfakefs import fake_filesystem, fake_open, fake_os
from pyfakefs.helpers import is_byte_string, to_string
class DummyTime:
"""Mock replacement for time.time. Increases returned time on access."""
def __init__(self, curr_time, increment):
self.current_time = curr_time
self.increment = increment
def __call__(self, *args, **kwargs):
current_time = self.current_time
self.current_time += self.increment
return current_time
class DummyMock:
def start(self):
pass
def stop(self):
pass
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
pass
def time_mock(start=200, step=20):
return mock.patch("pyfakefs.helpers.now", DummyTime(start, step))
class TestCase(unittest.TestCase):
"""Test base class with some convenience methods and attributes"""
is_windows = sys.platform == "win32"
is_cygwin = sys.platform == "cygwin"
is_macos = sys.platform == "darwin"
symlinks_can_be_tested = None
def assert_mode_equal(self, expected, actual):
return self.assertEqual(stat.S_IMODE(expected), stat.S_IMODE(actual))
@contextmanager
def raises_os_error(self, subtype):
try:
yield
self.fail("No exception was raised, OSError expected")
except OSError as exc:
if isinstance(subtype, list):
self.assertIn(exc.errno, subtype)
else:
self.assertEqual(subtype, exc.errno)
class RealFsTestMixin:
"""Test mixin to allow tests to run both in the fake filesystem and in the
real filesystem.
To run tests in the real filesystem, a new test class can be derived from
the test class testing the fake filesystem which overwrites
`use_real_fs()` to return `True`.
All tests in the real file system operate inside the local temp path.
In order to make a test able to run in the real FS, it must not use the
fake filesystem functions directly. For access to `os` and `open`,
the respective attributes must be used, which point either to the native
or to the fake modules. A few convenience methods allow to compose
paths, create files and directories.
"""
def __init__(self):
self.filesystem = None
self.open = open
self.os = os
self.base_path = None
def setUp(self):
if not os.environ.get("TEST_REAL_FS"):
self.skip_real_fs()
if self.use_real_fs():
self.base_path = tempfile.mkdtemp()
def tearDown(self):
if self.use_real_fs():
self.os.chdir(os.path.dirname(self.base_path))
shutil.rmtree(self.base_path, ignore_errors=True)
os.chdir(self.cwd)
@property
def is_windows_fs(self):
return TestCase.is_windows
def set_windows_fs(self, value):
if self.filesystem is not None:
self.filesystem._is_windows_fs = value
if value:
self.filesystem._is_macos = False
self.create_basepath()
@property
def is_macos(self):
return TestCase.is_macos
@property
def is_pypy(self):
return platform.python_implementation() == "PyPy"
def use_real_fs(self):
"""Return True if the real file system shall be tested."""
return False
def setUpFileSystem(self):
pass
def path_separator(self):
"""Can be overwritten to use a specific separator in the
fake filesystem."""
if self.use_real_fs():
return os.path.sep
return "/"
def check_windows_only(self):
"""If called at test start, the real FS test is executed only under
Windows, and the fake filesystem test emulates a Windows system.
"""
if self.use_real_fs():
if not TestCase.is_windows:
raise unittest.SkipTest("Testing Windows specific functionality")
else:
self.set_windows_fs(True)
def check_linux_only(self):
"""If called at test start, the real FS test is executed only under
Linux, and the fake filesystem test emulates a Linux system.
"""
if self.use_real_fs():
if TestCase.is_macos or TestCase.is_windows:
raise unittest.SkipTest("Testing Linux specific functionality")
else:
self.set_windows_fs(False)
self.filesystem._is_macos = False
def check_macos_only(self):
"""If called at test start, the real FS test is executed only under
MacOS, and the fake filesystem test emulates a MacOS system.
"""
if self.use_real_fs():
if not TestCase.is_macos:
raise unittest.SkipTest("Testing MacOS specific functionality")
else:
self.set_windows_fs(False)
self.filesystem._is_macos = True
def check_linux_and_windows(self):
"""If called at test start, the real FS test is executed only under
Linux and Windows, and the fake filesystem test emulates a Linux
system under MacOS.
"""
if self.use_real_fs():
if TestCase.is_macos:
raise unittest.SkipTest("Testing non-MacOs functionality")
else:
self.filesystem._is_macos = False
def check_case_insensitive_fs(self):
"""If called at test start, the real FS test is executed only in a
case-insensitive FS (e.g. Windows or MacOS), and the fake filesystem
test emulates a case-insensitive FS under the running OS.
"""
if self.use_real_fs():
if not TestCase.is_macos and not TestCase.is_windows:
raise unittest.SkipTest(
"Testing case insensitive specific functionality"
)
else:
self.filesystem.is_case_sensitive = False
def check_case_sensitive_fs(self):
"""If called at test start, the real FS test is executed only in a
case-sensitive FS (e.g. under Linux), and the fake file system test
emulates a case-sensitive FS under the running OS.
"""
if self.use_real_fs():
if TestCase.is_macos or TestCase.is_windows:
raise unittest.SkipTest("Testing case sensitive specific functionality")
else:
self.filesystem.is_case_sensitive = True
def check_posix_only(self):
"""If called at test start, the real FS test is executed only under
Linux and MacOS, and the fake filesystem test emulates a Linux
system under Windows.
"""
if self.use_real_fs():
if TestCase.is_windows:
raise unittest.SkipTest("Testing Posix specific functionality")
else:
self.set_windows_fs(False)
def skip_real_fs(self):
"""If called at test start, no real FS test is executed."""
if self.use_real_fs():
raise unittest.SkipTest("Only tests fake FS")
def skip_real_fs_failure(
self,
skip_windows=True,
skip_posix=True,
skip_macos=True,
skip_linux=True,
):
"""If called at test start, no real FS test is executed for the given
conditions. This is used to mark tests that do not pass correctly under
certain systems and shall eventually be fixed.
"""
if True:
if self.use_real_fs() and (
TestCase.is_windows
and skip_windows
or not TestCase.is_windows
and skip_macos
and skip_linux
or TestCase.is_macos
and skip_macos
or not TestCase.is_windows
and not TestCase.is_macos
and skip_linux
or not TestCase.is_windows
and skip_posix
):
raise unittest.SkipTest(
"Skipping because FakeFS does not match real FS"
)
def symlink_can_be_tested(self, force_real_fs=False):
"""Used to check if symlinks and hard links can be tested under
Windows. All tests are skipped under Windows for Python versions
not supporting links, and real tests are skipped if running without
administrator rights.
"""
if not TestCase.is_windows or (not force_real_fs and not self.use_real_fs()):
return True
if TestCase.symlinks_can_be_tested is None:
if force_real_fs:
self.base_path = tempfile.mkdtemp()
link_path = self.make_path("link")
try:
self.os.symlink(self.base_path, link_path)
TestCase.symlinks_can_be_tested = True
self.os.remove(link_path)
except (OSError, NotImplementedError):
TestCase.symlinks_can_be_tested = False
if force_real_fs:
self.base_path = None
return TestCase.symlinks_can_be_tested
def skip_if_symlink_not_supported(self, force_real_fs=False):
"""If called at test start, tests are skipped if symlinks are not
supported."""
if not self.symlink_can_be_tested(force_real_fs):
raise unittest.SkipTest("Symlinks under Windows need admin privileges")
def make_path(self, *args):
"""Create a path with the given component(s). A base path is prepended
to the path which represents a temporary directory in the real FS,
and a fixed path in the fake filesystem.
Always use to compose absolute paths for tests also running in the
real FS.
"""
if isinstance(args[0], (list, tuple)):
path = self.base_path
for arg in args[0]:
path = self.os.path.join(path, to_string(arg))
return path
args = [to_string(arg) for arg in args]
return self.os.path.join(self.base_path, *args)
def create_dir(self, dir_path, perm=0o777):
"""Create the directory at `dir_path`, including subdirectories.
`dir_path` shall be composed using `make_path()`.
"""
if not dir_path:
return
existing_path = dir_path
components = []
while existing_path and not self.os.path.exists(existing_path):
existing_path, component = self.os.path.split(existing_path)
if not component and existing_path:
# existing path is a drive or UNC root
if not self.os.path.exists(existing_path):
self.filesystem.add_mount_point(existing_path)
break
components.insert(0, component)
for component in components:
existing_path = self.os.path.join(existing_path, component)
self.os.mkdir(existing_path)
self.os.chmod(existing_path, 0o777)
self.os.chmod(dir_path, perm)
def create_file(self, file_path, contents=None, encoding=None, perm=0o666):
"""Create the given file at `file_path` with optional contents,
including subdirectories. `file_path` shall be composed using
`make_path()`.
"""
self.create_dir(self.os.path.dirname(file_path))
mode = "wb" if encoding is not None or is_byte_string(contents) else "w"
if encoding is not None and contents is not None:
contents = contents.encode(encoding)
with self.open(file_path, mode) as f:
if contents is not None:
f.write(contents)
self.os.chmod(file_path, perm)
def create_symlink(self, link_path, target_path):
"""Create the path at `link_path`, and a symlink to this path at
`target_path`. `link_path` shall be composed using `make_path()`.
"""
self.create_dir(self.os.path.dirname(link_path))
self.os.symlink(target_path, link_path)
def check_contents(self, file_path, contents):
"""Compare `contents` with the contents of the file at `file_path`.
Asserts equality.
"""
mode = "rb" if is_byte_string(contents) else "r"
with self.open(file_path, mode) as f:
self.assertEqual(contents, f.read())
def create_basepath(self):
"""Create the path used as base path in `make_path`."""
if self.filesystem is not None:
old_base_path = self.base_path
self.base_path = self.filesystem.path_separator + "basepath"
if self.filesystem.is_windows_fs:
self.base_path = "C:" + self.base_path
if old_base_path != self.base_path:
if old_base_path is not None:
self.filesystem.reset()
if not self.filesystem.exists(self.base_path):
self.filesystem.create_dir(self.base_path)
if old_base_path is not None:
self.setUpFileSystem()
def assert_equal_paths(self, actual, expected):
if self.is_windows:
actual = str(actual).replace("\\\\?\\", "")
expected = str(expected).replace("\\\\?\\", "")
if os.name == "nt" and self.use_real_fs():
# work around a problem that the user name, but not the full
# path is shown as the short name
self.assertEqual(
self.path_with_short_username(actual),
self.path_with_short_username(expected),
)
else:
self.assertEqual(actual, expected)
elif self.is_macos:
self.assertEqual(
str(actual).replace("/private/var/", "/var/"),
str(expected).replace("/private/var/", "/var/"),
)
else:
self.assertEqual(actual, expected)
@staticmethod
def path_with_short_username(path):
components = path.split(os.sep)
if len(components) >= 3:
components[2] = components[2][:6].upper() + "~1"
return os.sep.join(components)
def mock_time(self, start=200, step=20):
if not self.use_real_fs():
return time_mock(start, step)
return DummyMock()
def assert_raises_os_error(self, subtype, expression, *args, **kwargs):
"""Asserts that a specific subtype of OSError is raised."""
try:
expression(*args, **kwargs)
self.fail("No exception was raised, OSError expected")
except OSError as exc:
if isinstance(subtype, list):
self.assertIn(exc.errno, subtype)
else:
self.assertEqual(subtype, exc.errno)
class RealFsTestCase(TestCase, RealFsTestMixin):
"""Can be used as base class for tests also running in the real
file system."""
def __init__(self, methodName="runTest"):
TestCase.__init__(self, methodName)
RealFsTestMixin.__init__(self)
def setUp(self):
RealFsTestMixin.setUp(self)
self.cwd = os.getcwd()
if not self.use_real_fs():
self.filesystem = fake_filesystem.FakeFilesystem(
path_separator=self.path_separator()
)
self.setup_fake_fs()
self.setUpFileSystem()
def setup_fake_fs(self):
if not self.use_real_fs():
self.open = fake_open.FakeFileOpen(self.filesystem)
self.os = fake_os.FakeOsModule(self.filesystem)
self.create_basepath()
def tearDown(self):
RealFsTestMixin.tearDown(self)
@property
def is_windows_fs(self):
if self.use_real_fs():
return self.is_windows
return self.filesystem.is_windows_fs
@property
def is_macos(self):
if self.use_real_fs():
return TestCase.is_macos
return self.filesystem.is_macos
Zerion Mini Shell 1.0