[#85251] Check if required directories exist

Signed-off-by: Maciej Sobkowski <msobkowski@antmicro.com>

Author: msobkowski

protoplaster/protoplaster.py

@@ -44,6 +44,11 @@ def parse_args():
                         type=str,
                         default=f"{ARTIFACTS_DIR}",
                         help="Path to the test artifacts directory")
+    parser.add_argument(
+        "-m",
+        "--mkdir",
+        action="store_true",
+        help="Try to create test/reports/artifacts directories if missing")
     parser.add_argument(
         "-t",
         "--test-file",
@@ -133,6 +138,29 @@ def main():
     if args.sudo:
         os.execv(shutil.which("sudo"),
                  [__file__] + list(filter(lambda a: a != "--sudo", sys.argv)))
+
+    if not os.path.exists(args.test_dir):
+        if args.mkdir:
+            os.makedirs(args.test_dir, exist_ok=True)
+        else:
+            print(error(f"Tests dir ({args.test_dir}) does not exist!"))
+            exit(1)
+
+    if not os.path.exists(args.reports_dir):
+        if args.mkdir:
+            os.makedirs(args.reports_dir, exist_ok=True)
+        else:
+            print(error(f"Reports dir ({args.reports_dir}) does not exist!"))
+            exit(1)
+
+    if not os.path.exists(args.artifacts_dir):
+        if args.mkdir:
+            os.makedirs(args.artifacts_dir, exist_ok=True)
+        else:
+            print(
+                error(f"Artifacts dir ({args.artifacts_dir}) does not exist!"))
+            exit(1)
+
     if not os.path.exists(
             f"{args.test_dir}/{args.test_file}") and not args.server:
         print(

protoplaster/runner/runner.py

@@ -157,8 +157,6 @@ def run_tests(args):
     test_modules = test_file.list_test_modules()
     metadata_cmds = test_file.list_metadata_commands()
 
-    os.makedirs(args.artifacts_dir, exist_ok=True)
-
     if metadata_cmds:
         metadata = [
             result.output_file

protoplaster/runner/worker.py

@@ -27,6 +27,13 @@ def run_test(run, base_args):
     args.csv = run["id"] + ".csv"
     args.artifacts_dir = os.path.join(args.artifacts_dir, run["id"])
 
+    os.makedirs(args.artifacts_dir, exist_ok=True)
+
+    if not (os.path.exists(args.test_dir) or os.path.exists(args.reports_dir)
+            or os.path.exists(args.reports_dir)):
+        run["status"] = RunStatus.FAILED
+        return
+
     ret, metadata = run_tests(args)
 
     run["metadata"] = {