def test_validate_application_files__fails_if_application_directory_does_not_exist(
tmp_path):
application_path = tmp_path / "dummy"
match_pattern = re.compile(
f"application files in {application_path} were invalid.*directory {application_path} does not exist",
flags=re.DOTALL,
)
with pytest.raises(Abort, match=match_pattern):
validate_application_files(application_path)
def test_validate_application_files__fails_if_application_config_does_not_exist(
tmp_path):
application_path = tmp_path / "dummy"
application_path.mkdir()
match_pattern = re.compile(
f"application files in {application_path} were invalid.*does not contain required configuration file",
flags=re.DOTALL,
)
with pytest.raises(Abort, match=match_pattern):
validate_application_files(application_path)
def test_validate_application_files__fails_if_application_module_is_not_valid_python(
tmp_path):
application_path = tmp_path / "dummy"
application_path.mkdir()
application_module = application_path / JOBBERGATE_APPLICATION_MODULE_FILE_NAME
application_module.write_text("[")
match_pattern = re.compile(
f"application files in {application_path} were invalid.*not valid python",
flags=re.DOTALL,
)
with pytest.raises(Abort, match=match_pattern):
validate_application_files(application_path)
def test_validate_application_files__fails_if_application_config_is_not_valid_yaml(
tmp_path):
application_path = tmp_path / "dummy"
application_path.mkdir()
application_config = application_path / JOBBERGATE_APPLICATION_CONFIG_FILE_NAME
application_config.write_text(":")
match_pattern = re.compile(
f"application files in {application_path} were invalid.*not valid YAML",
flags=re.DOTALL,
)
with pytest.raises(Abort, match=match_pattern):
validate_application_files(application_path)
def test_validate_application_files__success(tmp_path):
application_path = tmp_path / "dummy"
application_path.mkdir()
application_module = application_path / JOBBERGATE_APPLICATION_MODULE_FILE_NAME
application_module.write_text(
dedent("""
import sys
print(f"Got some args, yo: {sys.argv}")
"""))
application_config = application_path / JOBBERGATE_APPLICATION_CONFIG_FILE_NAME
application_config.write_text(
dedent("""
foo:
bar: baz
"""))
validate_application_files(application_path)
def update(
ctx: typer.Context,
id: int = typer.Option(
...,
help=f"The specific id of the application to update. {ID_NOTE}",
),
application_path: Optional[pathlib.Path] = typer.Option(
None,
help="The path to the directory where the application files are located",
),
identifier: Optional[str] = typer.Option(
None,
help="Optional new application identifier to be set",
),
application_desc: Optional[str] = typer.Option(
None,
help="Optional new application description to be set",
),
):
"""
Update an existing application.
"""
req_data = dict()
if identifier:
req_data["application_identifier"] = identifier
if application_desc:
req_data["application_description"] = application_desc
if application_path is not None:
validate_application_files(application_path)
req_data["application_config"] = dump_full_config(application_path)
req_data["application_file"] = read_application_module(application_path)
jg_ctx: JobbergateContext = ctx.obj
# Make static type checkers happy
assert jg_ctx.client is not None
result = cast(
Dict[str, Any],
make_request(
jg_ctx.client,
f"/applications/{id}",
"PUT",
expected_status=200,
abort_message="Request to update application was not accepted by the API",
support=True,
json=req_data,
),
)
if application_path is not None:
successful_upload = _upload_application(jg_ctx, application_path, id)
if not successful_upload:
terminal_message(
f"""
The zipped application files could not be uploaded.
[yellow]If the problem persists, please contact [bold]{OV_CONTACT}[/bold]
for support and trouble-shooting[/yellow]
""",
subject="File upload failed",
color="yellow",
)
else:
result["application_uploaded"] = True
render_single_result(
jg_ctx,
result,
hidden_fields=HIDDEN_FIELDS,
title="Updated Application",
)
def create(
ctx: typer.Context,
name: str = typer.Option(
...,
help="The name of the application to create",
),
identifier: Optional[str] = typer.Option(
None,
help=f"The human-friendly identifier of the application. {IDENTIFIER_NOTE}",
),
application_path: pathlib.Path = typer.Option(
...,
help="The path to the directory where the application files are located",
),
application_desc: Optional[str] = typer.Option(
None,
help="A helpful description of the application",
),
):
"""
Create a new application.
"""
req_data = load_default_config()
req_data["application_name"] = name
if identifier:
req_data["application_identifier"] = identifier
if application_desc:
req_data["application_description"] = application_desc
validate_application_files(application_path)
req_data["application_config"] = dump_full_config(application_path)
req_data["application_file"] = read_application_module(application_path)
jg_ctx: JobbergateContext = ctx.obj
# Make static type checkers happy
assert jg_ctx.client is not None
result = cast(
Dict[str, Any],
make_request(
jg_ctx.client,
"/applications",
"POST",
expected_status=201,
abort_message="Request to create application was not accepted by the API",
support=True,
json=req_data,
),
)
application_id = result["id"]
successful_upload = upload_application(jg_ctx, application_path, application_id)
if not successful_upload:
terminal_message(
f"""
The zipped application files could not be uploaded.
Try running the `update` command including the application path to re-upload.
[yellow]If the problem persists, please contact [bold]{OV_CONTACT}[/bold]
for support and trouble-shooting[/yellow]
""",
subject="File upload failed",
color="yellow",
)
else:
result["application_uploaded"] = True
render_single_result(
jg_ctx,
result,
hidden_fields=HIDDEN_FIELDS,
title="Created Application",
)