cppreference-doc/tests/test_devhelp_qch.py at master · dataaip/cppreference-doc

50 lines (39 loc) · 1.87 KB

# This file is part of cppreference-doc

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program. If not, see http://www.gnu.org/licenses/.

import os

import unittest

from lxml import etree

from index_transform.devhelp_qch import convert_devhelp_to_qch

class TestConvertDevhelpToQch(unittest.TestCase):

def test_convert_devhelp_to_qch(self):

dir_path = os.path.dirname(__file__)

index_fn = os.path.join(dir_path, 'devhelp_qch_data/devhelp-index.xml')

file_list_fn = os.path.join(dir_path, 'devhelp_qch_data/file-list.xml')

expected_path = os.path.join(dir_path, 'devhelp_qch_data/expected.xml')

dest_path = os.path.join(dir_path, 'devhelp_qch_data/result.xml')

parser = etree.XMLParser(encoding='UTF-8', recover=True)

in_tree = etree.parse(index_fn, parser)

in_root = in_tree.getroot()

file_tree = etree.parse(file_list_fn, parser)

files_root = file_tree.getroot()

with open(expected_path, 'rb') as expected_f:

expected = expected_f.read()

result = convert_devhelp_to_qch(in_root, files_root, 'virtual_folder')

if expected != result:

with open(dest_path, 'wb') as result_f:

result_f.write(result)

self.assertEqual(expected, result)

Provide feedback