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