diff options
Diffstat (limited to 'test/test_fileutils.py')
-rw-r--r-- | test/test_fileutils.py | 76 |
1 files changed, 40 insertions, 36 deletions
diff --git a/test/test_fileutils.py b/test/test_fileutils.py index 555e73f..49955cd 100644 --- a/test/test_fileutils.py +++ b/test/test_fileutils.py @@ -27,47 +27,48 @@ from logilab.common.testlib import TestCase, unittest_main, unittest from logilab.common.fileutils import * -DATA_DIR = join(os.path.abspath(os.path.dirname(__file__)), 'data') -NEWLINES_TXT = join(DATA_DIR, 'newlines.txt') +DATA_DIR = join(os.path.abspath(os.path.dirname(__file__)), "data") +NEWLINES_TXT = join(DATA_DIR, "newlines.txt") class FirstleveldirectoryTC(TestCase): - def test_known_values_first_level_directory(self): """return the first level directory of a path""" - self.assertEqual(first_level_directory('truc/bidule/chouette'), 'truc', None) - self.assertEqual(first_level_directory('/truc/bidule/chouette'), '/', None) + self.assertEqual(first_level_directory("truc/bidule/chouette"), "truc", None) + self.assertEqual(first_level_directory("/truc/bidule/chouette"), "/", None) + class IsBinaryTC(TestCase): def test(self): - self.assertEqual(is_binary('toto.txt'), 0) - #self.assertEqual(is_binary('toto.xml'), 0) - self.assertEqual(is_binary('toto.bin'), 1) - self.assertEqual(is_binary('toto.sxi'), 1) - self.assertEqual(is_binary('toto.whatever'), 1) + self.assertEqual(is_binary("toto.txt"), 0) + # self.assertEqual(is_binary('toto.xml'), 0) + self.assertEqual(is_binary("toto.bin"), 1) + self.assertEqual(is_binary("toto.sxi"), 1) + self.assertEqual(is_binary("toto.whatever"), 1) + class GetModeTC(TestCase): def test(self): - self.assertEqual(write_open_mode('toto.txt'), 'w') - #self.assertEqual(write_open_mode('toto.xml'), 'w') - self.assertEqual(write_open_mode('toto.bin'), 'wb') - self.assertEqual(write_open_mode('toto.sxi'), 'wb') + self.assertEqual(write_open_mode("toto.txt"), "w") + # self.assertEqual(write_open_mode('toto.xml'), 'w') + self.assertEqual(write_open_mode("toto.bin"), "wb") + self.assertEqual(write_open_mode("toto.sxi"), "wb") + class NormReadTC(TestCase): def test_known_values_norm_read(self): with io.open(NEWLINES_TXT) as f: data = f.read() - self.assertEqual(data.strip(), '\n'.join(['# mixed new lines', '1', '2', '3'])) + self.assertEqual(data.strip(), "\n".join(["# mixed new lines", "1", "2", "3"])) class LinesTC(TestCase): def test_known_values_lines(self): - self.assertEqual(lines(NEWLINES_TXT), - ['# mixed new lines', '1', '2', '3']) + self.assertEqual(lines(NEWLINES_TXT), ["# mixed new lines", "1", "2", "3"]) def test_known_values_lines_comment(self): - self.assertEqual(lines(NEWLINES_TXT, comments='#'), - ['1', '2', '3']) + self.assertEqual(lines(NEWLINES_TXT, comments="#"), ["1", "2", "3"]) + class ExportTC(TestCase): def setUp(self): @@ -76,18 +77,19 @@ class ExportTC(TestCase): def test(self): export(DATA_DIR, self.tempdir, verbose=0) - self.assertTrue(exists(join(self.tempdir, '__init__.py'))) - self.assertTrue(exists(join(self.tempdir, 'sub'))) - self.assertTrue(not exists(join(self.tempdir, '__init__.pyc'))) - self.assertTrue(not exists(join(self.tempdir, 'CVS'))) + self.assertTrue(exists(join(self.tempdir, "__init__.py"))) + self.assertTrue(exists(join(self.tempdir, "sub"))) + self.assertTrue(not exists(join(self.tempdir, "__init__.pyc"))) + self.assertTrue(not exists(join(self.tempdir, "CVS"))) def tearDown(self): shutil.rmtree(self.tempdir) + class ProtectedFileTC(TestCase): def setUp(self): - self.rpath = join(DATA_DIR, 'write_protected_file.txt') - self.rwpath = join(DATA_DIR, 'normal_file.txt') + self.rpath = join(DATA_DIR, "write_protected_file.txt") + self.rwpath = join(DATA_DIR, "normal_file.txt") # Make sure rpath is not writable ! os.chmod(self.rpath, 33060) # Make sure rwpath is writable ! @@ -96,51 +98,53 @@ class ProtectedFileTC(TestCase): def test_mode_change(self): """tests that mode is changed when needed""" # test on non-writable file - #self.assertTrue(not os.access(self.rpath, os.W_OK)) + # self.assertTrue(not os.access(self.rpath, os.W_OK)) self.assertTrue(not os.stat(self.rpath).st_mode & S_IWRITE) - wp_file = ProtectedFile(self.rpath, 'w') + wp_file = ProtectedFile(self.rpath, "w") self.assertTrue(os.stat(self.rpath).st_mode & S_IWRITE) self.assertTrue(os.access(self.rpath, os.W_OK)) # test on writable-file self.assertTrue(os.stat(self.rwpath).st_mode & S_IWRITE) self.assertTrue(os.access(self.rwpath, os.W_OK)) - wp_file = ProtectedFile(self.rwpath, 'w') + wp_file = ProtectedFile(self.rwpath, "w") self.assertTrue(os.stat(self.rwpath).st_mode & S_IWRITE) self.assertTrue(os.access(self.rwpath, os.W_OK)) def test_restore_on_close(self): """tests original mode is restored on close""" # test on non-writable file - #self.assertTrue(not os.access(self.rpath, os.W_OK)) + # self.assertTrue(not os.access(self.rpath, os.W_OK)) self.assertTrue(not os.stat(self.rpath).st_mode & S_IWRITE) - ProtectedFile(self.rpath, 'w').close() - #self.assertTrue(not os.access(self.rpath, os.W_OK)) + ProtectedFile(self.rpath, "w").close() + # self.assertTrue(not os.access(self.rpath, os.W_OK)) self.assertTrue(not os.stat(self.rpath).st_mode & S_IWRITE) # test on writable-file self.assertTrue(os.access(self.rwpath, os.W_OK)) self.assertTrue(os.stat(self.rwpath).st_mode & S_IWRITE) - ProtectedFile(self.rwpath, 'w').close() + ProtectedFile(self.rwpath, "w").close() self.assertTrue(os.access(self.rwpath, os.W_OK)) self.assertTrue(os.stat(self.rwpath).st_mode & S_IWRITE) def test_mode_change_on_append(self): """tests that mode is changed when file is opened in 'a' mode""" - #self.assertTrue(not os.access(self.rpath, os.W_OK)) + # self.assertTrue(not os.access(self.rpath, os.W_OK)) self.assertTrue(not os.stat(self.rpath).st_mode & S_IWRITE) - wp_file = ProtectedFile(self.rpath, 'a') + wp_file = ProtectedFile(self.rpath, "a") self.assertTrue(os.access(self.rpath, os.W_OK)) self.assertTrue(os.stat(self.rpath).st_mode & S_IWRITE) wp_file.close() - #self.assertTrue(not os.access(self.rpath, os.W_OK)) + # self.assertTrue(not os.access(self.rpath, os.W_OK)) self.assertTrue(not os.stat(self.rpath).st_mode & S_IWRITE) if sys.version_info < (3, 0): + def load_tests(loader, tests, ignore): from logilab.common import fileutils + tests.addTests(doctest.DocTestSuite(fileutils)) return tests -if __name__ == '__main__': +if __name__ == "__main__": unittest_main() |