sh_cpl/unittests/unittests_shared/cli_commands.py

88 lines
2.5 KiB
Python

import os
import subprocess
from unittests_cli.constants import CLI_PATH
class CLICommands:
@staticmethod
def _run(cmd: str, *args, output=False):
env_vars = os.environ
env_vars['CPL_IS_UNITTEST'] = 'NO' if output else 'YES'
command = ['python', CLI_PATH, cmd]
for arg in args:
command.append(arg)
if output:
subprocess.run(command, env=env_vars)
else:
subprocess.run(command, env=env_vars, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, stdin=subprocess.DEVNULL)
@staticmethod
def _run_with_output(cmd: str, *args) -> str:
env_vars = os.environ
# env_vars['CPL_IS_UNITTEST'] = 'NO'
command = ['python', CLI_PATH, cmd]
for arg in args:
command.append(arg)
return subprocess.run(command, env=env_vars, check=True, capture_output=True, text=True).stdout
@classmethod
def add(cls, source: str, target: str, output=False):
cls._run('add', source, target, output=output)
@classmethod
def build(cls, output=False):
cls._run('build', output=output)
@classmethod
def generate(cls, schematic: str, name: str, output=False):
cls._run('generate', schematic, name, output=output)
@classmethod
def install(cls, package: str = None, is_dev=False, output=False):
if package is None:
cls._run('install', output=output)
return
cls._run('install', package, '--dev' if is_dev else '', output=output)
@classmethod
def new(cls, project_type: str, name: str, *args, output=False):
cls._run('new', project_type, name, *args, output=output)
@classmethod
def publish(cls, output=False):
cls._run('publish', output=output)
@classmethod
def remove(cls, project: str, output=False):
cls._run('remove', project, output=output)
@classmethod
def run(cls, project: str = None, output=False):
if project is None:
cls._run('run', output=output)
return
cls._run('run', project, output=output)
@classmethod
def start(cls, output=False):
cls._run('start', output=output)
@classmethod
def uninstall(cls, package: str, is_dev=False, output=False):
cls._run('uninstall', package, '--dev' if is_dev else '', output=output)
@classmethod
def update(cls, output=False):
cls._run('update', output=output)
@classmethod
def version(cls) -> str:
return cls._run_with_output('version')