| | |
| |
|
| | from io import BytesIO |
| | import json |
| |
|
| | from hwp5 import recordstream as RS |
| | from hwp5.recordstream import RecordStream |
| | from hwp5.recordstream import dump_record |
| | from hwp5.recordstream import read_record |
| | from hwp5.recordstream import record_to_json |
| | from hwp5.storage import ExtraItemStorage |
| | from hwp5.tagids import HWPTAG_DOCUMENT_PROPERTIES |
| | from hwp5.tagids import HWPTAG_ID_MAPPINGS |
| | from hwp5.tagids import HWPTAG_PARA_HEADER |
| | from hwp5.utils import cached_property |
| |
|
| | from . import test_filestructure |
| |
|
| |
|
| | class TestBase(test_filestructure.TestBase): |
| |
|
| | @property |
| | def hwp5file_rec(self): |
| | return RS.Hwp5File(self.olestg) |
| |
|
| | hwp5file = hwp5file_rec |
| |
|
| |
|
| | class TestRecord(TestBase): |
| |
|
| | def test_read_record(self): |
| | docinfo_stream = self.hwp5file['DocInfo'] |
| |
|
| | record = read_record(docinfo_stream.open(), 0) |
| | self.assertEqual(HWPTAG_DOCUMENT_PROPERTIES, record['tagid']) |
| |
|
| | def test_dump_record(self): |
| | docinfo_stream = self.hwp5file['DocInfo'] |
| | record = read_record(docinfo_stream.open(), 0) |
| | stream = BytesIO() |
| | dump_record(stream, record) |
| | stream.seek(0) |
| | record2 = read_record(stream, 0) |
| | self.assertEqual(record2, record) |
| |
|
| |
|
| | class TestRecordStream(TestBase): |
| |
|
| | @cached_property |
| | def docinfo(self): |
| | return RecordStream(self.hwp5file_fs['DocInfo'], |
| | self.hwp5file_fs.header.version) |
| |
|
| | def test_records(self): |
| | self.assertEqual(67, len(list(self.docinfo.records()))) |
| |
|
| | def test_records_kwargs_treegroup(self): |
| | records = self.docinfo.records(treegroup=1) |
| | self.assertEqual(66, len(records)) |
| | self.assertEqual(HWPTAG_ID_MAPPINGS, records[0]['tagid']) |
| |
|
| | records = self.docinfo.records(treegroup=0) |
| | self.assertEqual(1, len(records)) |
| | self.assertEqual(HWPTAG_DOCUMENT_PROPERTIES, records[0]['tagid']) |
| |
|
| | records = self.bodytext.section(0).records(treegroup=5) |
| | self.assertEqual(26, records[0]['seqno']) |
| | self.assertEqual(37, len(records)) |
| |
|
| | def test_record(self): |
| | record = self.docinfo.record(0) |
| | self.assertEqual(0, record['seqno']) |
| |
|
| | record = self.docinfo.record(10) |
| | self.assertEqual(10, record['seqno']) |
| |
|
| | def test_records_treegrouped(self): |
| | groups = self.docinfo.records_treegrouped() |
| | document_properties_treerecords = next(groups) |
| | self.assertEqual(1, len(document_properties_treerecords)) |
| | idmappings_treerecords = next(groups) |
| | self.assertEqual(66, len(idmappings_treerecords)) |
| |
|
| | section = self.bodytext.section(0) |
| | for group_idx, records in enumerate(section.records_treegrouped()): |
| | |
| | self.assertEqual(HWPTAG_PARA_HEADER, records[0]['tagid']) |
| |
|
| | def test_records_treegrouped_as_iterable(self): |
| | groups = self.docinfo.records_treegrouped(group_as_list=False) |
| | group = next(groups) |
| | self.assertFalse(isinstance(group, list)) |
| |
|
| | def test_records_treegroped_as_list(self): |
| | groups = self.docinfo.records_treegrouped() |
| | group = next(groups) |
| | self.assertTrue(isinstance(group, list)) |
| |
|
| | def test_records_treegroup(self): |
| | records = self.docinfo.records_treegroup(1) |
| | self.assertEqual(66, len(records)) |
| | self.assertEqual(HWPTAG_ID_MAPPINGS, records[0]['tagid']) |
| |
|
| | records = self.docinfo.records_treegroup(0) |
| | self.assertEqual(1, len(records)) |
| | self.assertEqual(HWPTAG_DOCUMENT_PROPERTIES, records[0]['tagid']) |
| |
|
| | records = self.bodytext.section(0).records_treegroup(5) |
| | self.assertEqual(26, records[0]['seqno']) |
| | self.assertEqual(37, len(records)) |
| |
|
| |
|
| | class TestHwp5File(TestBase): |
| |
|
| | def test_if_hwp5file_contains_other_formats(self): |
| | stg = ExtraItemStorage(self.hwp5file) |
| | self.assertTrue('DocInfo.records' in list(stg)) |
| |
|
| | def test_docinfo(self): |
| | docinfo = self.hwp5file.docinfo |
| | self.assertTrue(isinstance(docinfo, RS.RecordStream)) |
| | records = list(docinfo.records()) |
| | self.assertEqual(67, len(records)) |
| |
|
| | def test_bodytext(self): |
| | bodytext = self.hwp5file.bodytext |
| | self.assertTrue(isinstance(bodytext, RS.Sections)) |
| | stg = ExtraItemStorage(bodytext) |
| | self.assertEqual(['Section0', 'Section0.records'], list(stg)) |
| |
|
| |
|
| | class TestJson(TestBase): |
| | def test_record_to_json(self): |
| | record = next(self.hwp5file.docinfo.records()) |
| | json_string = record_to_json(record) |
| | jsonobject = json.loads(json_string) |
| | self.assertEqual(16, jsonobject['tagid']) |
| | self.assertEqual(0, jsonobject['level']) |
| | self.assertEqual(26, jsonobject['size']) |
| | self.assertEqual(['01 00 01 00 01 00 01 00 01 00 01 00 01 00 00 00', |
| | '00 00 07 00 00 00 05 00 00 00'], |
| | jsonobject['payload']) |
| | self.assertEqual(0, jsonobject['seqno']) |
| | self.assertEqual('HWPTAG_DOCUMENT_PROPERTIES', jsonobject['tagname']) |
| |
|
| | def test_generate_json(self): |
| | records_json = self.hwp5file.docinfo.records_json() |
| | json_string = ''.join(records_json.generate()) |
| |
|
| | jsonobject = json.loads(json_string) |
| | self.assertEqual(67, len(jsonobject)) |
| |
|
