Source code for daily_med.tests.test_spl_document

# -*- coding: utf-8 -*-
# Copyright 2017-TODAY LasLabs Inc.
# License MIT (https://opensource.org/licenses/MIT).

import os
import unittest

from ..models.spl_document import parse, parseString, get_all_values


class TestSPLDocument(unittest.TestCase):

    TEST_FILES = [
        'spl_doc_1.xml',
        'spl_doc_1.xml',
    ]
    TEST_DIR = os.path.abspath(os.path.dirname(__file__))

    def _parse_file(self, test_file, as_string=True):
        test_file = os.path.join(self.TEST_DIR, test_file)
        if not as_string:
            return parse(test_file, True)
        with open(test_file, 'r') as fh:
            return parseString(fh.read(), True)

    def test_parse(self):
        """ It should parse the sample files without errors. """
        for test_file in self.TEST_FILES:
            spl_doc = self._parse_file(test_file, as_string=False)
            self.assertTrue(spl_doc.id.root)
            self.assertTrue(spl_doc.title)
            self.assertTrue(spl_doc.code.code)

    def test_parse_string(self):
        """ It should parse the sample files as string without errors. """
        for test_file in self.TEST_FILES:
            spl_doc = self._parse_file(test_file, as_string=True)
            self.assertTrue(spl_doc.id.root)
            self.assertTrue(spl_doc.title)
            self.assertTrue(spl_doc.code.code)

    def test_get_all_values(self):
        """ It should return a list of the node values. """
        spl_doc = self._parse_file(self.TEST_FILES[0])
        expect = ['RENESE', '\n\t\t', '\n\t\t', '(polythiazide)',
                  '\n\t\t', 'TABLETS', '\n\t\t', 'for Oral Administration']
        self.assertEqual(
            get_all_values(spl_doc.title.content_), expect,
        )