| | |
| | from __future__ import absolute_import |
| | from __future__ import print_function |
| | from __future__ import unicode_literals |
| | from contextlib import closing |
| | import io |
| | import os.path |
| | import shutil |
| |
|
| | from hwp5.hwp5html import HTMLTransform |
| | from hwp5.plat import get_xslt |
| | from hwp5.storage.fs import FileSystemStorage |
| |
|
| | from . import test_xmlmodel |
| |
|
| |
|
| | class TestBase(test_xmlmodel.TestBase): |
| |
|
| | def make_base_dir(self): |
| | base_dir = self.id() |
| | if os.path.exists(base_dir): |
| | shutil.rmtree(base_dir) |
| | os.mkdir(base_dir) |
| | return base_dir |
| |
|
| |
|
| | class HtmlConvTest(TestBase): |
| |
|
| | @property |
| | def xslt(self): |
| | return get_xslt() |
| |
|
| | @property |
| | def xhwp5_path(self): |
| | return self.id() + '.xhwp5' |
| |
|
| | @property |
| | def transform(self): |
| | return HTMLTransform() |
| |
|
| | def create_xhwp5(self): |
| | xhwp5_path = self.xhwp5_path |
| | with io.open(xhwp5_path, 'wb') as xhwp5_file: |
| | self.hwp5file.xmlevents(embedbin=False).dump(xhwp5_file) |
| | return xhwp5_path |
| |
|
| | def test_generate_css_file(self): |
| | base_dir = self.make_base_dir() |
| | css_path = os.path.join(base_dir, 'styles.css') |
| |
|
| | with closing(self.hwp5file) as hwp5file: |
| | with io.open(css_path, 'wb+') as f: |
| | self.transform.transform_hwp5_to_css(hwp5file, f) |
| |
|
| | def test_generate_html_file(self): |
| | base_dir = self.make_base_dir() |
| | html_path = os.path.join(base_dir, 'index.xhtml') |
| |
|
| | with closing(self.hwp5file) as hwp5file: |
| | with io.open(html_path, 'wb+') as f: |
| | self.transform.transform_hwp5_to_xhtml(hwp5file, f) |
| |
|
| | def test_extract_bindata_dir(self): |
| | base_dir = self.make_base_dir() |
| | hwp5file = self.hwp5file |
| |
|
| | bindata_dir = os.path.join(base_dir, 'bindata') |
| |
|
| | self.transform.extract_bindata_dir(hwp5file, bindata_dir) |
| |
|
| | bindata_stg = hwp5file['BinData'] |
| |
|
| | self.assertEqual(set(bindata_stg), |
| | set(FileSystemStorage(bindata_dir))) |
| |
|
| | def test_extract_bindata_dir_without_bindata(self): |
| | self.hwp5file_name = 'charshape.hwp' |
| | base_dir = self.make_base_dir() |
| | hwp5file = self.hwp5file |
| |
|
| | bindata_dir = os.path.join(base_dir, 'bindata') |
| |
|
| | self.transform.extract_bindata_dir(hwp5file, bindata_dir) |
| | self.assertFalse(os.path.exists(bindata_dir)) |
| |
|
