122 lines
2.9 KiB
Python
122 lines
2.9 KiB
Python
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License").
|
|
# You may not use this file except in compliance with the License.
|
|
# A copy of the License is located at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# or in the "license" file accompanying this file. This file is distributed
|
|
# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
|
|
# express or implied. See the License for the specific language governing
|
|
# permissions and limitations under the License.
|
|
|
|
|
|
import json
|
|
import os
|
|
import pathlib
|
|
import sys
|
|
|
|
|
|
def get_init_args():
|
|
return {
|
|
"kwargs": {
|
|
"project": "foo",
|
|
}
|
|
}
|
|
|
|
|
|
def get_jobs():
|
|
litani_path = pathlib.Path(os.environ["LITANI_E2E_LITANI_PATH"])
|
|
if not litani_path.exists():
|
|
raise UserWarning("Could not find litani executable")
|
|
|
|
return [{
|
|
"kwargs": {
|
|
"command": f"sleep 2",
|
|
"description": "SLEEP",
|
|
"outputs": "sleep-output",
|
|
"ci-stage": "build",
|
|
"pipeline": "foo",
|
|
}
|
|
}, {
|
|
"kwargs": {
|
|
"command": f"{litani_path.resolve()} dump-run",
|
|
"description": "DUMP",
|
|
"inputs": "sleep-output",
|
|
"timeout": 20,
|
|
"ci-stage": "build",
|
|
"pipeline": "foo",
|
|
}
|
|
}, {
|
|
"kwargs": {
|
|
"command": "sleep 10",
|
|
"description": "LONG",
|
|
"ci-stage": "build",
|
|
"pipeline": "foo",
|
|
}
|
|
}]
|
|
|
|
|
|
def get_run_build_args():
|
|
return {}
|
|
|
|
|
|
def perror(*args, **kwargs):
|
|
kwargs = {
|
|
**kwargs, "file": sys.stderr
|
|
}
|
|
print(*args, **kwargs)
|
|
|
|
|
|
def check_run(run):
|
|
pipe = run["pipelines"][0]
|
|
jobs = pipe["ci_stages"][0]["jobs"]
|
|
dump_job = [
|
|
j for j in jobs
|
|
if j["wrapper_arguments"]["description"] == "DUMP"
|
|
]
|
|
if not dump_job:
|
|
perror("Dump job doesn't exist")
|
|
return False
|
|
|
|
run_dump = "\n".join(dump_job[0]["stdout"])
|
|
try:
|
|
run_so_far = json.loads(run_dump)
|
|
except json.decoder.JSONDecodeError:
|
|
perror("JSON decode")
|
|
perror(run_dump)
|
|
return False
|
|
|
|
pipe_so_far = run_so_far["pipelines"][0]
|
|
jobs = pipe_so_far["ci_stages"][0]["jobs"]
|
|
|
|
sleep_job = [
|
|
j for j in jobs
|
|
if j["wrapper_arguments"]["description"] == "SLEEP"
|
|
]
|
|
if not sleep_job:
|
|
perror("sleep job doesn't exist")
|
|
return False
|
|
|
|
if not all((
|
|
sleep_job[0]["complete"],
|
|
sleep_job[0]["duration"] > 1,
|
|
)):
|
|
perror("sleep job format")
|
|
return False
|
|
|
|
long_job = [
|
|
j for j in jobs
|
|
if j["wrapper_arguments"]["description"] == "LONG"
|
|
]
|
|
if not long_job:
|
|
perror("long job doesn't exist")
|
|
return False
|
|
|
|
if long_job[0]["complete"]:
|
|
perror("long job already complete before dumping run")
|
|
return False
|
|
|
|
return True
|