diff options
Diffstat (limited to 'pyfakefs/tests/example_test.py')
-rw-r--r-- | pyfakefs/tests/example_test.py | 107 |
1 files changed, 51 insertions, 56 deletions
diff --git a/pyfakefs/tests/example_test.py b/pyfakefs/tests/example_test.py index 91225d0..126bbe7 100644 --- a/pyfakefs/tests/example_test.py +++ b/pyfakefs/tests/example_test.py @@ -38,15 +38,14 @@ from pyfakefs.tests import example # The module under test def load_tests(loader, tests, ignore): """Load the pyfakefs/example.py doctest tests into unittest.""" - return fake_filesystem_unittest.load_doctests( - loader, tests, ignore, example) + return fake_filesystem_unittest.load_doctests(loader, tests, ignore, example) class TestExample(fake_filesystem_unittest.TestCase): # pylint: disable=R0904 """Test the example module. - The os and shutil modules have been replaced with the fake modules, - so that all of the calls to os and shutil in the tested example code - occur in the fake filesystem. + The os and shutil modules have been replaced with the fake modules, + so that all of the calls to os and shutil in the tested example code + occur in the fake filesystem. """ def setUp(self): @@ -63,7 +62,7 @@ class TestExample(fake_filesystem_unittest.TestCase): # pylint: disable=R0904 # This is before setUpPyfakefs(), so still using the real file system self.filepath = os.path.realpath(__file__) - with io.open(self.filepath, 'rb') as f: + with io.open(self.filepath, "rb") as f: self.real_contents = f.read() self.setUpPyfakefs() @@ -73,61 +72,57 @@ class TestExample(fake_filesystem_unittest.TestCase): # pylint: disable=R0904 pass def test_create_file(self): - """Test example.create_file() which uses `open()` and `file.write()`. + """Test example.create_file() which uses `open()` + and `file.write()`. """ - self.assertFalse(os.path.isdir('/test')) - os.mkdir('/test') - self.assertTrue(os.path.isdir('/test')) + self.assertFalse(os.path.isdir("/test")) + os.mkdir("/test") + self.assertTrue(os.path.isdir("/test")) - self.assertFalse(os.path.exists('/test/file.txt')) - example.create_file('/test/file.txt') - self.assertTrue(os.path.exists('/test/file.txt')) + self.assertFalse(os.path.exists("/test/file.txt")) + example.create_file("/test/file.txt") + self.assertTrue(os.path.exists("/test/file.txt")) def test_delete_file(self): """Test example.delete_file() which uses `os.remove()`.""" - self.fs.create_file('/test/full.txt', - contents='First line\n' - 'Second Line\n') - self.assertTrue(os.path.exists('/test/full.txt')) - example.delete_file('/test/full.txt') - self.assertFalse(os.path.exists('/test/full.txt')) + self.fs.create_file("/test/full.txt", contents="First line\n" "Second Line\n") + self.assertTrue(os.path.exists("/test/full.txt")) + example.delete_file("/test/full.txt") + self.assertFalse(os.path.exists("/test/full.txt")) def test_file_exists(self): """Test example.path_exists() which uses `os.path.exists()`.""" - self.assertFalse(example.path_exists('/test/empty.txt')) - self.fs.create_file('/test/empty.txt') - self.assertTrue(example.path_exists('/test/empty.txt')) + self.assertFalse(example.path_exists("/test/empty.txt")) + self.fs.create_file("/test/empty.txt") + self.assertTrue(example.path_exists("/test/empty.txt")) def test_get_globs(self): """Test example.get_glob().""" - self.assertFalse(os.path.isdir('/test')) - self.fs.create_dir('/test/dir1/dir2a') - self.assertTrue(os.path.isdir('/test/dir1/dir2a')) + self.assertFalse(os.path.isdir("/test")) + self.fs.create_dir("/test/dir1/dir2a") + self.assertTrue(os.path.isdir("/test/dir1/dir2a")) # os.mkdirs() works, too. - os.makedirs('/test/dir1/dir2b') - self.assertTrue(os.path.isdir('/test/dir1/dir2b')) + os.makedirs("/test/dir1/dir2b") + self.assertTrue(os.path.isdir("/test/dir1/dir2b")) - self.assertEqual(example.get_glob('/test/dir1/nonexistent*'), - []) - is_windows = sys.platform.startswith('win') - matching_paths = sorted(example.get_glob('/test/dir1/dir*')) + self.assertEqual(example.get_glob("/test/dir1/nonexistent*"), []) + is_windows = sys.platform.startswith("win") + matching_paths = sorted(example.get_glob("/test/dir1/dir*")) if is_windows: - self.assertEqual(matching_paths, - [r'/test/dir1\dir2a', r'/test/dir1\dir2b']) + self.assertEqual(matching_paths, [r"/test/dir1\dir2a", r"/test/dir1\dir2b"]) else: - self.assertEqual(matching_paths, - ['/test/dir1/dir2a', '/test/dir1/dir2b']) + self.assertEqual(matching_paths, ["/test/dir1/dir2a", "/test/dir1/dir2b"]) def test_rm_tree(self): """Test example.rm_tree() using `shutil.rmtree()`.""" - self.fs.create_dir('/test/dir1/dir2a') + self.fs.create_dir("/test/dir1/dir2a") # os.mkdirs() works, too. - os.makedirs('/test/dir1/dir2b') - self.assertTrue(os.path.isdir('/test/dir1/dir2b')) - self.assertTrue(os.path.isdir('/test/dir1/dir2a')) + os.makedirs("/test/dir1/dir2b") + self.assertTrue(os.path.isdir("/test/dir1/dir2b")) + self.assertTrue(os.path.isdir("/test/dir1/dir2a")) - example.rm_tree('/test/dir1') - self.assertFalse(os.path.exists('/test/dir1')) + example.rm_tree("/test/dir1") + self.assertFalse(os.path.exists("/test/dir1")) def test_os_scandir(self): """Test example.scandir() which uses `os.scandir()`. @@ -136,20 +131,21 @@ class TestExample(fake_filesystem_unittest.TestCase): # pylint: disable=R0904 fake filesystem path entries are returned instead of `os.DirEntry` objects. """ - self.fs.create_file('/test/text.txt') - self.fs.create_dir('/test/dir') - self.fs.create_file('/linktest/linked') - self.fs.create_symlink('/test/linked_file', '/linktest/linked') + self.fs.create_file("/test/text.txt") + self.fs.create_dir("/test/dir") + self.fs.create_file("/linktest/linked") + self.fs.create_symlink("/test/linked_file", "/linktest/linked") - entries = sorted(example.scan_dir('/test'), key=lambda e: e.name) + entries = sorted(example.scan_dir("/test"), key=lambda e: e.name) self.assertEqual(3, len(entries)) - self.assertEqual('linked_file', entries[1].name) + self.assertEqual("linked_file", entries[1].name) self.assertTrue(entries[0].is_dir()) self.assertTrue(entries[1].is_symlink()) self.assertTrue(entries[2].is_file()) - @unittest.skipIf(not use_scandir_package, - 'Testing only if scandir module is installed') + @unittest.skipIf( + not use_scandir_package, "Testing only if scandir module is installed" + ) def test_scandir_scandir(self): """Test example.scandir() which uses `scandir.scandir()`. @@ -157,23 +153,22 @@ class TestExample(fake_filesystem_unittest.TestCase): # pylint: disable=R0904 the fake filesystem path entries are returned instead of `scandir.DirEntry` objects. """ - self.fs.create_file('/test/text.txt') - self.fs.create_dir('/test/dir') + self.fs.create_file("/test/text.txt") + self.fs.create_dir("/test/dir") - entries = sorted(example.scan_dir('/test'), key=lambda e: e.name) + entries = sorted(example.scan_dir("/test"), key=lambda e: e.name) self.assertEqual(2, len(entries)) - self.assertEqual('text.txt', entries[1].name) + self.assertEqual("text.txt", entries[1].name) self.assertTrue(entries[0].is_dir()) self.assertTrue(entries[1].is_file()) def test_real_file_access(self): """Test `example.file_contents()` for a real file after adding it using - `add_real_file()`.""" + `add_real_file()`.""" with self.assertRaises(OSError): example.file_contents(self.filepath) self.fs.add_real_file(self.filepath) - self.assertEqual(example.file_contents(self.filepath), - self.real_contents) + self.assertEqual(example.file_contents(self.filepath), self.real_contents) if __name__ == "__main__": |