diff options
Diffstat (limited to 'systrace/catapult/common/py_utils/py_utils/tempfile_ext_unittest.py')
-rw-r--r-- | systrace/catapult/common/py_utils/py_utils/tempfile_ext_unittest.py | 37 |
1 files changed, 1 insertions, 36 deletions
diff --git a/systrace/catapult/common/py_utils/py_utils/tempfile_ext_unittest.py b/systrace/catapult/common/py_utils/py_utils/tempfile_ext_unittest.py index 76a0efd..6844623 100644 --- a/systrace/catapult/common/py_utils/py_utils/tempfile_ext_unittest.py +++ b/systrace/catapult/common/py_utils/py_utils/tempfile_ext_unittest.py @@ -2,15 +2,14 @@ # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. -import filecmp import os -import shutil from py_utils import tempfile_ext from pyfakefs import fake_filesystem_unittest class NamedTemporaryDirectoryTest(fake_filesystem_unittest.TestCase): + def setUp(self): self.setUpPyfakefs() @@ -38,37 +37,3 @@ class NamedTemporaryDirectoryTest(fake_filesystem_unittest.TestCase): self.fs.CreateDirectory(test_dir) with tempfile_ext.NamedTemporaryDirectory(dir=test_dir) as d: self.assertEquals(test_dir, os.path.dirname(d)) - - -class TemporaryFilesTest(fake_filesystem_unittest.TestCase): - def setUp(self): - self.setUpPyfakefs() - - def tearDown(self): - self.tearDownPyfakefs() - - def testNamedTemporaryFile(self): - with tempfile_ext.NamedTemporaryFile() as f: - self.assertTrue(os.path.isfile(f.name)) - f.write('<data>') - f.close() - self.assertTrue(os.path.exists(f.name)) - with open(f.name) as f2: - self.assertEqual(f2.read(), '<data>') - - self.assertFalse(os.path.exists(f.name)) - - def testTemporaryFileName(self): - with tempfile_ext.TemporaryFileName('foo') as filepath: - self.assertTrue(os.path.basename(filepath), 'foo') - self.assertFalse(os.path.exists(filepath)) - - with open(filepath, 'w') as f: - f.write('<data>') - self.assertTrue(os.path.exists(filepath)) - - shutil.copyfile(filepath, filepath + '.bak') - self.assertTrue(filecmp.cmp(filepath, filepath + '.bak')) - - self.assertFalse(os.path.exists(filepath)) - self.assertFalse(os.path.exists(os.path.dirname(filepath))) |