sh_cpl/unittests/unittests_cli/start_test_case.py

import json
import os
import shutil
import time
import unittest

from cpl_core.utils import String
from unittests_cli.abc.command_test_case import CommandTestCase
from unittests_cli.constants import PLAYGROUND_PATH
from unittests_cli.threads.start_test_thread import StartTestThread
from unittests_shared.cli_commands import CLICommands


class StartTestCase(CommandTestCase):
    def __init__(self, method_name: str):
        CommandTestCase.__init__(self, method_name)
        self._source = "start-test"
        self._project_file = f"src/{String.convert_to_snake_case(self._source)}/{self._source}.json"
        self._appsettings = f"src/{String.convert_to_snake_case(self._source)}/appsettings.json"
        self._application = f"src/{String.convert_to_snake_case(self._source)}/application.py"
        self._test_code = f"""
        import json
        import os
        settings = dict()
        with open('appsettings.json', 'r', encoding='utf-8') as cfg:
            # load json
            settings = json.load(cfg)
            cfg.close()
        
        if settings['RunTest']['WasStarted'] == 'True':
            settings['RunTest']['WasRestarted'] = 'True'
        
        settings['RunTest']['WasStarted'] = 'True'
        settings['RunTest']['Path'] = os.path.dirname(os.path.realpath(__file__))
        
        with open('appsettings.json', 'w', encoding='utf-8') as project_file:
            project_file.write(json.dumps(settings, indent=2))
            project_file.close()
        """

    def _get_appsettings(self, is_dev=False):
        appsettings = f"dist/{self._source}/build/{String.convert_to_snake_case(self._source)}/appsettings.json"
        if is_dev:
            appsettings = f"src/{String.convert_to_snake_case(self._source)}/appsettings.json"

        with open(os.path.join(os.getcwd(), appsettings), "r", encoding="utf-8") as cfg:
            # load json
            project_json = json.load(cfg)
            cfg.close()

        return project_json

    def _save_appsettings(self, settings: dict):
        with open(
            os.path.join(os.getcwd(), f"src/{String.convert_to_snake_case(self._source)}/appsettings.json"),
            "w",
            encoding="utf-8",
        ) as project_file:
            project_file.write(json.dumps(settings, indent=2))
            project_file.close()

    def setUp(self):
        if not os.path.exists(PLAYGROUND_PATH):
            os.makedirs(PLAYGROUND_PATH)

        os.chdir(PLAYGROUND_PATH)
        # create projects
        CLICommands.new("console", self._source, "--ab", "--s")
        os.chdir(os.path.join(os.getcwd(), self._source))
        settings = {"RunTest": {"WasStarted": "False", "WasRestarted": "False"}}
        self._save_appsettings(settings)
        with open(os.path.join(os.getcwd(), self._application), "a", encoding="utf-8") as file:
            file.write(f"\t\t{self._test_code}")
            file.close()

    def test_start(self):
        thread = StartTestThread()
        thread.start()
        time.sleep(5)
        settings = self._get_appsettings()
        self.assertNotEqual(settings, {})
        self.assertIn("RunTest", settings)
        self.assertIn("WasStarted", settings["RunTest"])
        self.assertEqual("True", settings["RunTest"]["WasStarted"])

        with open(os.path.join(os.getcwd(), self._application), "a", encoding="utf-8") as file:
            file.write(f"# trigger restart (comment generated by unittest)")
            file.close()

        time.sleep(5)

        settings = self._get_appsettings()
        self.assertNotEqual(settings, {})
        self.assertIn("RunTest", settings)
        self.assertIn("WasStarted", settings["RunTest"])
        self.assertIn("WasRestarted", settings["RunTest"])
        self.assertEqual("True", settings["RunTest"]["WasStarted"])
        self.assertEqual("True", settings["RunTest"]["WasRestarted"])

    def test_start_dev(self):
        thread = StartTestThread(is_dev=True)
        thread.start()
        time.sleep(1)
        settings = self._get_appsettings()
        self.assertNotEqual(settings, {})
        self.assertIn("RunTest", settings)
        self.assertIn("WasStarted", settings["RunTest"])
        self.assertEqual("True", settings["RunTest"]["WasStarted"])

        with open(os.path.join(os.getcwd(), self._application), "a", encoding="utf-8") as file:
            file.write(f"# trigger restart (comment generated by unittest)")
            file.close()

        time.sleep(1)

        settings = self._get_appsettings(is_dev=True)
        self.assertNotEqual(settings, {})
        self.assertIn("RunTest", settings)
        self.assertIn("WasStarted", settings["RunTest"])
        self.assertIn("WasRestarted", settings["RunTest"])
        self.assertEqual("True", settings["RunTest"]["WasStarted"])
        self.assertEqual("True", settings["RunTest"]["WasRestarted"])