mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-27 02:16:19 +03:00
151 lines
3.7 KiB
Python
151 lines
3.7 KiB
Python
from helper import unittest, PillowTestCase
|
|
|
|
import datetime
|
|
|
|
import PIL.OleFileIO as OleFileIO
|
|
|
|
|
|
class TestOleFileIo(PillowTestCase):
|
|
|
|
def test_isOleFile(self):
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
|
|
self.assertTrue(OleFileIO.isOleFile(ole_file))
|
|
with open(ole_file, 'rb') as fp:
|
|
self.assertTrue(OleFileIO.isOleFile(fp))
|
|
self.assertTrue(OleFileIO.isOleFile(fp.read()))
|
|
|
|
non_ole_file = "Tests/images/flower.jpg"
|
|
|
|
self.assertFalse(OleFileIO.isOleFile(non_ole_file))
|
|
with open(non_ole_file, 'rb') as fp:
|
|
self.assertFalse(OleFileIO.isOleFile(fp))
|
|
self.assertFalse(OleFileIO.isOleFile(fp.read()))
|
|
|
|
def test_exists_worddocument(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
exists = ole.exists('worddocument')
|
|
|
|
# Assert
|
|
self.assertTrue(exists)
|
|
ole.close()
|
|
|
|
def test_exists_no_vba_macros(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
exists = ole.exists('macros/vba')
|
|
|
|
# Assert
|
|
self.assertFalse(exists)
|
|
ole.close()
|
|
|
|
def test_get_type(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
entry_type = ole.get_type('worddocument')
|
|
|
|
# Assert
|
|
self.assertEqual(entry_type, OleFileIO.STGTY_STREAM)
|
|
ole.close()
|
|
|
|
def test_get_size(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
size = ole.get_size('worddocument')
|
|
|
|
# Assert
|
|
self.assertGreater(size, 0)
|
|
ole.close()
|
|
|
|
def test_get_rootentry_name(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
root = ole.get_rootentry_name()
|
|
|
|
# Assert
|
|
self.assertEqual(root, "Root Entry")
|
|
ole.close()
|
|
|
|
def test_meta(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
meta = ole.get_metadata()
|
|
|
|
# Assert
|
|
self.assertEqual(meta.author, b"Laurence Ipsum")
|
|
self.assertEqual(meta.num_pages, 1)
|
|
ole.close()
|
|
|
|
def test_gettimes(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
root_entry = ole.direntries[0]
|
|
|
|
# Act
|
|
ctime = root_entry.getctime()
|
|
mtime = root_entry.getmtime()
|
|
|
|
# Assert
|
|
self.assertIsInstance(ctime, type(None))
|
|
self.assertIsInstance(mtime, datetime.datetime)
|
|
self.assertEqual(ctime, None)
|
|
self.assertEqual(mtime.year, 2014)
|
|
ole.close()
|
|
|
|
def test_listdir(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
|
|
# Act
|
|
dirlist = ole.listdir()
|
|
|
|
# Assert
|
|
self.assertIn(['WordDocument'], dirlist)
|
|
ole.close()
|
|
|
|
def test_debug(self):
|
|
# Arrange
|
|
ole_file = "Tests/images/test-ole-file.doc"
|
|
ole = OleFileIO.OleFileIO(ole_file)
|
|
meta = ole.get_metadata()
|
|
|
|
# Act
|
|
OleFileIO.set_debug_mode(True)
|
|
ole.dumpdirectory()
|
|
meta.dump()
|
|
|
|
OleFileIO.set_debug_mode(False)
|
|
ole.dumpdirectory()
|
|
meta.dump()
|
|
|
|
# Assert
|
|
# No assert, just check they run ok
|
|
ole.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# End of file
|