import unittest

import sys

from zensols.cli import CliHarness, ApplicationFailure

from zensols.config import ConfigFactory

from zensols.mednlp import Application, ApplicationFactory, surpress_warnings

class TestBase(unittest.TestCase):

    def setUp(self):

        import warnings

        warnings.simplefilter("ignore", ResourceWarning)

        surpress_warnings()

        self.text_1 = 'He was diagnosed with kidney failure and heart disease.'

        self.text_2 = 'He loved to smoke but Marlboro cigarettes gave John Smith lung cancer while he was in Chicago.'

        self.maxDiff = sys.maxsize

    def _get_doc_parser(self, config: str = 'default', section: str = None):

        harness: CliHarness = ApplicationFactory.create_harness()

        args: str = f'--config test-resources/config/{config}.conf --level=err'

        if section is None:

            app: Application = harness.get_instance(f'show _ {args}')

            if isinstance(app, ApplicationFailure):

                raise app.exception

            return app.doc_parser

        else:

            harness: CliHarness = ApplicationFactory.create_harness()

            fac: ConfigFactory = harness.get_config_factory(args)

            return fac(section)