| |
| 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)) |
|
|
