def test_export_img_with_non_visio_file(self):
try:
tmpdir = mkdtemp()
with self.assertRaises(IOError):
export_img(__file__,
os.path.join(tmpdir, 'output.png'), None, None)
finally:
rmtree(tmpdir)
def test_export_img_from_vsd(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'multipages.vsd'),
os.path.join(tmpdir, 'output.png'), None, None)
self.assertEqual(['output1.png', 'output2.png'],
os.listdir(tmpdir))
finally:
rmtree(tmpdir)
def test_export_img_from_vsd(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'multipages.vsd'),
os.path.join(tmpdir, 'output.png'), None, None)
self.assertEqual(['output1.png', 'output2.png'],
os.listdir(tmpdir))
finally:
rmtree(tmpdir)
def test_export_img_if_visio_not_found(self, win32com_client):
from pywintypes import com_error
win32com_client.Dispatch.side_effect = com_error
try:
tmpdir = mkdtemp()
with self.assertRaises(OSError):
export_img(os.path.join(EXAMPLE_DIR, 'singlepage.vsdx'),
os.path.join(tmpdir, 'output.png'), None, None)
finally:
rmtree(tmpdir)
def test_export_img_multipages2(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'multipages2.vsdx'),
os.path.join(tmpdir, 'output.png'), None, None)
expected = ['output01.png', 'output02.png', 'output03.png',
'output04.png', 'output05.png', 'output06.png',
'output07.png', 'output08.png', 'output09.png',
'output10.png']
self.assertEqual(expected, os.listdir(tmpdir))
finally:
rmtree(tmpdir)
def test_export_img_singlepage_to_gif(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'singlepage.vsdx'),
os.path.join(tmpdir, 'output.gif'), None, None)
self.assertEqual(['output.gif'], os.listdir(tmpdir))
expected = os.path.join(EXAMPLE_DIR, 'singlepage', 'output.gif')
actual = os.path.join(tmpdir, 'output.gif')
self.assertEqual(open(expected, 'rb').read(),
open(actual, 'rb').read())
finally:
rmtree(tmpdir)
def test_export_img_multipages2(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'multipages2.vsdx'),
os.path.join(tmpdir, 'output.png'), None, None)
expected = [
'output01.png', 'output02.png', 'output03.png', 'output04.png',
'output05.png', 'output06.png', 'output07.png', 'output08.png',
'output09.png', 'output10.png'
]
self.assertEqual(expected, os.listdir(tmpdir))
finally:
rmtree(tmpdir)
def test_export_img_singlepage_to_gif(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'singlepage.vsdx'),
os.path.join(tmpdir, 'output.gif'), None, None)
self.assertEqual(['output.gif'], os.listdir(tmpdir))
expected = os.path.join(EXAMPLE_DIR, 'singlepage', 'output.gif')
actual = os.path.join(tmpdir, 'output.gif')
self.assertEqual(
open(expected, 'rb').read(),
open(actual, 'rb').read())
finally:
rmtree(tmpdir)
def test_export_img_multipages_with_pagename(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'multipages.vsdx'),
os.path.join(tmpdir, 'output.png'), None, u"ページ - 2")
self.assertEqual(['output.png'], os.listdir(tmpdir))
expected = os.path.join(EXAMPLE_DIR, 'multipages', 'output2.png')
actual = os.path.join(tmpdir, 'output.png')
self.assertEqual(open(expected, 'rb').read(),
open(actual, 'rb').read())
finally:
rmtree(tmpdir)
def test_export_img_multipages_with_pagename(self):
try:
tmpdir = mkdtemp()
export_img(os.path.join(EXAMPLE_DIR, 'multipages.vsdx'),
os.path.join(tmpdir, 'output.png'), None, u"ページ - 2")
self.assertEqual(['output.png'], os.listdir(tmpdir))
expected = os.path.join(EXAMPLE_DIR, 'multipages', 'output2.png')
actual = os.path.join(tmpdir, 'output.png')
self.assertEqual(
open(expected, 'rb').read(),
open(actual, 'rb').read())
finally:
rmtree(tmpdir)
def test_export_img_output_dir_not_found(self):
with self.assertRaises(IOError):
export_img(os.path.join(EXAMPLE_DIR, 'singlepage.vsdx'),
'/path/to/output.png', None, None)
def test_export_img_visio_file_not_found(self):
with self.assertRaises(IOError):
export_img('/path/to/notexist.vsd', '/path/to/output.png', None,
None)
def test_export_img_output_dir_not_found(self):
with self.assertRaises(IOError):
export_img(os.path.join(EXAMPLE_DIR, 'singlepage.vsdx'),
'/path/to/output.png', None, None)
def test_export_img_visio_file_not_found(self):
with self.assertRaises(IOError):
export_img('/path/to/notexist.vsd', '/path/to/output.png',
None, None)
