diff options
Diffstat (limited to 'pyfakefs/tests/fake_tempfile_test.py')
-rw-r--r-- | pyfakefs/tests/fake_tempfile_test.py | 53 |
1 files changed, 25 insertions, 28 deletions
diff --git a/pyfakefs/tests/fake_tempfile_test.py b/pyfakefs/tests/fake_tempfile_test.py index 7ddd4e1..f208b56 100644 --- a/pyfakefs/tests/fake_tempfile_test.py +++ b/pyfakefs/tests/fake_tempfile_test.py @@ -39,79 +39,76 @@ class FakeTempfileModuleTest(fake_filesystem_unittest.TestCase): def test_named_temporary_file_no_delete(self): obj = tempfile.NamedTemporaryFile(delete=False) - obj.write(b'foo') + obj.write(b"foo") obj.close() file_obj = self.fs.get_object(obj.name) contents = file_obj.contents - self.assertEqual('foo', contents) - obj = tempfile.NamedTemporaryFile(mode='w', delete=False) - obj.write('foo') + self.assertEqual("foo", contents) + obj = tempfile.NamedTemporaryFile(mode="w", delete=False) + obj.write("foo") obj.close() file_obj = self.fs.get_object(obj.name) - self.assertEqual('foo', file_obj.contents) + self.assertEqual("foo", file_obj.contents) def test_mkstemp(self): next_fd = len(self.fs.open_files) temporary = tempfile.mkstemp() self.assertEqual(2, len(temporary)) self.assertTrue( - temporary[1].startswith( - os.path.join(tempfile.gettempdir(), 'tmp'))) + temporary[1].startswith(os.path.join(tempfile.gettempdir(), "tmp")) + ) self.assertEqual(next_fd, temporary[0]) self.assertTrue(self.fs.exists(temporary[1])) mode = 0o666 if self.fs.is_windows_fs else 0o600 - self.assertEqual(self.fs.get_object(temporary[1]).st_mode, - stat.S_IFREG | mode) - fh = os.fdopen(temporary[0], 'w+b') + self.assertEqual(self.fs.get_object(temporary[1]).st_mode, stat.S_IFREG | mode) + fh = os.fdopen(temporary[0], "w+b") self.assertEqual(temporary[0], fh.fileno()) def test_mkstemp_dir(self): """test tempfile.mkstemp(dir=).""" # expect fail: /dir does not exist with self.assertRaises(OSError): - tempfile.mkstemp(dir='/dir') + tempfile.mkstemp(dir="/dir") # expect pass: /dir exists - self.fs.create_dir('/dir') + self.fs.create_dir("/dir") next_fd = len(self.fs.open_files) - temporary = tempfile.mkstemp(dir='/dir') + temporary = tempfile.mkstemp(dir="/dir") self.assertEqual(2, len(temporary)) self.assertEqual(next_fd, temporary[0]) - self.assertTrue(temporary[1].startswith( - os.path.join(os.sep, 'dir', 'tmp'))) + self.assertTrue( + temporary[1].startswith(os.path.join(self.fs.root_dir_name, "dir", "tmp")) + ) self.assertTrue(self.fs.exists(temporary[1])) mode = 0o666 if self.fs.is_windows_fs else 0o600 - self.assertEqual(self.fs.get_object(temporary[1]).st_mode, - stat.S_IFREG | mode) + self.assertEqual(self.fs.get_object(temporary[1]).st_mode, stat.S_IFREG | mode) def test_mkdtemp(self): dirname = tempfile.mkdtemp() self.assertTrue(dirname) self.assertTrue(self.fs.exists(dirname)) - self.assertEqual(self.fs.get_object(dirname).st_mode, - stat.S_IFDIR | 0o700) + self.assertEqual(self.fs.get_object(dirname).st_mode, stat.S_IFDIR | 0o700) def test_temporary_directory(self): with tempfile.TemporaryDirectory() as tmpdir: self.assertTrue(tmpdir) self.assertTrue(self.fs.exists(tmpdir)) - self.assertEqual(self.fs.get_object(tmpdir).st_mode, - stat.S_IFDIR | 0o700) + self.assertEqual(self.fs.get_object(tmpdir).st_mode, stat.S_IFDIR | 0o700) def test_temporary_file(self): with tempfile.TemporaryFile() as f: - f.write(b'test') + f.write(b"test") f.seek(0) - self.assertEqual(b'test', f.read()) + self.assertEqual(b"test", f.read()) def test_temporay_file_with_dir(self): with self.assertRaises(FileNotFoundError): - tempfile.TemporaryFile(dir='/parent') - os.mkdir('/parent') + tempfile.TemporaryFile(dir="/parent") + os.mkdir("/parent") with tempfile.TemporaryFile() as f: - f.write(b'test') + f.write(b"test") f.seek(0) - self.assertEqual(b'test', f.read()) + self.assertEqual(b"test", f.read()) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main() |