def test_comparison(self):
the_file = Py7File(self.test_file)
identical_file = the_file.copy('to_compare.file')
different_file = Py7File(self.test_file_zip)
self.assertEqual(the_file, identical_file)
self.assertEqual(the_file, os.path.abspath('to_compare.file'))
self.assertEqual(the_file, u'to_compare.file')
self.assertEqual(the_file, file('to_compare.file'))
self.assertNotEqual(the_file, different_file)
identical_file.delete()
def test_unzip_gz(self):
the_file = Py7File(self.test_file_gz)
unzipped = the_file.unzip()
filenames = [f.filename for f in unzipped]
self.assertEqual(len(filenames), 1)
self.assertIn('gz_test.txt', filenames)
the_file.cleanup()
def test_unzip_returns(self):
the_file = Py7File(self.test_file_zip)
unzipped = the_file.unzip()
filenames = [f.filename for f in unzipped]
self.assertEqual(len(filenames), 2)
self.assertIn('file_in_root.txt', filenames)
self.assertIn('file_in_subfolder.txt', filenames)
the_file.cleanup()
def test_unzip(self):
the_file = Py7File(self.test_file_zip)
the_file.unzip()
self.assertTrue(os.path.isdir('zip_test_unzipped'))
self.assertTrue((os.path.isfile('zip_test_unzipped/file_in_root.txt')))
the_file.rezip()
self.assertTrue(os.path.exists(self.test_file_zip))
self.assertFalse(os.path.isdir('zip_test_unzipped'))
def test_restore(self):
test_file = Py7File(self.test_file)
test_file.backup()
test_file.backup()
test_file.delete()
test_file.restore()
self.assertTrue(os.path.isfile(self.test_file))
test_file.cleanup()
def test_from_cwd(self):
the_file = Py7File(os.path.basename(self.test_file))
self.assertEqual(os.path.basename(self.test_file), the_file.filename)
self.assertEqual(self.root, the_file.location)
self.assertEqual(
os.path.splitext(self.test_file)[-1].lstrip('.'),
the_file.extension)
self.assertEqual(
os.path.basename(self.test_file).split('.')[0], the_file.trunc)
def test_no_extension(self):
the_file = Py7File(self.test_file_noext)
self.assertEqual(the_file.get_mimeptype(), None)
backup1 = the_file.backup()
backup2 = the_file.backup()
self.assertEqual(backup1.filename, the_file.filename + '_backup_001')
self.assertEqual(backup2.filename, the_file.filename + '_backup_002')
the_file.delete_backups()
self.assertFalse(backup1.exists())
self.assertFalse(backup2.exists())
def test_epub_file(self):
the_file = Py7File(self.test_epub)
copied_file = the_file.copy(
os.path.join(self.root, 'test', 'test_copy.epub'))
epub_file_obj = file(copied_file.filepath)
the_epub = EpubFile(epub_file_obj)
the_epub.unzip()
self.assertTrue(
os.path.isfile(os.path.join(the_epub.zipdir, 'mimetype')))
the_epub.rezip()
self.assertTrue(zipfile.is_zipfile(copied_file.filepath))
copied_file.delete()
def test_copy(self):
# create testfile
test_file = Py7File(self.test_file)
new_file = test_file.copy('test_copy.txt')
self.assertTrue(
os.path.isfile(os.path.join(self.root, 'test_copy.txt')))
self.assertEqual(new_file.filename, 'test_copy.txt')
# testing secure default
self.assertRaises(IOError, test_file.copy, 'test_copy.txt')
# again with overwrite
new_file2 = test_file.copy('test_copy.txt', secure=False)
self.assertTrue(
os.path.isfile(os.path.join(self.root, 'test_copy.txt')))
self.assertEqual(new_file2.filename, 'test_copy.txt')
# cleanup
new_file.delete()
def test_backup(self):
test_file = Py7File(self.test_file)
test_file.backup()
self.assertTrue(
os.path.isfile(test_file.trunc + '_backup_001.' +
test_file.extension))
test_file.backup()
self.assertTrue(
os.path.isfile(test_file.trunc + '_backup_002.' +
test_file.extension))
test_file.delete_backups()
self.assertFalse(
os.path.isfile(test_file.trunc + '_backup_001.' +
test_file.extension))
self.assertFalse(
os.path.isfile(test_file.trunc + '_backup_002.' +
test_file.extension))
def test_file_object(self):
f_obj = file(self.test_file)
the_file = Py7File(f_obj)
self.assertEqual(self.root, the_file.location)
def test_init(self):
self.assertEqual(Py7File(self.test_file).filepath, self.test_file)
self.assertEqual(
Py7File(self.test_file_utf8).filepath, self.test_file_utf8)
self.assertEqual(Py7File(self.test_file_zip).extension, 'zip')
self.assertEqual(Py7File(self.test_file_noext).extension, '')
def test_object(self):
return Py7File(self.test_file)
def test_get_sanitized_filename(self):
obj = Py7File(self.test_file_utf8)
self.assertEqual("mu_t_be_german.txt", obj.get_sanitized_filename())
obj = Py7File(unicode(self.test_file_utf8))
self.assertEqual("mu_t_be_german.txt", obj.get_sanitized_filename())
def test_get_filesize(self):
the_file = Py7File(self.test_file)
self.assertTrue(the_file.get_filesize())
self.assertTrue(isinstance(float(the_file.get_filesize()), float))
def test_get_number(self):
test_file_no_num = Py7File(self.test_file)
self.assertIsNone(test_file_no_num.get_number())
test_file_numbered = Py7File(self.numbered_file)
self.assertIsInstance(test_file_numbered.get_number(), int)
self.assertEqual(test_file_numbered.get_number(), 26)
def test_repr(self):
the_file = Py7File(self.test_file)
self.assertEqual(the_file, eval(repr(the_file)))
def test_read(self):
self.assertIsInstance(Py7File(self.test_file).read(), str)
self.assertEqual(
Py7File(self.test_file).read(), 'This is a file for testing')
def test_is_zip_file(self):
self.assertTrue(Py7File(self.test_file_zip).is_zip_file())
self.assertTrue(Py7File(self.test_file_zip_noext).is_zip_file())
self.assertFalse(Py7File(self.test_file).is_zip_file())
self.assertFalse(Py7File(self.test_file_utf16).is_zip_file())
def test_special_chars(self):
the_file = Py7File(self.test_file_utf8)
the_file.backup()
the_file.cleanup()
def test_is_binary(self):
self.assertFalse(Py7File(self.test_file).is_binary())
self.assertFalse(Py7File(self.test_file_utf8).is_binary())
self.assertFalse(Py7File(self.test_file_utf16).is_binary())
self.assertTrue(Py7File(self.test_file_zip).is_binary())