def dry_run_workload(yaml_file_path: str, is_darwin: bool,
genny_repo_root: str, workspace_root: str):
if os.path.basename(yaml_file_path) in WIP_YML_FILES:
SLOG.info("Skipping dry run for workloads for future functionality.",
file=yaml_file_path)
return
if os.path.basename(yaml_file_path) in [
"MixedWorkloadsGennyStress.yml",
]:
SLOG.info(
"TIG-3290 skipping dry run for MixedWorkloadsGennyStress.yml.",
file=yaml_file_path)
return
if is_darwin and os.path.basename(yaml_file_path) in [
"AuthNInsert.yml",
"ParallelInsert.yml",
"LoggingActorExample.yml",
]:
SLOG.info("TIG-1435 skipping dry run on macOS", file=yaml_file_path)
return
genny_runner.main_genny_runner(
genny_args=["dry-run", yaml_file_path],
genny_repo_root=genny_repo_root,
workspace_root=workspace_root,
cleanup_metrics=True,
)
def dry_run_workload(yaml_file_path: str, is_darwin: bool,
genny_repo_root: str, workspace_root: str):
if os.path.basename(yaml_file_path) in WIP_YML_FILES:
SLOG.info("Skipping dry run for workloads for future functionality.",
file=yaml_file_path)
return
if os.path.basename(yaml_file_path) in [
"MixedWorkloadsGennyStress.yml",
]:
SLOG.info(
"TIG-3290 skipping dry run for MixedWorkloadsGennyStress.yml.",
file=yaml_file_path)
return
if is_darwin and os.path.basename(yaml_file_path) in [
"AuthNInsert.yml",
"ParallelInsert.yml",
"LoggingActorExample.yml",
]:
SLOG.info("TIG-1435 skipping dry run on macOS", file=yaml_file_path)
return
genny_runner.main_genny_runner(
workload_yaml_path=yaml_file_path,
mongo_uri="mongodb://localhost:27017",
verbosity="info",
override=None,
dry_run=True,
smoke_test=False,
genny_repo_root=genny_repo_root,
workspace_root=workspace_root,
cleanup_metrics=True,
)
def workload(
ctx: click.Context,
workload_yaml: str,
mongo_uri: str,
verbosity: str,
override: str,
dry_run: bool,
smoke_test: bool,
debug: bool,
):
from genny.tasks import genny_runner
ctx.ensure_object(dict)
genny_runner.main_genny_runner(
workload_yaml_path=workload_yaml[0],
mongo_uri=mongo_uri,
verbosity=verbosity,
override=override,
dry_run=dry_run,
smoke_test=smoke_test,
genny_repo_root=ctx.obj["GENNY_REPO_ROOT"],
workspace_root=ctx.obj["WORKSPACE_ROOT"],
cleanup_metrics=True,
hang=debug,
)
def workload(ctx: click.Context, genny_args: List[str]):
from genny.tasks import genny_runner
ctx.ensure_object(dict)
ctx.obj["GENNY_ARGS"] = genny_args
genny_runner.main_genny_runner(
genny_args=ctx.obj["GENNY_ARGS"],
genny_repo_root=ctx.obj["GENNY_REPO_ROOT"],
workspace_root=ctx.obj["WORKSPACE_ROOT"],
cleanup_metrics=True,
)
def dry_run_workload(yaml_file_path: str, is_darwin: bool,
genny_repo_root: str, workspace_root: str):
if is_darwin and os.path.basename(yaml_file_path) in [
"AuthNInsert.yml",
"ParallelInsert.yml",
"LoggingActorExample.yml",
]:
SLOG.info("TIG-1435 skipping dry run on macOS", file=yaml_file_path)
return
genny_runner.main_genny_runner(
genny_args=["dry-run", yaml_file_path],
genny_repo_root=genny_repo_root,
workspace_root=workspace_root,
cleanup_metrics=True,
)