sh_cpl/unittests/unittests_cli/new_test_case.py

import json
import os
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_shared.cli_commands import CLICommands


class NewTestCase(CommandTestCase):

    def __init__(self, method_name: str):
        CommandTestCase.__init__(self, method_name)

    def _test_project(self, project_type: str, name: str, *args, test_venv=False, without_ws=False):
        CLICommands.new(project_type, name, *args, output=False)
        workspace_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, name))
        self.assertTrue(os.path.exists(workspace_path))
        if test_venv:
            with self.subTest(msg='Venv exists'):
                self.assertTrue(os.path.exists(os.path.join(workspace_path, 'venv')))
                self.assertTrue(os.path.exists(os.path.join(workspace_path, 'venv/bin')))
                self.assertTrue(os.path.exists(os.path.join(workspace_path, 'venv/bin/python')))
                self.assertTrue(os.path.islink(os.path.join(workspace_path, 'venv/bin/python')))

        base = 'src'
        if '--base' in args and '/' in name:
            base = name.split('/')[0]
            name = name.replace(f'{name.split("/")[0]}/', '')

        project_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, name, base, String.convert_to_snake_case(name)))
        if without_ws:
            project_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, base, name, 'src/', String.convert_to_snake_case(name)))

        with self.subTest(msg='Project json exists'):
            self.assertTrue(os.path.exists(project_path))
            self.assertTrue(os.path.exists(os.path.join(project_path, f'{name}.json')))

        if project_type == 'library':
            with self.subTest(msg='Library class1 exists'):
                self.assertTrue(os.path.exists(os.path.join(project_path, f'class1.py')))
            return

        with self.subTest(msg='Project main.py exists'):
            self.assertTrue(os.path.exists(os.path.join(project_path, f'main.py')))

        with self.subTest(msg='Application base'):
            if '--ab' in args:
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'application.py')))
            else:
                self.assertFalse(os.path.isfile(os.path.join(project_path, f'application.py')))

        # s depends on ab
        with self.subTest(msg='Startup'):
            if '--ab' in args and '--s' in args:
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'startup.py')))
            else:
                self.assertFalse(os.path.isfile(os.path.join(project_path, f'startup.py')))

        with self.subTest(msg='Unittest'):
            if project_type == 'unittest':
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'test_case.py')))
            else:
                self.assertFalse(os.path.isfile(os.path.join(project_path, f'test_case.py')))

        with self.subTest(msg='Discord'):
            if project_type == 'discord-bot':
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'events/__init__.py')))
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'events/on_ready_event.py')))
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'commands/__init__.py')))
                self.assertTrue(os.path.isfile(os.path.join(project_path, f'commands/ping_command.py')))
            else:
                self.assertFalse(os.path.isfile(os.path.join(project_path, f'events/on_ready_event.py')))
                self.assertFalse(os.path.isfile(os.path.join(project_path, f'commands/ping_command.py')))

    def _test_sub_project(self, project_type: str, name: str, workspace_name: str, *args, test_venv=False):
        os.chdir(os.path.abspath(os.path.join(os.getcwd(), workspace_name)))
        CLICommands.new(project_type, name, *args)
        workspace_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, workspace_name))
        self.assertTrue(os.path.exists(workspace_path))
        if test_venv:
            self.assertTrue(os.path.exists(os.path.join(workspace_path, 'venv')))
            self.assertTrue(os.path.exists(os.path.join(workspace_path, 'venv/bin')))
            self.assertTrue(os.path.exists(os.path.join(workspace_path, 'venv/bin/python')))
            self.assertTrue(os.path.islink(os.path.join(workspace_path, 'venv/bin/python')))

        base = 'src'
        if '--base' in args and '/' in name:
            base = name.split('/')[0]
            name = name.replace(f'{name.split("/")[0]}/', '')

        project_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, workspace_name, base, String.convert_to_snake_case(name)))
        self.assertTrue(os.path.exists(project_path))
        self.assertTrue(os.path.join(project_path, f'{name}.json'))

        if project_type == 'discord-bot':
            self.assertTrue(os.path.isfile(os.path.join(project_path, f'events/__init__.py')))
            self.assertTrue(os.path.isfile(os.path.join(project_path, f'events/on_ready_event.py')))
            self.assertTrue(os.path.isfile(os.path.join(project_path, f'commands/__init__.py')))
            self.assertTrue(os.path.isfile(os.path.join(project_path, f'commands/ping_command.py')))
        else:
            self.assertFalse(os.path.isfile(os.path.join(project_path, f'events/on_ready_event.py')))
            self.assertFalse(os.path.isfile(os.path.join(project_path, f'commands/ping_command.py')))

    def _test_sub_directory_project(self, project_type: str, directory: str, name: str, workspace_name: str, *args):
        os.chdir(os.path.abspath(os.path.join(os.getcwd(), workspace_name)))
        CLICommands.new(project_type, f'{directory}/{name}', *args)
        workspace_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, workspace_name))
        self.assertTrue(os.path.exists(workspace_path))

        project_path = os.path.abspath(os.path.join(PLAYGROUND_PATH, workspace_name, f'src/{directory}', String.convert_to_snake_case(name)))
        self.assertTrue(os.path.exists(project_path))
        project_file = os.path.join(project_path, f'{name}.json')
        self.assertTrue(os.path.exists(project_file))
        with open(project_file, 'r', encoding='utf-8') as cfg:
            # load json
            project_json = json.load(cfg)
            cfg.close()

        project_settings = project_json['ProjectSettings']
        build_settings = project_json['BuildSettings']

        self.assertEqual(project_settings['Name'], name)
        self.assertEqual(build_settings['ProjectType'], 'library')
        self.assertEqual(build_settings['OutputPath'], '../../dist')
        self.assertEqual(build_settings['Main'], f'{String.convert_to_snake_case(name)}.main')
        self.assertEqual(build_settings['EntryPoint'], name)

    def test_console(self):
        self._test_project('console', 'test-console', '--ab', '--s', '--venv', test_venv=True)

    def test_console_with_other_base(self):
        self._test_project('console', 'tools/test-console', '--ab', '--s', '--venv', '--base', test_venv=True, without_ws=True)

    def test_console_without_s(self):
        self._test_project('console', 'test-console-without-s', '--ab')

    def test_console_without_ab(self):
        self._test_project('console', 'test-console-without-ab', '--sp')

    def test_console_without_anything(self):
        self._test_project('console', 'test-console-without-anything', '--n')

    def test_console_sub(self):
        self._test_sub_project('console', 'test-sub-console', 'test-console', '--ab', '--s', '--sp', '--venv', test_venv=True)

    def test_console_sub_with_other_base(self):
        self._test_sub_project('console', 'tools/test-sub-console', 'test-console', '--ab', '--s', '--sp', '--venv', '--base', test_venv=True)

    def test_discord_bot(self):
        self._test_project('discord-bot', 'test-bot', '--ab', '--s', '--venv', test_venv=True)

    def test_discord_bot_sub(self):
        self._test_sub_project('discord-bot', 'test-bot-sub', 'test-console', '--ab', '--s', '--venv', test_venv=True)

    def test_library(self):
        self._test_project('library', 'test-library', '--ab', '--s', '--sp')

    def test_library_sub(self):
        self._test_sub_project('library', 'test-sub-library', 'test-console', '--ab', '--s', '--sp')

    def test_library_sub_directory(self):
        self._test_sub_directory_project('library', 'directory', 'test-sub-library', 'test-console', '--ab', '--s', '--sp')

    def test_unittest(self):
        self._test_project('unittest', 'test-unittest', '--ab')

    def test_unittest_sub(self):
        self._test_sub_project('unittest', 'test-unittest', 'test-console', '--ab', '--s', '--sp')