ocamltest: implement a copy action

Author: shindere

ocamltest/builtin_actions.ml

@@ -238,6 +238,34 @@ let file_exists_action _log env =
       end
 let file_exists = make "file-exists" file_exists_action
 
+let copy_action log env =
+  let do_copy src dst =
+    let (entry_type, f) =
+      if Sys.is_directory src
+      then ("directory", Sys.copy_directory)
+      else ("file", Sys.copy_file)
+    in
+    Printf.fprintf log "Copying %s %s to %s\n%!" entry_type src dst;
+    f src dst
+  in
+  let src = Environments.lookup Builtin_variables.src env in
+  let dst = Environments.lookup Builtin_variables.dst env in
+  match (src, dst) with
+    | (None, _) | (_, None) ->
+      let reason = reason_with_fallback env "src or dst are undefined" in
+      let result = Result.fail_with_reason reason in
+      (result, env)
+    | (Some src, Some dst) ->
+      let f =
+        if String.ends_with ~suffix:"/" dst
+        then fun src -> do_copy src (dst ^ (Filename.basename src))
+        else fun src -> do_copy src dst
+      in
+      List.iter f (String.words src);
+      (Result.pass, env)
+
+let copy = make "copy" copy_action
+
 let initialize_test_exit_status_variables _log env =
   Environments.add_bindings
   [
@@ -281,4 +309,5 @@ let _ =
     function_sections;
     naked_pointers;
     file_exists;
+    copy;
   ]

ocamltest/builtin_actions.mli

@@ -49,3 +49,5 @@ val script : Actions.t
 val check_program_output : Actions.t
 
 val file_exists : Actions.t
+
+val copy : Actions.t

ocamltest/builtin_variables.ml

@@ -31,6 +31,8 @@ let cwd = Variables.make ("cwd",
 let commandline = Variables.make ("commandline",
   "Specify the commandline of a tool")
 
+let dst = Variables.make ("dst", "Location where to copy files and directories")
+
 let exit_status = Variables.make ("exit_status",
   "Expected program exit status")
 
@@ -79,6 +81,8 @@ let skip_header_bytes =
 let script = Variables.make ("script",
   "External script to run")
 
+let src = Variables.make ("src", "Files and directories to copy")
+
 let stdin = Variables.make ("stdin", "Default standard input")
 let stdout = Variables.make ("stdout", "Default standard output")
 let stderr = Variables.make ("stderr", "Default standard error")
@@ -115,6 +119,7 @@ let _ = List.iter Variables.register_variable
     arguments;
     cwd;
     commandline;
+    dst;
     exit_status;
     file;
     readonly_files;
@@ -125,6 +130,7 @@ let _ = List.iter Variables.register_variable
     program; program2;
     reason;
     reference;
+    src;
     skip_header_lines;
     skip_header_bytes;
     script;

ocamltest/builtin_variables.mli

@@ -23,6 +23,8 @@ val cwd : Variables.t
 
 val commandline : Variables.t
 
+val dst : Variables.t
+
 val exit_status : Variables.t
 
 val file : Variables.t
@@ -51,6 +53,8 @@ val skip_header_bytes : Variables.t
 
 val script : Variables.t
 
+val src : Variables.t
+
 val stdin : Variables.t
 val stdout : Variables.t
 val stderr : Variables.t