Viewing file:  fake_filesystem_test.py (103.17 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# Copyright 2009 Google 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.
"""Unittest for fake_filesystem module."""
import contextlib
import errno
import os
import stat
import sys
import unittest
from pyfakefs import fake_filesystem, fake_os, fake_open
from pyfakefs.fake_filesystem import (
set_uid,
set_gid,
is_root,
reset_ids,
OSType,
)
from pyfakefs.helpers import IS_WIN
from pyfakefs.tests.test_utils import TestCase, RealFsTestCase, time_mock
class FakeDirectoryUnitTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.os = fake_os.FakeOsModule(self.filesystem)
self.time = time_mock(10, 1)
self.time.start()
self.fake_file = fake_filesystem.FakeFile(
"foobar", contents="dummy_file", filesystem=self.filesystem
)
self.fake_dir = fake_filesystem.FakeDirectory(
"somedir", filesystem=self.filesystem
)
def tearDown(self):
self.time.stop()
def test_new_file_and_directory(self):
self.assertTrue(stat.S_IFREG & self.fake_file.st_mode)
self.assertTrue(stat.S_IFDIR & self.fake_dir.st_mode)
self.assertEqual({}, self.fake_dir.entries)
def test_add_entry(self):
self.fake_dir.add_entry(self.fake_file)
self.assertEqual({"foobar": self.fake_file}, self.fake_dir.entries)
def test_get_entry(self):
self.fake_dir.add_entry(self.fake_file)
self.assertEqual(self.fake_file, self.fake_dir.get_entry("foobar"))
def test_path(self):
root_dir = self.filesystem.root_dir_name
self.filesystem.root.add_entry(self.fake_dir)
self.fake_dir.add_entry(self.fake_file)
self.assertEqual(f"{root_dir}somedir/foobar", self.fake_file.path)
self.assertEqual(f"{root_dir}somedir", self.fake_dir.path)
def test_path_with_drive(self):
self.filesystem.is_windows_fs = True
dir_path = "C:/foo/bar/baz"
self.filesystem.create_dir(dir_path)
dir_object = self.filesystem.get_object(dir_path)
self.assertEqual(dir_path, dir_object.path)
def test_path_after_chdir(self):
root_dir = self.filesystem.root_dir_name
dir_path = "/foo/bar/baz"
self.filesystem.create_dir(dir_path)
self.os.chdir(dir_path)
dir_object = self.filesystem.get_object(dir_path)
self.assertEqual(f"{root_dir}foo/bar/baz", dir_object.path)
def test_path_after_chdir_with_drive(self):
self.filesystem.is_windows_fs = True
dir_path = "C:/foo/bar/baz"
self.filesystem.create_dir(dir_path)
self.os.chdir(dir_path)
dir_object = self.filesystem.get_object(dir_path)
self.assertEqual(dir_path, dir_object.path)
def test_remove_entry(self):
self.fake_dir.add_entry(self.fake_file)
self.assertEqual(self.fake_file, self.fake_dir.get_entry("foobar"))
self.fake_dir.remove_entry("foobar")
with self.assertRaises(KeyError):
self.fake_dir.get_entry("foobar")
def test_should_throw_if_set_size_is_not_integer(self):
with self.raises_os_error(errno.ENOSPC):
self.fake_file.size = 0.1
def test_should_throw_if_set_size_is_negative(self):
with self.raises_os_error(errno.ENOSPC):
self.fake_file.size = -1
def test_produce_empty_file_if_set_size_is_zero(self):
self.fake_file.size = 0
self.assertEqual("", self.fake_file.contents)
def test_sets_content_empty_if_set_size_is_zero(self):
self.fake_file.size = 0
self.assertEqual("", self.fake_file.contents)
def test_truncate_file_if_size_is_smaller_than_current_size(self):
self.fake_file.size = 6
self.assertEqual("dummy_", self.fake_file.contents)
def test_leave_file_unchanged_if_size_is_equal_to_current_size(self):
self.fake_file.size = 10
self.assertEqual("dummy_file", self.fake_file.contents)
def test_set_contents_to_dir_raises(self):
# Regression test for #276
self.filesystem.is_windows_fs = True
with self.raises_os_error(errno.EISDIR):
self.fake_dir.set_contents("a")
self.filesystem.is_windows_fs = False
with self.raises_os_error(errno.EISDIR):
self.fake_dir.set_contents("a")
def test_pads_with_nullbytes_if_size_is_greater_than_current_size(self):
self.fake_file.size = 13
self.assertEqual("dummy_file\0\0\0", self.fake_file.contents)
def test_set_m_time(self):
self.assertEqual(10, self.fake_file.st_mtime)
self.fake_file.st_mtime = 14
self.assertEqual(14, self.fake_file.st_mtime)
self.fake_file.st_mtime = 131
self.assertEqual(131, self.fake_file.st_mtime)
def test_file_inode(self):
filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
fake_os_module = fake_os.FakeOsModule(filesystem)
file_path = "some_file1"
filesystem.create_file(file_path, contents="contents here1")
self.assertLess(0, fake_os_module.stat(file_path)[stat.ST_INO])
file_obj = filesystem.get_object(file_path)
file_obj.st_ino = 43
self.assertEqual(43, fake_os_module.stat(file_path)[stat.ST_INO])
def test_directory_inode(self):
filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
fake_os_module = fake_os.FakeOsModule(filesystem)
dirpath = "testdir"
filesystem.create_dir(dirpath)
self.assertLess(0, fake_os_module.stat(dirpath)[stat.ST_INO])
dir_obj = filesystem.get_object(dirpath)
dir_obj.st_ino = 43
self.assertEqual(43, fake_os_module.stat(dirpath)[stat.ST_INO])
def test_directory_size(self):
fs = fake_filesystem.FakeFilesystem(path_separator="/")
foo_dir = fs.create_dir("/foo")
fs.create_file("/foo/bar.txt", st_size=20)
bar_dir = fs.create_dir("/foo/bar/")
fs.create_file("/foo/bar/baz1.txt", st_size=30)
fs.create_file("/foo/bar/baz2.txt", st_size=40)
foo1_dir = fs.create_dir("/foo1")
fs.create_file("/foo1/bar.txt", st_size=50)
fs.create_file("/foo1/bar/baz/file", st_size=60)
self.assertEqual(90, foo_dir.size)
self.assertEqual(70, bar_dir.size)
self.assertEqual(110, foo1_dir.size)
self.assertEqual(200, fs.root_dir.size)
with self.raises_os_error(errno.EISDIR):
foo1_dir.size = 100
def test_ordered_dirs(self):
filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
filesystem.create_dir("/foo")
filesystem.create_file("/foo/2")
filesystem.create_file("/foo/4")
filesystem.create_file("/foo/1")
filesystem.create_file("/foo/3")
fake_dir = filesystem.get_object("/foo")
self.assertEqual(["2", "4", "1", "3"], fake_dir.ordered_dirs)
class SetLargeFileSizeTest(TestCase):
def setUp(self):
filesystem = fake_filesystem.FakeFilesystem()
self.fake_file = fake_filesystem.FakeFile("foobar", filesystem=filesystem)
def test_should_throw_if_size_is_not_integer(self):
with self.raises_os_error(errno.ENOSPC):
self.fake_file.set_large_file_size(0.1)
def test_should_throw_if_size_is_negative(self):
with self.raises_os_error(errno.ENOSPC):
self.fake_file.set_large_file_size(-1)
def test_sets_content_none_if_size_is_non_negative_integer(self):
self.fake_file.set_large_file_size(1000000000)
self.assertEqual(None, self.fake_file.contents)
self.assertEqual(1000000000, self.fake_file.st_size)
class NormalizePathTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.root_name = self.filesystem.root_dir_name
def test_empty_path_should_get_normalized_to_root_path(self):
self.assertEqual(self.root_name, self.filesystem.absnormpath(""))
def test_root_path_remains_unchanged(self):
self.assertEqual(self.root_name, self.filesystem.absnormpath(self.root_name))
def test_relative_path_forced_to_cwd(self):
path = "bar"
self.filesystem.cwd = "/foo"
self.assertEqual("/foo/bar", self.filesystem.absnormpath(path))
def test_absolute_path_remains_unchanged(self):
path = "foo/bar"
self.assertEqual(self.root_name + path, self.filesystem.absnormpath(path))
def test_dotted_path_is_normalized(self):
path = "/foo/.."
self.assertEqual(
self.filesystem.root_dir_name, self.filesystem.absnormpath(path)
)
path = "foo/../bar"
self.assertEqual(
f"{self.filesystem.root_dir_name}bar",
self.filesystem.absnormpath(path),
)
def test_dot_path_is_normalized(self):
path = "."
self.assertEqual(self.root_name, self.filesystem.absnormpath(path))
class GetPathComponentsTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.root_name = "/"
def test_root_path_should_return_empty_list(self):
self.assertEqual([], self.filesystem._path_components(self.root_name))
def test_empty_path_should_return_empty_list(self):
self.assertEqual([], self.filesystem._path_components(""))
def test_relative_path_with_one_component_should_return_component(self):
self.assertEqual(["foo"], self.filesystem._path_components("foo"))
def test_absolute_path_with_one_component_should_return_component(self):
self.assertEqual(["foo"], self.filesystem._path_components("/foo"))
def test_two_level_relative_path_should_return_components(self):
self.assertEqual(["foo", "bar"], self.filesystem._path_components("foo/bar"))
def test_two_level_absolute_path_should_return_components(self):
self.assertEqual(["foo", "bar"], self.filesystem._path_components("/foo/bar"))
class FakeFilesystemUnitTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.root_name = self.filesystem.root_dir_name
self.fake_file = fake_filesystem.FakeFile("foobar", filesystem=self.filesystem)
self.fake_child = fake_filesystem.FakeDirectory(
"foobaz", filesystem=self.filesystem
)
self.fake_grandchild = fake_filesystem.FakeDirectory(
"quux", filesystem=self.filesystem
)
def test_new_filesystem(self):
self.assertEqual("/", self.filesystem.path_separator)
self.assertTrue(stat.S_IFDIR & self.filesystem.root.st_mode)
self.assertEqual({}, self.filesystem.root_dir.entries)
def test_none_raises_type_error(self):
with self.assertRaises(TypeError):
self.filesystem.exists(None)
def test_empty_string_does_not_exist(self):
self.assertFalse(self.filesystem.exists(""))
def test_exists_root(self):
self.assertTrue(self.filesystem.exists(self.root_name))
def test_exists_unadded_file(self):
self.assertFalse(self.filesystem.exists(self.fake_file.name))
def test_not_exists_subpath_named_like_file_contents(self):
# Regression test for #219
file_path = "/foo/bar"
self.filesystem.create_file(file_path, contents="baz")
self.assertFalse(self.filesystem.exists(file_path + "/baz"))
def test_get_root_object(self):
self.assertEqual(
self.filesystem.root_dir,
self.filesystem.get_object(self.root_name),
)
def test_add_object_to_root(self):
self.filesystem.add_object(self.root_name, self.fake_file)
self.assertEqual({"foobar": self.fake_file}, self.filesystem.root_dir.entries)
def test_windows_root_dir_name(self):
self.filesystem.is_windows_fs = True
self.assertEqual("C:/", self.filesystem.root_dir_name)
self.filesystem.cwd = "E:/foo"
self.assertEqual("E:/", self.filesystem.root_dir_name)
self.filesystem.cwd = "//foo/bar"
self.assertEqual("//foo/bar/", self.filesystem.root_dir_name)
def test_exists_added_file(self):
self.filesystem.add_object(self.root_name, self.fake_file)
self.assertTrue(self.filesystem.exists(self.fake_file.name))
def test_exists_relative_path_posix(self):
self.filesystem.is_windows_fs = False
self.filesystem.create_file("/a/b/file_one")
self.filesystem.create_file("/a/c/file_two")
self.assertTrue(self.filesystem.exists("a/b/../c/file_two"))
self.assertTrue(self.filesystem.exists("/a/c/../b/file_one"))
self.assertTrue(self.filesystem.exists("/a/c/../../a/b/file_one"))
self.assertFalse(self.filesystem.exists("a/b/../z/d"))
self.assertFalse(self.filesystem.exists("a/b/../z/../c/file_two"))
self.filesystem.cwd = "/a/c"
self.assertTrue(self.filesystem.exists("../b/file_one"))
self.assertTrue(self.filesystem.exists("../../a/b/file_one"))
self.assertTrue(self.filesystem.exists("../../a/b/../../a/c/file_two"))
self.assertFalse(self.filesystem.exists("../z/file_one"))
self.assertFalse(self.filesystem.exists("../z/../c/file_two"))
def test_exists_relative_path_windows(self):
self.filesystem.is_windows_fs = True
self.filesystem.is_macos = False
self.filesystem.create_file("/a/b/file_one")
self.filesystem.create_file("/a/c/file_two")
self.assertTrue(self.filesystem.exists("a/b/../c/file_two"))
self.assertTrue(self.filesystem.exists("/a/c/../b/file_one"))
self.assertTrue(self.filesystem.exists("/a/c/../../a/b/file_one"))
self.assertFalse(self.filesystem.exists("a/b/../z/d"))
self.assertTrue(self.filesystem.exists("a/b/../z/../c/file_two"))
self.filesystem.cwd = "C:/a/c"
self.assertTrue(self.filesystem.exists("../b/file_one"))
self.assertTrue(self.filesystem.exists("../../a/b/file_one"))
self.assertTrue(self.filesystem.exists("../../a/b/../../a/c/file_two"))
self.assertFalse(self.filesystem.exists("../z/file_one"))
self.assertTrue(self.filesystem.exists("../z/../c/file_two"))
def test_get_object_from_root(self):
self.filesystem.add_object(self.root_name, self.fake_file)
self.assertEqual(self.fake_file, self.filesystem.get_object("foobar"))
def test_get_nonexistent_object_from_root_error(self):
self.filesystem.add_object(self.root_name, self.fake_file)
self.assertEqual(self.fake_file, self.filesystem.get_object("foobar"))
with self.raises_os_error(errno.ENOENT):
self.filesystem.get_object("some_bogus_filename")
def test_remove_object_from_root(self):
self.filesystem.add_object(self.root_name, self.fake_file)
self.filesystem.remove_object(self.fake_file.name)
with self.raises_os_error(errno.ENOENT):
self.filesystem.get_object(self.fake_file.name)
def test_remove_nonexisten_object_from_root_error(self):
with self.raises_os_error(errno.ENOENT):
self.filesystem.remove_object("some_bogus_filename")
def test_exists_removed_file(self):
self.filesystem.add_object(self.root_name, self.fake_file)
self.filesystem.remove_object(self.fake_file.name)
self.assertFalse(self.filesystem.exists(self.fake_file.name))
def test_add_object_to_child(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_file)
self.assertEqual(
{self.fake_file.name: self.fake_file},
self.filesystem.root_dir.get_entry(self.fake_child.name).entries,
)
def test_add_object_to_regular_file_error_posix(self):
self.filesystem.is_windows_fs = False
self.filesystem.add_object(self.filesystem.root_dir_name, self.fake_file)
with self.raises_os_error(errno.ENOTDIR):
self.filesystem.add_object(self.fake_file.name, self.fake_file)
def test_add_object_to_regular_file_error_windows(self):
self.filesystem.is_windows_fs = True
self.filesystem.add_object(self.root_name, self.fake_file)
with self.raises_os_error(errno.ENOENT):
self.filesystem.add_object(self.fake_file.name, self.fake_file)
def test_exists_file_added_to_child(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_file)
path = self.filesystem.joinpaths(self.fake_child.name, self.fake_file.name)
self.assertTrue(self.filesystem.exists(path))
def test_get_object_from_child(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_file)
self.assertEqual(
self.fake_file,
self.filesystem.get_object(
self.filesystem.joinpaths(self.fake_child.name, self.fake_file.name)
),
)
def test_get_nonexistent_object_from_child_error(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_file)
with self.raises_os_error(errno.ENOENT):
self.filesystem.get_object(
self.filesystem.joinpaths(self.fake_child.name, "some_bogus_filename")
)
def test_remove_object_from_child(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_file)
target_path = self.filesystem.joinpaths(
self.fake_child.name, self.fake_file.name
)
self.filesystem.remove_object(target_path)
with self.raises_os_error(errno.ENOENT):
self.filesystem.get_object(target_path)
def test_remove_object_from_child_error(self):
self.filesystem.add_object(self.root_name, self.fake_child)
with self.raises_os_error(errno.ENOENT):
self.filesystem.remove_object(
self.filesystem.joinpaths(self.fake_child.name, "some_bogus_filename")
)
def test_remove_object_from_non_directory_error(self):
self.filesystem.add_object(self.root_name, self.fake_file)
with self.raises_os_error(errno.ENOTDIR):
self.filesystem.remove_object(
self.filesystem.joinpaths(
"%s" % self.fake_file.name,
"file_does_not_matter_since_parent_not_a_directory",
)
)
def test_exists_file_removed_from_child(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_file)
path = self.filesystem.joinpaths(self.fake_child.name, self.fake_file.name)
self.filesystem.remove_object(path)
self.assertFalse(self.filesystem.exists(path))
def test_operate_on_grandchild_directory(self):
self.filesystem.add_object(self.root_name, self.fake_child)
self.filesystem.add_object(self.fake_child.name, self.fake_grandchild)
grandchild_directory = self.filesystem.joinpaths(
self.fake_child.name, self.fake_grandchild.name
)
grandchild_file = self.filesystem.joinpaths(
grandchild_directory, self.fake_file.name
)
with self.assertRaises(OSError):
self.filesystem.get_object(grandchild_file)
self.filesystem.add_object(grandchild_directory, self.fake_file)
self.assertEqual(self.fake_file, self.filesystem.get_object(grandchild_file))
self.assertTrue(self.filesystem.exists(grandchild_file))
self.filesystem.remove_object(grandchild_file)
with self.assertRaises(OSError):
self.filesystem.get_object(grandchild_file)
self.assertFalse(self.filesystem.exists(grandchild_file))
def test_create_directory_in_root_directory(self):
path = "foo"
self.filesystem.create_dir(path)
new_dir = self.filesystem.get_object(path)
self.assertEqual(os.path.basename(path), new_dir.name)
self.assertTrue(stat.S_IFDIR & new_dir.st_mode)
def test_create_directory_in_root_directory_already_exists_error(self):
path = "foo"
self.filesystem.create_dir(path)
with self.raises_os_error(errno.EEXIST):
self.filesystem.create_dir(path)
def test_create_directory(self):
path = "foo/bar/baz"
self.filesystem.create_dir(path)
new_dir = self.filesystem.get_object(path)
self.assertEqual(os.path.basename(path), new_dir.name)
self.assertTrue(stat.S_IFDIR & new_dir.st_mode)
# Create second directory to make sure first is OK.
path = "%s/quux" % path
self.filesystem.create_dir(path)
new_dir = self.filesystem.get_object(path)
self.assertEqual(os.path.basename(path), new_dir.name)
self.assertTrue(stat.S_IFDIR & new_dir.st_mode)
def test_create_directory_already_exists_error(self):
path = "foo/bar/baz"
self.filesystem.create_dir(path)
with self.raises_os_error(errno.EEXIST):
self.filesystem.create_dir(path)
def test_create_file_in_read_only_directory_raises_in_posix(self):
self.filesystem.is_windows_fs = False
dir_path = "/foo/bar"
self.filesystem.create_dir(dir_path, perm_bits=0o555)
file_path = dir_path + "/baz"
if not is_root():
with self.raises_os_error(errno.EACCES):
self.filesystem.create_file(file_path)
else:
self.filesystem.create_file(file_path)
self.assertTrue(self.filesystem.exists(file_path))
def test_create_file_in_read_only_directory_possible_in_windows(self):
self.filesystem.is_windows_fs = True
dir_path = "C:/foo/bar"
self.filesystem.create_dir(dir_path, perm_bits=0o555)
file_path = dir_path + "/baz"
self.filesystem.create_file(file_path)
self.assertTrue(self.filesystem.exists(file_path))
def test_create_file_in_current_directory(self):
path = "foo"
contents = "dummy data"
self.filesystem.create_file(path, contents=contents)
self.assertTrue(self.filesystem.exists(path))
self.assertFalse(self.filesystem.exists(os.path.dirname(path)))
path = "./%s" % path
self.assertTrue(self.filesystem.exists(os.path.dirname(path)))
def test_create_file_in_root_directory(self):
path = "/foo"
contents = "dummy data"
self.filesystem.create_file(path, contents=contents)
new_file = self.filesystem.get_object(path)
self.assertTrue(self.filesystem.exists(path))
self.assertTrue(self.filesystem.exists(os.path.dirname(path)))
self.assertEqual(os.path.basename(path), new_file.name)
self.assertTrue(stat.S_IFREG & new_file.st_mode)
self.assertEqual(contents, new_file.contents)
def test_create_file_with_size_but_no_content_creates_large_file(self):
path = "large_foo_bar"
self.filesystem.create_file(path, st_size=100000000)
new_file = self.filesystem.get_object(path)
self.assertEqual(None, new_file.contents)
self.assertEqual(100000000, new_file.st_size)
def test_create_file_in_root_directory_already_exists_error(self):
path = "foo"
self.filesystem.create_file(path)
with self.raises_os_error(errno.EEXIST):
self.filesystem.create_file(path)
def test_create_file(self):
path = "foo/bar/baz"
retval = self.filesystem.create_file(path, contents="dummy_data")
self.assertTrue(self.filesystem.exists(path))
self.assertTrue(self.filesystem.exists(os.path.dirname(path)))
new_file = self.filesystem.get_object(path)
self.assertEqual(os.path.basename(path), new_file.name)
if IS_WIN:
self.assertEqual(1, new_file.st_uid)
self.assertEqual(1, new_file.st_gid)
else:
self.assertEqual(os.getuid(), new_file.st_uid)
self.assertEqual(os.getgid(), new_file.st_gid)
self.assertEqual(new_file, retval)
def test_create_file_with_changed_ids(self):
path = "foo/bar/baz"
set_uid(42)
set_gid(2)
self.filesystem.create_file(path)
self.assertTrue(self.filesystem.exists(path))
new_file = self.filesystem.get_object(path)
self.assertEqual(42, new_file.st_uid)
self.assertEqual(2, new_file.st_gid)
reset_ids()
def test_empty_file_created_for_none_contents(self):
fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
path = "foo/bar/baz"
self.filesystem.create_file(path, contents=None)
with fake_open(path) as f:
self.assertEqual("", f.read())
def test_create_file_with_incorrect_mode_type(self):
with self.assertRaises(TypeError):
self.filesystem.create_file("foo", "bar")
def test_create_file_already_exists_error(self):
path = "foo/bar/baz"
self.filesystem.create_file(path, contents="dummy_data")
with self.raises_os_error(errno.EEXIST):
self.filesystem.create_file(path)
def test_create_link(self):
path = "foo/bar/baz"
target_path = "foo/bar/quux"
new_file = self.filesystem.create_symlink(path, "quux")
# Neither the path nor the final target exists before we actually
# write to one of them, even though the link appears in the file
# system.
self.assertFalse(self.filesystem.exists(path))
self.assertFalse(self.filesystem.exists(target_path))
self.assertTrue(stat.S_IFLNK & new_file.st_mode)
# but once we write the linked to file, they both will exist.
self.filesystem.create_file(target_path)
self.assertTrue(self.filesystem.exists(path))
self.assertTrue(self.filesystem.exists(target_path))
def test_resolve_object(self):
target_path = "dir/target"
target_contents = "0123456789ABCDEF"
link_name = "x"
self.filesystem.create_dir("dir")
self.filesystem.create_file("dir/target", contents=target_contents)
self.filesystem.create_symlink(link_name, target_path)
obj = self.filesystem.resolve(link_name)
self.assertEqual("target", obj.name)
self.assertEqual(target_contents, obj.contents)
def check_lresolve_object(self):
target_path = "dir/target"
target_contents = "0123456789ABCDEF"
link_name = "x"
self.filesystem.create_dir("dir")
self.filesystem.create_file("dir/target", contents=target_contents)
self.filesystem.create_symlink(link_name, target_path)
obj = self.filesystem.lresolve(link_name)
self.assertEqual(link_name, obj.name)
self.assertEqual(target_path, obj.contents)
def test_lresolve_object_windows(self):
self.filesystem.is_windows_fs = True
self.check_lresolve_object()
def test_lresolve_object_posix(self):
self.filesystem.is_windows_fs = False
self.check_lresolve_object()
def check_directory_access_on_file(self, error_subtype):
self.filesystem.create_file("not_a_dir")
with self.raises_os_error(error_subtype):
self.filesystem.resolve("not_a_dir/foo")
with self.raises_os_error(error_subtype):
self.filesystem.lresolve("not_a_dir/foo/bar")
def test_directory_access_on_file_windows(self):
self.filesystem.is_windows_fs = True
self.check_directory_access_on_file(errno.ENOENT)
def test_directory_access_on_file_posix(self):
self.filesystem.is_windows_fs = False
self.check_directory_access_on_file(errno.ENOTDIR)
def test_pickle_fs(self):
"""Regression test for #445"""
import pickle
self.filesystem.open_files = []
p = pickle.dumps(self.filesystem)
fs = pickle.loads(p)
self.assertEqual(str(fs.root), str(self.filesystem.root))
self.assertEqual(fs.mount_points, self.filesystem.mount_points)
class CaseInsensitiveFakeFilesystemTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.filesystem.is_case_sensitive = False
self.os = fake_os.FakeOsModule(self.filesystem)
self.path = self.os.path
def test_get_object(self):
self.filesystem.create_dir("/foo/bar")
self.filesystem.create_file("/foo/bar/baz")
self.assertTrue(self.filesystem.get_object("/Foo/Bar/Baz"))
def test_remove_object(self):
self.filesystem.create_dir("/foo/bar")
self.filesystem.create_file("/foo/bar/baz")
self.filesystem.remove_object("/Foo/Bar/Baz")
self.assertFalse(self.filesystem.exists("/foo/bar/baz"))
def test_exists(self):
self.filesystem.create_dir("/Foo/Bar")
self.assertTrue(self.filesystem.exists("/Foo/Bar"))
self.assertTrue(self.filesystem.exists("/foo/bar"))
self.filesystem.create_file("/foo/Bar/baz")
self.assertTrue(self.filesystem.exists("/Foo/bar/BAZ"))
self.assertTrue(self.filesystem.exists("/foo/bar/baz"))
def test_create_directory_with_different_case_root(self):
self.filesystem.create_dir("/Foo/Bar")
self.filesystem.create_dir("/foo/bar/baz")
dir1 = self.filesystem.get_object("/Foo/Bar")
dir2 = self.filesystem.get_object("/foo/bar")
self.assertEqual(dir1, dir2)
def test_create_file_with_different_case_dir(self):
self.filesystem.create_dir("/Foo/Bar")
self.filesystem.create_file("/foo/bar/baz")
dir1 = self.filesystem.get_object("/Foo/Bar")
dir2 = self.filesystem.get_object("/foo/bar")
self.assertEqual(dir1, dir2)
def test_resolve_path(self):
self.filesystem.create_dir("/foo/baz")
self.filesystem.create_symlink("/Foo/Bar", "./baz/bip")
self.assertEqual(
f"{self.filesystem.root_dir_name}foo/baz/bip",
self.filesystem.resolve_path("/foo/bar"),
)
def test_isdir_isfile(self):
self.filesystem.create_file("foo/bar")
self.assertTrue(self.path.isdir("Foo"))
self.assertFalse(self.path.isfile("Foo"))
self.assertTrue(self.path.isfile("Foo/Bar"))
self.assertFalse(self.path.isdir("Foo/Bar"))
def test_getsize(self):
file_path = "foo/bar/baz"
self.filesystem.create_file(file_path, contents="1234567")
self.assertEqual(7, self.path.getsize("FOO/BAR/BAZ"))
def test_getsize_with_looping_symlink(self):
self.filesystem.is_windows_fs = False
dir_path = "/foo/bar"
self.filesystem.create_dir(dir_path)
link_path = dir_path + "/link"
link_target = link_path + "/link"
self.os.symlink(link_target, link_path)
with self.raises_os_error(errno.ELOOP):
self.os.path.getsize(link_path)
def test_get_mtime(self):
test_file = self.filesystem.create_file("foo/bar1.txt")
test_file.st_mtime = 24
self.assertEqual(24, self.path.getmtime("Foo/Bar1.TXT"))
def test_get_object_with_file_size(self):
self.filesystem.create_file("/Foo/Bar", st_size=10)
self.assertTrue(self.filesystem.get_object("/foo/bar"))
class CaseSensitiveFakeFilesystemTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.filesystem.is_case_sensitive = True
self.os = fake_os.FakeOsModule(self.filesystem)
self.path = self.os.path
def test_get_object(self):
self.filesystem.create_dir("/foo/bar")
self.filesystem.create_file("/foo/bar/baz")
with self.assertRaises(OSError):
self.filesystem.get_object("/Foo/Bar/Baz")
def test_remove_object(self):
self.filesystem.create_dir("/foo/bar")
self.filesystem.create_file("/foo/bar/baz")
with self.assertRaises(OSError):
self.filesystem.remove_object("/Foo/Bar/Baz")
self.assertTrue(self.filesystem.exists("/foo/bar/baz"))
def test_exists(self):
self.filesystem.create_dir("/Foo/Bar")
self.assertTrue(self.filesystem.exists("/Foo/Bar"))
self.assertFalse(self.filesystem.exists("/foo/bar"))
self.filesystem.create_file("/foo/Bar/baz")
self.assertFalse(self.filesystem.exists("/Foo/bar/BAZ"))
self.assertFalse(self.filesystem.exists("/foo/bar/baz"))
def test_create_directory_with_different_case_root(self):
self.filesystem.create_dir("/Foo/Bar")
self.filesystem.create_dir("/foo/bar/baz")
dir1 = self.filesystem.get_object("/Foo/Bar")
dir2 = self.filesystem.get_object("/foo/bar")
self.assertNotEqual(dir1, dir2)
def test_create_file_with_different_case_dir(self):
self.filesystem.create_dir("/Foo/Bar")
self.filesystem.create_file("/foo/bar/baz")
dir1 = self.filesystem.get_object("/Foo/Bar")
dir2 = self.filesystem.get_object("/foo/bar")
self.assertNotEqual(dir1, dir2)
def test_isdir_isfile(self):
self.filesystem.create_file("foo/bar")
self.assertFalse(self.path.isdir("Foo"))
self.assertFalse(self.path.isfile("Foo"))
self.assertFalse(self.path.isfile("Foo/Bar"))
self.assertFalse(self.path.isdir("Foo/Bar"))
def test_getsize(self):
file_path = "foo/bar/baz"
self.filesystem.create_file(file_path, contents="1234567")
with self.assertRaises(os.error):
self.path.getsize("FOO/BAR/BAZ")
def test_get_mtime(self):
test_file = self.filesystem.create_file("foo/bar1.txt")
test_file.st_mtime = 24
with self.raises_os_error(errno.ENOENT):
self.path.getmtime("Foo/Bar1.TXT")
class OsPathInjectionRegressionTest(TestCase):
"""Test faking os.path before calling os.walk.
Found when investigating a problem with
gws/tools/labrat/rat_utils_unittest, which was faking out os.path
before calling os.walk.
"""
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.os_path = os.path
# The bug was that when os.path gets faked, the FakePathModule doesn't
# get called in self.os.walk(). FakePathModule now insists that it is
# created as part of FakeOsModule.
self.os = fake_os.FakeOsModule(self.filesystem)
def tearDown(self):
os.path = self.os_path
def test_create_top_level_directory(self):
top_level_dir = "/x"
self.assertFalse(self.filesystem.exists(top_level_dir))
self.filesystem.create_dir(top_level_dir)
self.assertTrue(self.filesystem.exists("/"))
self.assertTrue(self.filesystem.exists(top_level_dir))
self.filesystem.create_dir("%s/po" % top_level_dir)
self.filesystem.create_file("%s/po/control" % top_level_dir)
self.filesystem.create_file("%s/po/experiment" % top_level_dir)
self.filesystem.create_dir("%s/gv" % top_level_dir)
self.filesystem.create_file("%s/gv/control" % top_level_dir)
expected = [
("/", ["x"], []),
("/x", ["gv", "po"], []),
("/x/gv", [], ["control"]),
("/x/po", [], ["control", "experiment"]),
]
# as the result is unsorted, we have to check against sorted results
result = sorted([step for step in self.os.walk("/")], key=lambda v: v[0])
self.assertEqual(len(expected), len(result))
for entry, expected_entry in zip(result, expected):
self.assertEqual(expected_entry[0], entry[0])
self.assertEqual(expected_entry[1], sorted(entry[1]))
self.assertEqual(expected_entry[2], sorted(entry[2]))
class FakePathModuleTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="!")
self.os = fake_os.FakeOsModule(self.filesystem)
self.path = self.os.path
def check_abspath(self, is_windows):
# the implementation differs in Windows and Posix, so test both
self.filesystem.is_windows_fs = is_windows
filename = "foo"
abspath = self.filesystem.root_dir_name + filename
self.filesystem.create_file(abspath)
self.assertEqual(abspath, self.path.abspath(abspath))
self.assertEqual(abspath, self.path.abspath(filename))
self.assertEqual(abspath, self.path.abspath("..!%s" % filename))
def test_abspath_windows(self):
self.check_abspath(is_windows=True)
def test_abspath_posix(self):
"""abspath should return a consistent representation of a file."""
self.check_abspath(is_windows=False)
def check_abspath_bytes(self, is_windows):
"""abspath should return a consistent representation of a file."""
self.filesystem.is_windows_fs = is_windows
filename = b"foo"
abspath = self.filesystem.root_dir_name.encode() + filename
self.filesystem.create_file(abspath)
self.assertEqual(abspath, self.path.abspath(abspath))
self.assertEqual(abspath, self.path.abspath(filename))
self.assertEqual(abspath, self.path.abspath(b"..!" + filename))
def test_abspath_bytes_windows(self):
self.check_abspath_bytes(is_windows=True)
def test_abspath_bytes_posix(self):
self.check_abspath_bytes(is_windows=False)
def test_abspath_deals_with_relative_non_root_path(self):
"""abspath should correctly handle relative paths from a
non-! directory.
This test is distinct from the basic functionality test because
fake_filesystem has historically been based in !.
"""
filename = "!foo!bar!baz"
file_components = filename.split(self.path.sep)
root_name = self.filesystem.root_dir_name
basedir = f"{root_name}{file_components[0]}"
self.filesystem.create_file(filename)
self.os.chdir(basedir)
self.assertEqual(basedir, self.path.abspath(self.path.curdir))
self.assertEqual(root_name, self.path.abspath(".."))
self.assertEqual(
self.path.join(basedir, file_components[1]),
self.path.abspath(file_components[1]),
)
def test_abs_path_with_drive_component(self):
self.filesystem.is_windows_fs = True
self.filesystem.cwd = "C:!foo"
self.assertEqual("C:!foo!bar", self.path.abspath("bar"))
self.assertEqual("C:!foo!bar", self.path.abspath("C:bar"))
self.assertEqual("C:!foo!bar", self.path.abspath("!foo!bar"))
def test_isabs_with_drive_component(self):
self.filesystem.is_windows_fs = False
self.assertFalse(self.path.isabs("C:!foo"))
self.assertFalse(self.path.isabs(b"C:!foo"))
self.assertTrue(self.path.isabs("!"))
self.assertTrue(self.path.isabs(b"!"))
self.filesystem.is_windows_fs = True
self.assertTrue(self.path.isabs("C:!foo"))
self.assertTrue(self.path.isabs(b"C:!foo"))
self.assertTrue(self.path.isabs("!"))
self.assertTrue(self.path.isabs(b"!"))
def test_relpath(self):
path_foo = "!path!to!foo"
path_bar = "!path!to!bar"
path_other = "!some!where!else"
with self.assertRaises(ValueError):
self.path.relpath(None)
with self.assertRaises(ValueError):
self.path.relpath("")
self.assertEqual("path!to!foo", self.path.relpath(path_foo))
self.assertEqual("..!foo", self.path.relpath(path_foo, path_bar))
self.assertEqual(
"..!..!..%s" % path_other, self.path.relpath(path_other, path_bar)
)
self.assertEqual(".", self.path.relpath(path_bar, path_bar))
def test_realpath_vs_abspath(self):
self.filesystem.is_windows_fs = False
self.filesystem.create_file("!george!washington!bridge")
self.filesystem.create_symlink("!first!president", "!george!washington")
self.assertEqual(
"!first!president!bridge",
self.os.path.abspath("!first!president!bridge"),
)
self.assertEqual(
"!george!washington!bridge",
self.os.path.realpath("!first!president!bridge"),
)
self.os.chdir("!first!president")
self.assertEqual("!george!washington!bridge", self.os.path.realpath("bridge"))
@unittest.skipIf(sys.version_info < (3, 10), "'strict' new in Python 3.10")
def test_realpath_strict(self):
self.filesystem.create_file("!foo!bar")
root_dir = self.filesystem.root_dir_name
self.filesystem.cwd = f"{root_dir}foo"
self.assertEqual(
f"{root_dir}foo!baz", self.os.path.realpath("baz", strict=False)
)
with self.raises_os_error(errno.ENOENT):
self.os.path.realpath("baz", strict=True)
self.assertEqual(
f"{root_dir}foo!bar", self.os.path.realpath("bar", strict=True)
)
def test_samefile(self):
file_path1 = "!foo!bar!baz"
file_path2 = "!foo!bar!boo"
self.filesystem.create_file(file_path1)
self.filesystem.create_file(file_path2)
self.assertTrue(self.path.samefile(file_path1, file_path1))
self.assertFalse(self.path.samefile(file_path1, file_path2))
self.assertTrue(self.path.samefile(file_path1, "!foo!..!foo!bar!..!bar!baz"))
self.assertTrue(self.path.samefile(file_path1, b"!foo!..!foo!bar!..!bar!baz"))
def test_exists(self):
file_path = "foo!bar!baz"
file_path_bytes = b"foo!bar!baz"
self.filesystem.create_file(file_path)
self.assertTrue(self.path.exists(file_path))
self.assertTrue(self.path.exists(file_path_bytes))
self.assertFalse(self.path.exists("!some!other!bogus!path"))
def test_exists_with_drive(self):
self.filesystem.os = OSType.WINDOWS
self.filesystem.add_mount_point("F:")
self.assertTrue(self.path.exists("C:"))
self.assertTrue(self.path.exists("c:\\"))
self.assertTrue(self.path.exists("f:"))
self.assertTrue(self.path.exists("F:\\"))
self.assertFalse(self.path.exists("Z:"))
self.assertFalse(self.path.exists("z:\\"))
def test_lexists(self):
file_path = "foo!bar!baz"
file_path_bytes = b"foo!bar!baz"
self.filesystem.create_dir("foo!bar")
self.filesystem.create_symlink(file_path, "bogus")
self.assertTrue(self.path.lexists(file_path))
self.assertTrue(self.path.lexists(file_path_bytes))
self.assertFalse(self.path.exists(file_path))
self.assertFalse(self.path.exists(file_path_bytes))
self.filesystem.create_file("foo!bar!bogus")
self.assertTrue(self.path.exists(file_path))
def test_dirname_with_drive(self):
self.filesystem.is_windows_fs = True
self.assertEqual("c:!foo", self.path.dirname("c:!foo!bar"))
self.assertEqual(b"c:!", self.path.dirname(b"c:!foo"))
self.assertEqual("!foo", self.path.dirname("!foo!bar"))
self.assertEqual(b"!", self.path.dirname(b"!foo"))
self.assertEqual("c:foo", self.path.dirname("c:foo!bar"))
self.assertEqual(b"c:", self.path.dirname(b"c:foo"))
self.assertEqual("foo", self.path.dirname("foo!bar"))
def test_dirname(self):
dirname = "foo!bar"
self.assertEqual(dirname, self.path.dirname("%s!baz" % dirname))
def test_join_strings(self):
components = ["foo", "bar", "baz"]
self.assertEqual("foo!bar!baz", self.path.join(*components))
def test_join_bytes(self):
components = [b"foo", b"bar", b"baz"]
self.assertEqual(b"foo!bar!baz", self.path.join(*components))
def test_expand_user(self):
if self.is_windows:
self.assertEqual(
self.path.expanduser("~"),
self.os.environ["USERPROFILE"].replace("\\", "!"),
)
else:
self.assertEqual(
self.path.expanduser("~"),
self.os.environ["HOME"].replace("/", "!"),
)
@unittest.skipIf(
TestCase.is_windows or TestCase.is_cygwin,
"only tested on unix systems",
)
def test_expand_root(self):
if sys.platform == "darwin":
roothome = "!var!root"
else:
roothome = "!root"
self.assertEqual(self.path.expanduser("~root"), roothome)
def test_getsize_path_nonexistent(self):
file_path = "foo!bar!baz"
with self.assertRaises(os.error):
self.path.getsize(file_path)
def test_getsize_file_empty(self):
file_path = "foo!bar!baz"
self.filesystem.create_file(file_path)
self.assertEqual(0, self.path.getsize(file_path))
def test_getsize_file_non_zero_size(self):
file_path = "foo!bar!baz"
file_path_bytes = b"foo!bar!baz"
self.filesystem.create_file(file_path, contents="1234567")
self.assertEqual(7, self.path.getsize(file_path))
self.assertEqual(7, self.path.getsize(file_path_bytes))
def test_getsize_dir_empty(self):
# For directories, only require that the size is non-negative.
dir_path = "foo!bar"
self.filesystem.create_dir(dir_path)
size = self.path.getsize(dir_path)
self.assertFalse(int(size) < 0, "expected non-negative size; actual: %s" % size)
def test_getsize_dir_non_zero_size(self):
# For directories, only require that the size is non-negative.
dir_path = "foo!bar"
self.filesystem.create_file(self.filesystem.joinpaths(dir_path, "baz"))
size = self.path.getsize(dir_path)
self.assertFalse(int(size) < 0, "expected non-negative size; actual: %s" % size)
def test_isdir(self):
self.filesystem.create_file("foo!bar")
self.assertTrue(self.path.isdir("foo"))
self.assertTrue(self.path.isdir(b"foo"))
self.assertFalse(self.path.isdir("foo!bar"))
self.assertFalse(self.path.isdir("it_dont_exist"))
def test_isdir_with_cwd_change(self):
self.filesystem.create_file("!foo!bar!baz")
self.assertTrue(self.path.isdir("!foo"))
self.assertTrue(self.path.isdir("!foo!bar"))
self.assertTrue(self.path.isdir("foo"))
self.assertTrue(self.path.isdir("foo!bar"))
self.filesystem.cwd = f"{self.filesystem.root_dir_name}foo"
self.assertTrue(self.path.isdir("!foo"))
self.assertTrue(self.path.isdir("!foo!bar"))
self.assertTrue(self.path.isdir("bar"))
def test_isfile(self):
self.filesystem.create_file("foo!bar")
self.assertFalse(self.path.isfile("foo"))
self.assertTrue(self.path.isfile("foo!bar"))
self.assertTrue(self.path.isfile(b"foo!bar"))
self.assertFalse(self.path.isfile("it_dont_exist"))
def test_get_mtime(self):
test_file = self.filesystem.create_file("foo!bar1.txt")
self.assertNotEqual(24, self.path.getmtime("foo!bar1.txt"))
test_file.st_mtime = 24
self.assertEqual(24, self.path.getmtime("foo!bar1.txt"))
self.assertEqual(24, self.path.getmtime(b"foo!bar1.txt"))
def test_get_mtime_raises_os_error(self):
self.assertFalse(self.path.exists("does_not_exist"))
with self.raises_os_error(errno.ENOENT):
self.path.getmtime("does_not_exist")
def test_islink(self):
self.filesystem.create_dir("foo")
self.filesystem.create_file("foo!regular_file")
self.filesystem.create_symlink("foo!link_to_file", "regular_file")
self.assertFalse(self.path.islink("foo"))
# An object can be both a link and a file or file, according to the
# comments in Python/Lib/posixpath.py.
self.assertTrue(self.path.islink("foo!link_to_file"))
self.assertTrue(self.path.isfile("foo!link_to_file"))
self.assertTrue(self.path.islink(b"foo!link_to_file"))
self.assertTrue(self.path.isfile(b"foo!link_to_file"))
self.assertTrue(self.path.isfile("foo!regular_file"))
self.assertFalse(self.path.islink("foo!regular_file"))
self.assertFalse(self.path.islink("it_dont_exist"))
def test_is_link_case_sensitive(self):
# Regression test for #306
self.filesystem.is_case_sensitive = False
self.filesystem.create_dir("foo")
self.filesystem.create_symlink("foo!bar", "foo")
self.assertTrue(self.path.islink("foo!Bar"))
def test_ismount(self):
self.assertFalse(self.path.ismount(""))
self.assertTrue(self.path.ismount("!"))
self.assertTrue(self.path.ismount(b"!"))
self.assertFalse(self.path.ismount("!mount!"))
self.filesystem.add_mount_point("!mount")
self.assertTrue(self.path.ismount("!mount"))
self.assertTrue(self.path.ismount(b"!mount"))
self.assertTrue(self.path.ismount("!mount!"))
def test_ismount_with_drive_letters(self):
self.filesystem.is_windows_fs = True
self.assertTrue(self.path.ismount("!"))
self.assertTrue(self.path.ismount("c:!"))
self.assertFalse(self.path.ismount("c:"))
self.assertTrue(self.path.ismount("z:!"))
self.filesystem.add_mount_point("!mount")
self.assertTrue(self.path.ismount("!mount"))
self.assertTrue(self.path.ismount("!mount!"))
def test_ismount_with_unc_paths(self):
self.filesystem.is_windows_fs = True
self.assertTrue(self.path.ismount("!!a!"))
self.assertTrue(self.path.ismount("!!a!b"))
self.assertTrue(self.path.ismount("!!a!b!"))
self.assertFalse(self.path.ismount("!a!b!"))
self.assertFalse(self.path.ismount("!!a!b!c"))
def test_ismount_with_alternate_path_separator(self):
self.filesystem.alternative_path_separator = "!"
self.filesystem.add_mount_point("!mount")
self.assertTrue(self.path.ismount("!mount"))
self.assertTrue(self.path.ismount("!mount!"))
self.assertTrue(self.path.ismount("!mount!!"))
self.filesystem.is_windows_fs = True
self.assertTrue(self.path.ismount("Z:!"))
def test_getattr_forward_to_real_os_path(self):
"""Forwards any non-faked calls to os.path."""
self.assertTrue(hasattr(self.path, "sep"), "Get a faked os.path function")
private_path_function = None
if sys.version_info < (3, 6):
if self.is_windows:
private_path_function = "_get_bothseps"
else:
private_path_function = "_join_real_path"
if private_path_function:
self.assertTrue(
hasattr(self.path, private_path_function),
"Get a real os.path function " "not implemented in fake os.path",
)
self.assertFalse(hasattr(self.path, "nonexistent"))
def test_splitroot_posix(self):
self.filesystem.is_windows_fs = False
self.assertEqual(("", "", "foo!bar"), self.filesystem.splitroot("foo!bar"))
self.assertEqual(("", "!", "foo!bar"), self.filesystem.splitroot("!foo!bar"))
self.assertEqual(
("", "!!", "foo!!bar"), self.filesystem.splitroot("!!foo!!bar")
)
class PathManipulationTestBase(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="|")
class CollapsePathPipeSeparatorTest(PathManipulationTestBase):
"""Tests CollapsePath (mimics os.path.normpath) using |
as path separator."""
def test_empty_path_becomes_dot_path(self):
self.assertEqual(".", self.filesystem.normpath(""))
def test_dot_path_unchanged(self):
self.assertEqual(".", self.filesystem.normpath("."))
def test_slashes_are_not_collapsed(self):
"""Tests that '/' is not treated specially if the
path separator is '|'.
In particular, multiple slashes should not be collapsed.
"""
self.assertEqual("/", self.filesystem.normpath("/"))
self.assertEqual("/////", self.filesystem.normpath("/////"))
def test_root_path(self):
self.assertEqual("|", self.filesystem.normpath("|"))
def test_multiple_separators_collapsed_into_root_path(self):
self.assertEqual("|", self.filesystem.normpath("|||||"))
def test_all_dot_paths_removed_but_one(self):
self.assertEqual(".", self.filesystem.normpath(".|.|.|."))
def test_all_dot_paths_removed_if_another_path_component_exists(self):
self.assertEqual("|", self.filesystem.normpath("|.|.|.|"))
self.assertEqual("foo|bar", self.filesystem.normpath("foo|.|.|.|bar"))
def test_ignores_up_level_references_starting_from_root(self):
self.assertEqual("|", self.filesystem.normpath("|..|..|..|"))
self.assertEqual("|", self.filesystem.normpath("|..|..|foo|bar|..|..|"))
self.filesystem.is_windows_fs = False # not an UNC path
self.assertEqual("|", self.filesystem.normpath("||..|.|..||"))
def test_conserves_up_level_references_starting_from_current_dir(self):
self.assertEqual("..|..", self.filesystem.normpath("..|foo|bar|..|..|.."))
def test_combine_dot_and_up_level_references_in_absolute_path(self):
self.assertEqual("|yes", self.filesystem.normpath("|||||.|..|||yes|no|..|.|||"))
def test_dots_in_path_collapses_to_last_path(self):
self.assertEqual("bar", self.filesystem.normpath("foo|..|bar"))
self.assertEqual("bar", self.filesystem.normpath("foo|..|yes|..|no|..|bar"))
class SplitPathTest(PathManipulationTestBase):
"""Tests SplitPath (which mimics os.path.split)
using | as path separator."""
def test_empty_path(self):
self.assertEqual(("", ""), self.filesystem.splitpath(""))
def test_no_separators(self):
self.assertEqual(("", "ab"), self.filesystem.splitpath("ab"))
def test_slashes_do_not_split(self):
"""Tests that '/' is not treated specially if the
path separator is '|'."""
self.assertEqual(("", "a/b"), self.filesystem.splitpath("a/b"))
def test_eliminate_trailing_separators_from_head(self):
self.assertEqual(("a", "b"), self.filesystem.splitpath("a|b"))
self.assertEqual(("a", "b"), self.filesystem.splitpath("a|||b"))
self.assertEqual(("|a", "b"), self.filesystem.splitpath("|a||b"))
self.assertEqual(("a|b", "c"), self.filesystem.splitpath("a|b|c"))
self.assertEqual(("|a|b", "c"), self.filesystem.splitpath("|a|b|c"))
def test_root_separator_is_not_stripped(self):
self.assertEqual(("|||", ""), self.filesystem.splitpath("|||"))
self.assertEqual(("|", "a"), self.filesystem.splitpath("|a"))
self.assertEqual(("|||", "a"), self.filesystem.splitpath("|||a"))
def test_empty_tail_if_path_ends_in_separator(self):
self.assertEqual(("a|b", ""), self.filesystem.splitpath("a|b|"))
def test_empty_path_components_are_preserved_in_head(self):
self.assertEqual(("|a||b", "c"), self.filesystem.splitpath("|a||b||c"))
class JoinPathTest(PathManipulationTestBase):
"""Tests JoinPath (which mimics os.path.join) using | as path separator."""
def test_one_empty_component(self):
self.assertEqual("", self.filesystem.joinpaths(""))
def test_multiple_empty_components(self):
self.assertEqual("", self.filesystem.joinpaths("", "", ""))
def test_separators_not_stripped_from_single_component(self):
self.assertEqual("||a||", self.filesystem.joinpaths("||a||"))
def test_one_separator_added_between_components(self):
self.assertEqual("a|b|c|d", self.filesystem.joinpaths("a", "b", "c", "d"))
def test_no_separator_added_for_components_ending_in_separator(self):
self.assertEqual("a|b|c", self.filesystem.joinpaths("a|", "b|", "c"))
self.assertEqual("a|||b|||c", self.filesystem.joinpaths("a|||", "b|||", "c"))
def test_components_preceding_absolute_component_are_ignored(self):
self.assertEqual("|c|d", self.filesystem.joinpaths("a", "|b", "|c", "d"))
def test_one_separator_added_for_trailing_empty_components(self):
self.assertEqual("a|", self.filesystem.joinpaths("a", ""))
self.assertEqual("a|", self.filesystem.joinpaths("a", "", ""))
def test_no_separator_added_for_leading_empty_components(self):
self.assertEqual("a", self.filesystem.joinpaths("", "a"))
def test_internal_empty_components_ignored(self):
self.assertEqual("a|b", self.filesystem.joinpaths("a", "", "b"))
self.assertEqual("a|b|", self.filesystem.joinpaths("a|", "", "b|"))
class PathSeparatorTest(TestCase):
def test_os_path_sep_matches_fake_filesystem_separator(self):
filesystem = fake_filesystem.FakeFilesystem(path_separator="!")
fake_os_module = fake_os.FakeOsModule(filesystem)
self.assertEqual("!", fake_os_module.sep)
self.assertEqual("!", fake_os_module.path.sep)
class NormalizeCaseTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.filesystem.is_case_sensitive = False
def test_normalize_case(self):
self.filesystem.create_file("/Foo/Bar")
self.assertEqual(
f"{self.filesystem.root_dir_name}Foo/Bar",
self.filesystem._original_path("/foo/bar"),
)
self.assertEqual(
f"{self.filesystem.root_dir_name}Foo/Bar",
self.filesystem._original_path("/FOO/BAR"),
)
def test_normalize_case_for_drive(self):
self.filesystem.is_windows_fs = True
self.filesystem.create_file("C:/Foo/Bar")
self.assertEqual("C:/Foo/Bar", self.filesystem._original_path("c:/foo/bar"))
self.assertEqual("C:/Foo/Bar", self.filesystem._original_path("C:/FOO/BAR"))
def test_normalize_case_for_non_existing_file(self):
self.filesystem.create_dir("/Foo/Bar")
self.assertEqual(
f"{self.filesystem.root_dir_name}Foo/Bar/baz",
self.filesystem._original_path("/foo/bar/baz"),
)
self.assertEqual(
f"{self.filesystem.root_dir_name}Foo/Bar/BAZ",
self.filesystem._original_path("/FOO/BAR/BAZ"),
)
@unittest.skipIf(
not TestCase.is_windows, "Regression test for Windows problem only"
)
def test_normalize_case_for_lazily_added_empty_file(self):
# regression test for specific issue with added empty real files
filesystem = fake_filesystem.FakeFilesystem()
real_dir_path = os.path.split(os.path.dirname(os.path.abspath(__file__)))[0]
filesystem.add_real_directory(real_dir_path)
initPyPath = os.path.join(real_dir_path, "__init__.py")
self.assertEqual(initPyPath, filesystem._original_path(initPyPath.upper()))
class AlternativePathSeparatorTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="!")
self.filesystem.alternative_path_separator = "?"
def test_initial_value(self):
filesystem = fake_filesystem.FakeFilesystem()
if self.is_windows:
self.assertEqual("/", filesystem.alternative_path_separator)
else:
self.assertIsNone(filesystem.alternative_path_separator)
filesystem = fake_filesystem.FakeFilesystem(path_separator="/")
self.assertIsNone(filesystem.alternative_path_separator)
def test_alt_sep(self):
fake_os_module = fake_os.FakeOsModule(self.filesystem)
self.assertEqual("?", fake_os_module.altsep)
self.assertEqual("?", fake_os_module.path.altsep)
def test_collapse_path_with_mixed_separators(self):
self.assertEqual("!foo!bar", self.filesystem.normpath("!foo??bar"))
def test_normalize_path_with_mixed_separators(self):
path = "foo?..?bar"
self.assertEqual(
f"{self.filesystem.root_dir_name}bar",
self.filesystem.absnormpath(path),
)
def test_exists_with_mixed_separators(self):
self.filesystem.create_file("?foo?bar?baz")
self.filesystem.create_file("!foo!bar!xyzzy!plugh")
self.assertTrue(self.filesystem.exists("!foo!bar!baz"))
self.assertTrue(self.filesystem.exists("?foo?bar?xyzzy?plugh"))
class DriveLetterSupportTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(path_separator="!")
self.filesystem.alternative_path_separator = "^"
self.filesystem.is_windows_fs = True
def test_initial_value(self):
filesystem = fake_filesystem.FakeFilesystem()
if self.is_windows:
self.assertTrue(filesystem.is_windows_fs)
else:
self.assertFalse(filesystem.is_windows_fs)
def test_collapse_path(self):
self.assertEqual("c:!foo!bar", self.filesystem.normpath("c:!!foo!!bar"))
def test_collapse_unc_path(self):
self.assertEqual("!!foo!bar!baz", self.filesystem.normpath("!!foo!bar!!baz!!"))
def test_normalize_path_str(self):
self.filesystem.cwd = ""
self.assertEqual("c:!foo!bar", self.filesystem.absnormpath("c:!foo!!bar"))
self.filesystem.cwd = "c:!foo"
self.assertEqual("c:!foo!bar", self.filesystem.absnormpath("bar"))
def test_normalize_path_bytes(self):
self.filesystem.cwd = b""
self.assertEqual(b"c:!foo!bar", self.filesystem.absnormpath(b"c:!foo!!bar"))
self.filesystem.cwd = b"c:!foo"
self.assertEqual(b"c:!foo!bar", self.filesystem.absnormpath(b"bar"))
def test_split_path_str(self):
self.assertEqual(("c:!foo", "bar"), self.filesystem.splitpath("c:!foo!bar"))
self.assertEqual(("c:!", "foo"), self.filesystem.splitpath("c:!foo"))
self.assertEqual(("!foo", "bar"), self.filesystem.splitpath("!foo!bar"))
self.assertEqual(("!", "foo"), self.filesystem.splitpath("!foo"))
self.assertEqual(("c:foo", "bar"), self.filesystem.splitpath("c:foo!bar"))
self.assertEqual(("c:", "foo"), self.filesystem.splitpath("c:foo"))
self.assertEqual(("foo", "bar"), self.filesystem.splitpath("foo!bar"))
def test_split_with_alt_separator(self):
self.assertEqual(("a^b", "c"), self.filesystem.splitpath("a^b^c"))
self.assertEqual(("a^b!c", "d"), self.filesystem.splitpath("a^b!c^d"))
self.assertEqual(("a^b!c", "d"), self.filesystem.splitpath("a^b!c!d"))
self.assertEqual((b"a^b", b"c"), self.filesystem.splitpath(b"a^b^c"))
self.assertEqual((b"a^b!c", b"d"), self.filesystem.splitpath(b"a^b!c^d"))
self.assertEqual((b"a^b!c", b"d"), self.filesystem.splitpath(b"a^b!c!d"))
def test_split_path_bytes(self):
self.assertEqual((b"c:!foo", b"bar"), self.filesystem.splitpath(b"c:!foo!bar"))
self.assertEqual((b"c:!", b"foo"), self.filesystem.splitpath(b"c:!foo"))
self.assertEqual((b"!foo", b"bar"), self.filesystem.splitpath(b"!foo!bar"))
self.assertEqual((b"!", b"foo"), self.filesystem.splitpath(b"!foo"))
self.assertEqual((b"c:foo", b"bar"), self.filesystem.splitpath(b"c:foo!bar"))
self.assertEqual((b"c:", b"foo"), self.filesystem.splitpath(b"c:foo"))
self.assertEqual((b"foo", b"bar"), self.filesystem.splitpath(b"foo!bar"))
def test_characters_before_root_ignored_in_join_paths(self):
self.assertEqual("c:d", self.filesystem.joinpaths("b", "c:", "d"))
def test_resolve_path(self):
self.assertEqual("C:!foo!bar", self.filesystem.resolve_path("C:!foo!bar"))
def test_get_path_components(self):
self.assertEqual(
["c:", "foo", "bar"],
self.filesystem._path_components("c:!foo!bar"),
)
self.assertEqual(["c:"], self.filesystem._path_components("c:"))
def test_split_drive_str(self):
self.assertEqual(("c:", "!foo!bar"), self.filesystem.splitdrive("c:!foo!bar"))
self.assertEqual(("", "!foo!bar"), self.filesystem.splitdrive("!foo!bar"))
self.assertEqual(("c:", "foo!bar"), self.filesystem.splitdrive("c:foo!bar"))
self.assertEqual(("", "foo!bar"), self.filesystem.splitdrive("foo!bar"))
def test_split_drive_bytes(self):
self.assertEqual(
(b"c:", b"!foo!bar"), self.filesystem.splitdrive(b"c:!foo!bar")
)
self.assertEqual((b"", b"!foo!bar"), self.filesystem.splitdrive(b"!foo!bar"))
def test_split_drive_alt_sep(self):
self.assertEqual(("c:", "^foo^bar"), self.filesystem.splitdrive("c:^foo^bar"))
self.assertEqual(("", "foo^bar"), self.filesystem.splitdrive("foo^bar"))
self.assertEqual(("", "foo^bar!baz"), self.filesystem.splitdrive("foo^bar!baz"))
self.assertEqual(
(b"c:", b"^foo^bar"), self.filesystem.splitdrive(b"c:^foo^bar")
)
self.assertEqual((b"", b"^foo^bar"), self.filesystem.splitdrive(b"^foo^bar"))
self.assertEqual(
(b"", b"^foo^bar!baz"), self.filesystem.splitdrive(b"^foo^bar!baz")
)
def test_split_drive_with_unc_path(self):
self.assertEqual(
("!!foo!bar", "!baz"), self.filesystem.splitdrive("!!foo!bar!baz")
)
self.assertEqual(("", "!!foo"), self.filesystem.splitdrive("!!foo"))
self.assertEqual(("", "!!foo!!bar"), self.filesystem.splitdrive("!!foo!!bar"))
self.assertEqual(("!!foo!bar", "!!"), self.filesystem.splitdrive("!!foo!bar!!"))
def test_split_drive_with_unc_path_alt_sep(self):
self.assertEqual(
("^^foo^bar", "!baz"), self.filesystem.splitdrive("^^foo^bar!baz")
)
self.assertEqual(("", "^^foo"), self.filesystem.splitdrive("^^foo"))
self.assertEqual(("", "^^foo^^bar"), self.filesystem.splitdrive("^^foo^^bar"))
self.assertEqual(("^^foo^bar", "^^"), self.filesystem.splitdrive("^^foo^bar^^"))
def test_split_path_with_drive(self):
self.assertEqual(("d:!foo", "baz"), self.filesystem.splitpath("d:!foo!baz"))
self.assertEqual(("d:!foo!baz", ""), self.filesystem.splitpath("d:!foo!baz!"))
self.assertEqual(("c:", ""), self.filesystem.splitpath("c:"))
self.assertEqual(("c:!", ""), self.filesystem.splitpath("c:!"))
self.assertEqual(("c:!!", ""), self.filesystem.splitpath("c:!!"))
def test_split_path_with_drive_alt_sep(self):
self.assertEqual(("d:^foo", "baz"), self.filesystem.splitpath("d:^foo^baz"))
self.assertEqual(("d:^foo^baz", ""), self.filesystem.splitpath("d:^foo^baz^"))
self.assertEqual(("c:", ""), self.filesystem.splitpath("c:"))
self.assertEqual(("c:^", ""), self.filesystem.splitpath("c:^"))
self.assertEqual(("c:^^", ""), self.filesystem.splitpath("c:^^"))
def test_split_path_with_unc_path(self):
self.assertEqual(
("!!foo!bar!", "baz"), self.filesystem.splitpath("!!foo!bar!baz")
)
self.assertEqual(("!!foo!bar", ""), self.filesystem.splitpath("!!foo!bar"))
self.assertEqual(("!!foo!bar!!", ""), self.filesystem.splitpath("!!foo!bar!!"))
def test_split_path_with_unc_path_alt_sep(self):
self.assertEqual(
("^^foo^bar^", "baz"), self.filesystem.splitpath("^^foo^bar^baz")
)
self.assertEqual(("^^foo^bar", ""), self.filesystem.splitpath("^^foo^bar"))
self.assertEqual(("^^foo^bar^^", ""), self.filesystem.splitpath("^^foo^bar^^"))
def test_splitroot_with_drive(self):
self.assertEqual(
("E:", "!", "foo!bar"), self.filesystem.splitroot("E:!foo!bar")
)
self.assertEqual(
("E:", "!", "!foo!!!bar"), self.filesystem.splitroot("E:!!foo!!!bar")
)
self.assertEqual(
(b"E:", b"!", b"!foo!!!bar"), self.filesystem.splitroot(b"E:!!foo!!!bar")
)
self.assertEqual(
("C:", "^", "foo^bar"), self.filesystem.splitroot("C:^foo^bar")
)
def test_splitroot_with_unc_path(self):
self.assertEqual(
("!!foo!bar", "!", "baz"), self.filesystem.splitroot("!!foo!bar!baz")
)
self.assertEqual(
("!!?!UNC", "!", "foo!bar"), self.filesystem.splitroot("!!?!UNC!foo!bar")
)
self.assertEqual(
("^^foo^bar", "^", "baz"), self.filesystem.splitroot("^^foo^bar^baz")
)
self.assertEqual(
(b"!!foo!bar", b"!", b"baz"), self.filesystem.splitroot(b"!!foo!bar!baz")
)
def test_splitroot_with_empty_parts(self):
self.assertEqual(("", "", ""), self.filesystem.splitroot(""))
self.assertEqual(("", "!", "foo"), self.filesystem.splitroot("!foo"))
self.assertEqual(("!!foo!bar", "", ""), self.filesystem.splitroot("!!foo!bar"))
self.assertEqual(("!!foo", "", ""), self.filesystem.splitroot("!!foo"))
self.assertEqual(
("!!foo!bar", "!", ""), self.filesystem.splitroot("!!foo!bar!")
)
self.assertEqual(("C:", "", "foo!bar"), self.filesystem.splitroot("C:foo!bar"))
class DiskSpaceTest(TestCase):
def setUp(self):
self.fs = fake_filesystem.FakeFilesystem(path_separator="!", total_size=100)
self.os = fake_os.FakeOsModule(self.fs)
self.open = fake_open.FakeFileOpen(self.fs)
def test_disk_usage_on_file_creation(self):
total_size = 100
self.fs.add_mount_point("!mount", total_size)
def create_too_large_file():
with self.open("!mount!file", "w") as dest:
dest.write("a" * (total_size + 1))
with self.assertRaises(OSError):
create_too_large_file()
self.assertEqual(0, self.fs.get_disk_usage("!mount").used)
with self.open("!mount!file", "w") as dest:
dest.write("a" * total_size)
self.assertEqual(total_size, self.fs.get_disk_usage("!mount").used)
def test_disk_usage_on_automounted_drive(self):
self.fs.is_windows_fs = True
self.fs.reset(total_size=100)
self.fs.create_file("!foo!bar", st_size=50)
self.assertEqual(0, self.fs.get_disk_usage("D:!").used)
self.fs.cwd = "E:!foo"
self.assertEqual(0, self.fs.get_disk_usage("!foo").used)
def test_disk_usage_on_mounted_paths(self):
self.fs.add_mount_point("!foo", total_size=200)
self.fs.add_mount_point("!foo!bar", total_size=400)
self.fs.create_file("!baz", st_size=50)
self.fs.create_file("!foo!baz", st_size=60)
self.fs.create_file("!foo!bar!baz", st_size=100)
self.assertEqual(50, self.fs.get_disk_usage("!").used)
self.assertEqual(60, self.fs.get_disk_usage("!foo").used)
self.assertEqual(100, self.fs.get_disk_usage("!foo!bar").used)
self.assertEqual(400, self.fs.get_disk_usage("!foo!bar").total)
def test_file_system_size_after_large_file_creation(self):
filesystem = fake_filesystem.FakeFilesystem(
path_separator="!", total_size=1024 * 1024 * 1024 * 100
)
filesystem.create_file("!foo!baz", st_size=1024 * 1024 * 1024 * 10)
self.assertEqual(
(
1024 * 1024 * 1024 * 100,
1024 * 1024 * 1024 * 10,
1024 * 1024 * 1024 * 90,
),
filesystem.get_disk_usage(),
)
def test_file_system_size_after_binary_file_creation(self):
self.fs.create_file("!foo!bar", contents=b"xyzzy")
self.assertEqual((100, 5, 95), self.fs.get_disk_usage())
def test_file_system_size_after_ascii_string_file_creation(self):
self.fs.create_file("!foo!bar", contents="complicated")
self.assertEqual((100, 11, 89), self.fs.get_disk_usage())
def test_filesystem_size_after_2byte_unicode_file_creation(self):
self.fs.create_file("!foo!bar", contents="сложно", encoding="utf-8")
self.assertEqual((100, 12, 88), self.fs.get_disk_usage())
def test_filesystem_size_after_3byte_unicode_file_creation(self):
self.fs.create_file("!foo!bar", contents="複雑", encoding="utf-8")
self.assertEqual((100, 6, 94), self.fs.get_disk_usage())
def test_file_system_size_after_file_deletion(self):
self.fs.create_file("!foo!bar", contents=b"xyzzy")
self.fs.create_file("!foo!baz", st_size=20)
self.fs.remove_object("!foo!bar")
self.assertEqual((100, 20, 80), self.fs.get_disk_usage())
def test_file_system_size_after_directory_removal(self):
self.fs.create_file("!foo!bar", st_size=10)
self.fs.create_file("!foo!baz", st_size=20)
self.fs.create_file("!foo1!bar", st_size=40)
self.fs.remove_object("!foo")
self.assertEqual((100, 40, 60), self.fs.get_disk_usage())
def test_creating_file_with_fitting_content(self):
initial_usage = self.fs.get_disk_usage()
try:
self.fs.create_file("!foo!bar", contents=b"a" * 100)
except OSError:
self.fail(
"File with contents fitting into disk space " "could not be written."
)
self.assertEqual(initial_usage.used + 100, self.fs.get_disk_usage().used)
def test_creating_file_with_content_too_large(self):
def create_large_file():
self.fs.create_file("!foo!bar", contents=b"a" * 101)
initial_usage = self.fs.get_disk_usage()
with self.assertRaises(OSError):
create_large_file()
self.assertEqual(initial_usage, self.fs.get_disk_usage())
def test_creating_file_with_fitting_size(self):
initial_usage = self.fs.get_disk_usage()
try:
self.fs.create_file("!foo!bar", st_size=100)
except OSError:
self.fail("File with size fitting into disk space could not be written.")
self.assertEqual(initial_usage.used + 100, self.fs.get_disk_usage().used)
def test_creating_file_with_size_too_large(self):
initial_usage = self.fs.get_disk_usage()
def create_large_file():
self.fs.create_file("!foo!bar", st_size=101)
with self.assertRaises(OSError):
create_large_file()
self.assertEqual(initial_usage, self.fs.get_disk_usage())
def test_resize_file_with_fitting_size(self):
file_object = self.fs.create_file("!foo!bar", st_size=50)
try:
file_object.set_large_file_size(100)
file_object.set_contents(b"a" * 100)
except OSError:
self.fail("Resizing file failed although disk space was sufficient.")
def test_resize_file_with_size_too_large(self):
file_object = self.fs.create_file("!foo!bar", st_size=50)
with self.raises_os_error(errno.ENOSPC):
file_object.set_large_file_size(200)
with self.raises_os_error(errno.ENOSPC):
file_object.set_contents("a" * 150)
def test_file_system_size_after_directory_rename(self):
self.fs.create_file("!foo!bar", st_size=20)
self.os.rename("!foo", "!baz")
self.assertEqual(20, self.fs.get_disk_usage().used)
def test_file_system_size_after_file_rename(self):
self.fs.create_file("!foo!bar", st_size=20)
self.os.rename("!foo!bar", "!foo!baz")
self.assertEqual(20, self.fs.get_disk_usage().used)
def test_that_hard_link_does_not_change_used_size(self):
file1_path = "test_file1"
file2_path = "test_file2"
self.fs.create_file(file1_path, st_size=20)
self.assertEqual(20, self.fs.get_disk_usage().used)
# creating a hard link shall not increase used space
self.os.link(file1_path, file2_path)
self.assertEqual(20, self.fs.get_disk_usage().used)
# removing a file shall not decrease used space
# if a hard link still exists
self.os.unlink(file1_path)
self.assertEqual(20, self.fs.get_disk_usage().used)
self.os.unlink(file2_path)
self.assertEqual(0, self.fs.get_disk_usage().used)
def test_that_the_size_of_correct_mount_point_is_used(self):
self.fs.add_mount_point("!mount_limited", total_size=50)
self.fs.add_mount_point("!mount_unlimited")
with self.raises_os_error(errno.ENOSPC):
self.fs.create_file("!mount_limited!foo", st_size=60)
with self.raises_os_error(errno.ENOSPC):
self.fs.create_file("!bar", st_size=110)
try:
self.fs.create_file("!foo", st_size=60)
self.fs.create_file("!mount_limited!foo", st_size=40)
self.fs.create_file("!mount_unlimited!foo", st_size=1000000)
except OSError:
self.fail(
"File with contents fitting into " "disk space could not be written."
)
def test_that_disk_usage_of_correct_mount_point_is_used(self):
self.fs.add_mount_point("!mount1", total_size=20)
self.fs.add_mount_point("!mount1!bar!mount2", total_size=50)
self.fs.create_file("!foo!bar", st_size=10)
self.fs.create_file("!mount1!foo!bar", st_size=10)
self.fs.create_file("!mount1!bar!mount2!foo!bar", st_size=10)
self.assertEqual(90, self.fs.get_disk_usage("!foo").free)
self.assertEqual(10, self.fs.get_disk_usage("!mount1!foo").free)
self.assertEqual(40, self.fs.get_disk_usage("!mount1!bar!mount2").free)
def test_set_larger_disk_size(self):
self.fs.add_mount_point("!mount1", total_size=20)
with self.raises_os_error(errno.ENOSPC):
self.fs.create_file("!mount1!foo", st_size=100)
self.fs.set_disk_usage(total_size=200, path="!mount1")
self.fs.create_file("!mount1!foo", st_size=100)
self.assertEqual(100, self.fs.get_disk_usage("!mount1!foo").free)
def test_set_smaller_disk_size(self):
self.fs.add_mount_point("!mount1", total_size=200)
self.fs.create_file("!mount1!foo", st_size=100)
with self.raises_os_error(errno.ENOSPC):
self.fs.set_disk_usage(total_size=50, path="!mount1")
self.fs.set_disk_usage(total_size=150, path="!mount1")
self.assertEqual(50, self.fs.get_disk_usage("!mount1!foo").free)
def test_disk_size_on_unlimited_disk(self):
self.fs.add_mount_point("!mount1")
self.fs.create_file("!mount1!foo", st_size=100)
self.fs.set_disk_usage(total_size=1000, path="!mount1")
self.assertEqual(900, self.fs.get_disk_usage("!mount1!foo").free)
def test_disk_size_on_auto_mounted_drive_on_file_creation(self):
self.fs.is_windows_fs = True
# drive d: shall be auto-mounted and the used size adapted
self.fs.create_file("d:!foo!bar", st_size=100)
self.fs.set_disk_usage(total_size=1000, path="d:")
self.assertEqual(self.fs.get_disk_usage("d:!foo").free, 900)
def test_disk_size_on_auto_mounted_drive_on_directory_creation(self):
self.fs.is_windows_fs = True
self.fs.create_dir("d:!foo!bar")
self.fs.create_file("d:!foo!bar!baz", st_size=100)
self.fs.create_file("d:!foo!baz", st_size=100)
self.fs.set_disk_usage(total_size=1000, path="d:")
self.assertEqual(800, self.fs.get_disk_usage("d:!foo").free)
def test_copying_preserves_byte_contents(self):
source_file = self.fs.create_file("foo", contents=b"somebytes")
dest_file = self.fs.create_file("bar")
dest_file.set_contents(source_file.contents)
self.assertEqual(dest_file.contents, source_file.contents)
def test_diskusage_after_open_write(self):
with self.open("bar.txt", "w") as f:
f.write("a" * 60)
f.flush()
self.assertEqual(60, self.fs.get_disk_usage()[1])
def test_disk_full_after_reopened(self):
with self.open("bar.txt", "w") as f:
f.write("a" * 60)
with self.open("bar.txt") as f:
self.assertEqual("a" * 60, f.read())
with self.raises_os_error(errno.ENOSPC):
with self.open("bar.txt", "w") as f:
f.write("b" * 110)
with self.raises_os_error(errno.ENOSPC):
f.flush()
with self.open("bar.txt") as f:
self.assertEqual("", f.read())
def test_disk_full_append(self):
file_path = "bar.txt"
with self.open(file_path, "w") as f:
f.write("a" * 60)
with self.open(file_path) as f:
self.assertEqual("a" * 60, f.read())
with self.raises_os_error(errno.ENOSPC):
with self.open(file_path, "a") as f:
f.write("b" * 41)
with self.raises_os_error(errno.ENOSPC):
f.flush()
with self.open("bar.txt") as f:
self.assertEqual(f.read(), "a" * 60)
def test_disk_full_after_reopened_rplus_seek(self):
with self.open("bar.txt", "w") as f:
f.write("a" * 60)
with self.open("bar.txt") as f:
self.assertEqual(f.read(), "a" * 60)
with self.raises_os_error(errno.ENOSPC):
with self.open("bar.txt", "r+") as f:
f.seek(50)
f.write("b" * 60)
with self.raises_os_error(errno.ENOSPC):
f.flush()
with self.open("bar.txt") as f:
self.assertEqual(f.read(), "a" * 60)
class MountPointTest(TestCase):
def setUp(self):
self.filesystem = fake_filesystem.FakeFilesystem(
path_separator="!", total_size=100
)
def add_mount_points(self):
self.filesystem.add_mount_point("!foo")
self.filesystem.add_mount_point("!bar")
self.filesystem.add_mount_point("!foo!baz")
def test_that_new_mount_points_get_new_device_number(self):
self.add_mount_points()
self.assertEqual(1, self.filesystem.get_object("!").st_dev)
self.assertEqual(2, self.filesystem.get_object("!foo").st_dev)
self.assertEqual(3, self.filesystem.get_object("!bar").st_dev)
self.assertEqual(4, self.filesystem.get_object("!foo!baz").st_dev)
def test_that_new_directories_get_correct_device_number(self):
self.add_mount_points()
self.assertEqual(1, self.filesystem.create_dir("!foo1!bar").st_dev)
self.assertEqual(2, self.filesystem.create_dir("!foo!bar").st_dev)
self.assertEqual(4, self.filesystem.create_dir("!foo!baz!foo!bar").st_dev)
def test_that_new_files_get_correct_device_number(self):
self.add_mount_points()
self.assertEqual(1, self.filesystem.create_file("!foo1!bar").st_dev)
self.assertEqual(2, self.filesystem.create_file("!foo!bar").st_dev)
self.assertEqual(4, self.filesystem.create_file("!foo!baz!foo!bar").st_dev)
def test_that_mount_point_cannot_be_added_twice(self):
self.add_mount_points()
with self.raises_os_error(errno.EEXIST):
self.filesystem.add_mount_point("!foo")
with self.raises_os_error(errno.EEXIST):
self.filesystem.add_mount_point("!foo!")
def test_that_drives_are_auto_mounted(self):
self.filesystem.is_windows_fs = True
self.add_mount_points()
self.filesystem.create_dir("d:!foo!bar")
self.filesystem.create_file("d:!foo!baz")
self.filesystem.create_file("z:!foo!baz")
self.assertEqual(5, self.filesystem.get_object("d:").st_dev)
self.assertEqual(5, self.filesystem.get_object("d:!foo!bar").st_dev)
self.assertEqual(5, self.filesystem.get_object("d:!foo!baz").st_dev)
self.assertEqual(6, self.filesystem.get_object("z:!foo!baz").st_dev)
def test_that_drives_are_auto_mounted_case_insensitive(self):
self.filesystem.is_windows_fs = True
self.add_mount_points()
self.filesystem.is_case_sensitive = False
self.filesystem.create_dir("D:!foo!bar")
self.filesystem.create_file("e:!foo!baz")
self.assertEqual(5, self.filesystem.get_object("D:").st_dev)
self.assertEqual(5, self.filesystem.get_object("d:!foo!bar").st_dev)
self.assertEqual(6, self.filesystem.get_object("e:!foo").st_dev)
self.assertEqual(6, self.filesystem.get_object("E:!Foo!Baz").st_dev)
def test_that_unc_paths_are_auto_mounted(self):
self.filesystem.is_windows_fs = True
self.add_mount_points()
self.filesystem.create_dir("!!foo!bar!baz")
self.filesystem.create_file("!!foo!bar!bip!bop")
self.assertEqual(5, self.filesystem.get_object("!!foo!bar").st_dev)
self.assertEqual(5, self.filesystem.get_object("!!foo!bar!bip!bop").st_dev)
class ConvenienceMethodTest(RealFsTestCase):
def test_create_link_with_non_existent_parent(self):
self.skip_if_symlink_not_supported()
file1_path = self.make_path("test_file1")
link_path = self.make_path("nonexistent", "test_file2")
self.filesystem.create_file(file1_path, contents="link test")
self.assertEqual(self.os.stat(file1_path).st_nlink, 1)
self.filesystem.create_link(file1_path, link_path)
self.assertEqual(self.os.stat(file1_path).st_nlink, 2)
self.assertTrue(self.filesystem.exists(link_path))
def test_create_symlink_with_non_existent_parent(self):
self.skip_if_symlink_not_supported()
file1_path = self.make_path("test_file1")
link_path = self.make_path("nonexistent", "test_file2")
self.filesystem.create_file(file1_path, contents="symlink test")
self.filesystem.create_symlink(link_path, file1_path)
self.assertTrue(self.filesystem.exists(link_path))
self.assertTrue(self.filesystem.islink(link_path))
class RealFileSystemAccessTest(RealFsTestCase):
def setUp(self):
# use the real path separator to work with the real file system
self.filesystem = fake_filesystem.FakeFilesystem()
self.fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.pyfakefs_path = os.path.split(os.path.dirname(os.path.abspath(__file__)))[
0
]
self.root_path = os.path.split(self.pyfakefs_path)[0]
def test_add_non_existing_real_file_raises(self):
nonexisting_path = os.path.join("nonexisting", "test.txt")
with self.assertRaises(OSError):
self.filesystem.add_real_file(nonexisting_path)
self.assertFalse(self.filesystem.exists(nonexisting_path))
def test_add_non_existing_real_directory_raises(self):
nonexisting_path = "/nonexisting"
with self.raises_os_error(errno.ENOENT):
self.filesystem.add_real_directory(nonexisting_path)
self.assertFalse(self.filesystem.exists(nonexisting_path))
def test_existing_fake_file_raises(self):
real_file_path = __file__
self.filesystem.create_file(real_file_path)
with self.raises_os_error(errno.EEXIST):
self.filesystem.add_real_file(real_file_path)
def test_existing_fake_directory_raises(self):
self.filesystem.create_dir(self.root_path)
with self.raises_os_error(errno.EEXIST):
self.filesystem.add_real_directory(self.root_path)
def check_fake_file_stat(self, fake_file, real_file_path, target_path=None):
if target_path is None or target_path == real_file_path:
self.assertTrue(self.filesystem.exists(real_file_path))
else:
self.assertFalse(self.filesystem.exists(real_file_path))
self.assertTrue(self.filesystem.exists(target_path))
real_stat = os.stat(real_file_path)
self.assertIsNone(fake_file._byte_contents)
self.assertEqual(fake_file.st_size, real_stat.st_size)
self.assertAlmostEqual(fake_file.st_ctime, real_stat.st_ctime, places=5)
self.assertAlmostEqual(fake_file.st_atime, real_stat.st_atime, places=5)
self.assertAlmostEqual(fake_file.st_mtime, real_stat.st_mtime, places=5)
self.assertEqual(fake_file.st_uid, real_stat.st_uid)
self.assertEqual(fake_file.st_gid, real_stat.st_gid)
def check_read_only_file(self, fake_file, real_file_path):
with open(real_file_path, "rb") as f:
real_contents = f.read()
self.assertEqual(fake_file.byte_contents, real_contents)
if not is_root():
with self.raises_os_error(errno.EACCES):
self.fake_open(real_file_path, "w")
else:
with self.fake_open(real_file_path, "w"):
pass
def check_writable_file(self, fake_file, real_file_path):
with open(real_file_path, "rb") as f:
real_contents = f.read()
self.assertEqual(fake_file.byte_contents, real_contents)
with self.fake_open(real_file_path, "wb") as f:
f.write(b"test")
with open(real_file_path, "rb") as f:
real_contents1 = f.read()
self.assertEqual(real_contents1, real_contents)
with self.fake_open(real_file_path, "rb") as f:
fake_contents = f.read()
self.assertEqual(fake_contents, b"test")
def test_add_existing_real_file_read_only(self):
real_file_path = os.path.abspath(__file__)
fake_file = self.filesystem.add_real_file(real_file_path)
self.check_fake_file_stat(fake_file, real_file_path)
self.assertEqual(fake_file.st_mode & 0o333, 0)
self.check_read_only_file(fake_file, real_file_path)
def test_add_existing_real_file_read_write(self):
real_file_path = os.path.realpath(__file__)
fake_file = self.filesystem.add_real_file(real_file_path, read_only=False)
self.check_fake_file_stat(fake_file, real_file_path)
self.assertEqual(fake_file.st_mode, os.stat(real_file_path).st_mode)
self.check_writable_file(fake_file, real_file_path)
def test_add_real_file_to_existing_path(self):
real_file_path = os.path.abspath(__file__)
self.filesystem.create_file("/foo/bar")
with self.raises_os_error(errno.EEXIST):
self.filesystem.add_real_file(real_file_path, target_path="/foo/bar")
def test_add_real_file_to_non_existing_path(self):
real_file_path = os.path.abspath(__file__)
fake_file = self.filesystem.add_real_file(
real_file_path, target_path="/foo/bar"
)
self.check_fake_file_stat(fake_file, real_file_path, target_path="/foo/bar")
def test_write_to_real_file(self):
# regression test for #470
real_file_path = os.path.abspath(__file__)
self.filesystem.add_real_file(real_file_path, read_only=False)
with self.fake_open(real_file_path, "w") as f:
f.write("foo")
with self.fake_open(real_file_path, "rb") as f:
self.assertEqual(b"foo", f.read())
def test_add_existing_real_directory_read_only(self):
self.filesystem.add_real_directory(self.pyfakefs_path)
self.assertTrue(self.filesystem.exists(self.pyfakefs_path))
self.assertTrue(
self.filesystem.exists(
os.path.join(self.pyfakefs_path, "fake_filesystem.py")
)
)
self.assertTrue(
self.filesystem.exists(os.path.join(self.pyfakefs_path, "fake_pathlib.py"))
)
file_path = os.path.join(self.pyfakefs_path, "fake_filesystem_shutil.py")
fake_file = self.filesystem.resolve(file_path)
self.check_fake_file_stat(fake_file, file_path)
self.check_read_only_file(fake_file, file_path)
def test_add_existing_real_directory_tree(self):
self.filesystem.add_real_directory(self.root_path)
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fake_filesystem_test.py",
)
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(self.root_path, "pyfakefs", "fake_filesystem.py")
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(self.root_path, "pyfakefs", "__init__.py")
)
)
@contextlib.contextmanager
def create_symlinks(self, symlinks):
for link in symlinks:
os.symlink(link[0], link[1])
yield
for link in symlinks:
os.unlink(link[1])
def test_add_existing_real_directory_symlink(self):
fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
real_directory = os.path.join(self.root_path, "pyfakefs", "tests")
symlinks = [
(
"..",
os.path.join(real_directory, "fixtures", "symlink_dir_relative"),
),
(
"../all_tests.py",
os.path.join(real_directory, "fixtures", "symlink_file_relative"),
),
(
real_directory,
os.path.join(real_directory, "fixtures", "symlink_dir_absolute"),
),
(
os.path.join(real_directory, "all_tests.py"),
os.path.join(real_directory, "fixtures", "symlink_file_absolute"),
),
(
"/etc/something",
os.path.join(
real_directory, "fixtures", "symlink_file_absolute_outside"
),
),
]
self.filesystem.create_file("/etc/something")
with fake_open("/etc/something", "w") as f:
f.write("good morning")
try:
with self.create_symlinks(symlinks):
self.filesystem.add_real_directory(real_directory, lazy_read=False)
except OSError:
if self.is_windows:
raise unittest.SkipTest("Symlinks under Windows need admin privileges")
raise
for link in symlinks:
self.assertTrue(self.filesystem.islink(link[1]))
# relative
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_dir_relative",
)
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_dir_relative/all_tests.py",
)
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_file_relative",
)
)
)
# absolute
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_dir_absolute",
)
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_dir_absolute/all_tests.py",
)
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_file_absolute",
)
)
)
# outside
self.assertTrue(
self.filesystem.exists(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_file_absolute_outside",
)
)
)
self.assertEqual(
fake_open(
os.path.join(
self.root_path,
"pyfakefs",
"tests",
"fixtures/symlink_file_absolute_outside",
)
).read(),
"good morning",
)
def test_add_existing_real_directory_symlink_target_path(self):
self.skip_if_symlink_not_supported(force_real_fs=True)
real_directory = os.path.join(self.root_path, "pyfakefs", "tests")
symlinks = [
(
"..",
os.path.join(real_directory, "fixtures", "symlink_dir_relative"),
),
(
"../all_tests.py",
os.path.join(real_directory, "fixtures", "symlink_file_relative"),
),
]
with self.create_symlinks(symlinks):
self.filesystem.add_real_directory(
real_directory, target_path="/path", lazy_read=False
)
self.assertTrue(self.filesystem.exists("/path/fixtures/symlink_dir_relative"))
self.assertTrue(
self.filesystem.exists("/path/fixtures/symlink_dir_relative/all_tests.py")
)
self.assertTrue(self.filesystem.exists("/path/fixtures/symlink_file_relative"))
def test_add_existing_real_directory_symlink_lazy_read(self):
self.skip_if_symlink_not_supported(force_real_fs=True)
real_directory = os.path.join(self.root_path, "pyfakefs", "tests")
symlinks = [
(
"..",
os.path.join(real_directory, "fixtures", "symlink_dir_relative"),
),
(
"../all_tests.py",
os.path.join(real_directory, "fixtures", "symlink_file_relative"),
),
]
with self.create_symlinks(symlinks):
self.filesystem.add_real_directory(
real_directory, target_path="/path", lazy_read=True
)
self.assertTrue(
self.filesystem.exists("/path/fixtures/symlink_dir_relative")
)
self.assertTrue(
self.filesystem.exists(
"/path/fixtures/symlink_dir_relative/all_tests.py"
)
)
self.assertTrue(
self.filesystem.exists("/path/fixtures/symlink_file_relative")
)
def test_add_existing_real_directory_tree_to_existing_path(self):
self.filesystem.create_dir("/foo/bar")
with self.raises_os_error(errno.EEXIST):
self.filesystem.add_real_directory(self.root_path, target_path="/foo/bar")
def test_add_existing_real_directory_tree_to_other_path(self):
self.filesystem.add_real_directory(self.root_path, target_path="/foo/bar")
self.assertFalse(
self.filesystem.exists(
os.path.join(self.pyfakefs_path, "tests", "fake_filesystem_test.py")
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join(
"foo",
"bar",
"pyfakefs",
"tests",
"fake_filesystem_test.py",
)
)
)
self.assertFalse(
self.filesystem.exists(
os.path.join(self.root_path, "pyfakefs", "fake_filesystem.py")
)
)
self.assertTrue(
self.filesystem.exists(
os.path.join("foo", "bar", "pyfakefs", "__init__.py")
)
)
def test_get_object_from_lazily_added_real_directory(self):
self.filesystem.is_case_sensitive = True
self.filesystem.add_real_directory(self.root_path)
self.assertTrue(
self.filesystem.get_object(
os.path.join(self.root_path, "pyfakefs", "fake_filesystem.py")
)
)
self.assertTrue(
self.filesystem.get_object(
os.path.join(self.root_path, "pyfakefs", "__init__.py")
)
)
def test_add_existing_real_directory_lazily(self):
disk_size = 1024 * 1024 * 1024
real_dir_path = os.path.join(self.root_path, "pyfakefs")
self.filesystem.set_disk_usage(disk_size, real_dir_path)
self.filesystem.add_real_directory(real_dir_path)
# the directory contents have not been read, the the disk usage
# has not changed
self.assertEqual(disk_size, self.filesystem.get_disk_usage(real_dir_path).free)
# checking for existence shall read the directory contents
self.assertTrue(
self.filesystem.get_object(
os.path.join(real_dir_path, "fake_filesystem.py")
)
)
# so now the free disk space shall have decreased
self.assertGreater(
disk_size, self.filesystem.get_disk_usage(real_dir_path).free
)
def test_add_existing_real_directory_not_lazily(self):
disk_size = 1024 * 1024 * 1024
self.filesystem.set_disk_usage(disk_size, self.pyfakefs_path)
self.filesystem.add_real_directory(self.pyfakefs_path, lazy_read=False)
# the directory has been read, so the file sizes have
# been subtracted from the free space
self.assertGreater(
disk_size, self.filesystem.get_disk_usage(self.pyfakefs_path).free
)
def test_add_existing_real_directory_read_write(self):
self.filesystem.add_real_directory(self.pyfakefs_path, read_only=False)
self.assertTrue(self.filesystem.exists(self.pyfakefs_path))
self.assertTrue(
self.filesystem.exists(
os.path.join(self.pyfakefs_path, "fake_filesystem.py")
)
)
self.assertTrue(
self.filesystem.exists(os.path.join(self.pyfakefs_path, "fake_pathlib.py"))
)
file_path = os.path.join(self.pyfakefs_path, "pytest_plugin.py")
fake_file = self.filesystem.resolve(file_path)
self.check_fake_file_stat(fake_file, file_path)
self.check_writable_file(fake_file, file_path)
def test_add_existing_real_paths_read_only(self):
real_file_path = os.path.realpath(__file__)
fixture_path = os.path.join(self.pyfakefs_path, "tests", "fixtures")
self.filesystem.add_real_paths([real_file_path, fixture_path])
fake_file = self.filesystem.resolve(real_file_path)
self.check_fake_file_stat(fake_file, real_file_path)
self.check_read_only_file(fake_file, real_file_path)
real_file_path = os.path.join(fixture_path, "module_with_attributes.py")
fake_file = self.filesystem.resolve(real_file_path)
self.check_fake_file_stat(fake_file, real_file_path)
self.check_read_only_file(fake_file, real_file_path)
def test_add_existing_real_paths_read_write(self):
real_file_path = os.path.realpath(__file__)
fixture_path = os.path.join(self.pyfakefs_path, "tests", "fixtures")
self.filesystem.add_real_paths([real_file_path, fixture_path], read_only=False)
fake_file = self.filesystem.resolve(real_file_path)
self.check_fake_file_stat(fake_file, real_file_path)
self.check_writable_file(fake_file, real_file_path)
real_file_path = os.path.join(fixture_path, "module_with_attributes.py")
fake_file = self.filesystem.resolve(real_file_path)
self.check_fake_file_stat(fake_file, real_file_path)
self.check_writable_file(fake_file, real_file_path)
class FileSideEffectTests(TestCase):
def side_effect(self):
test_case = self
test_case.side_effect_called = False
def __side_effect(file_object):
test_case.side_effect_called = True
test_case.side_effect_file_object_content = file_object.contents
return __side_effect
def setUp(self):
# use the real path separator to work with the real file system
self.filesystem = fake_filesystem.FakeFilesystem()
self.filesystem.create_file("/a/b/file_one", side_effect=self.side_effect())
def test_side_effect_called(self):
fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.side_effect_called = False
with fake_open("/a/b/file_one", "w") as handle:
handle.write("foo")
self.assertTrue(self.side_effect_called)
def test_side_effect_file_object(self):
fake_open = fake_filesystem.FakeFileOpen(self.filesystem)
self.side_effect_called = False
with fake_open("/a/b/file_one", "w") as handle:
handle.write("foo")
self.assertEqual(self.side_effect_file_object_content, "foo")
if __name__ == "__main__":
unittest.main()
|