import unittest
import importlib
utils = importlib.import_module('extensions.sd-webui-controlnet.tests.utils', 'utils')
utils.setup_test_env()

from scripts import external_code


class TestImportlibReload(unittest.TestCase):
    def setUp(self):
        self.ControlNetUnit = external_code.ControlNetUnit

    def test_reload_does_not_redefine(self):
        importlib.reload(external_code)
        NewControlNetUnit = external_code.ControlNetUnit
        self.assertEqual(self.ControlNetUnit, NewControlNetUnit)

    def test_force_import_does_not_redefine(self):
        external_code_copy = importlib.import_module('extensions.sd-webui-controlnet.scripts.external_code', 'external_code')
        self.assertEqual(self.ControlNetUnit, external_code_copy.ControlNetUnit)


if __name__ == '__main__':
    unittest.main()