aboutsummaryrefslogtreecommitdiff
path: root/pyfakefs/tests/fake_tempfile_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'pyfakefs/tests/fake_tempfile_test.py')
-rw-r--r--pyfakefs/tests/fake_tempfile_test.py53
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()