Home | History | Annotate | Download | only in OCaml
      1 (* RUN: rm -rf %t && mkdir -p %t && cp %s %t/linker.ml
      2  * RUN: %ocamlc -g -w +A -package llvm.linker -linkpkg %t/linker.ml -o %t/executable
      3  * RUN: %t/executable
      4  * RUN: %ocamlopt -g -w +A -package llvm.linker -linkpkg %t/linker.ml -o %t/executable
      5  * RUN: %t/executable
      6  * XFAIL: vg_leak
      7  *)
      8 
      9 (* Note: It takes several seconds for ocamlopt to link an executable with
     10          libLLVMCore.a, so it's better to write a big test than a bunch of
     11          little ones. *)
     12 
     13 open Llvm
     14 open Llvm_linker
     15 
     16 let context = global_context ()
     17 let void_type = Llvm.void_type context
     18 
     19 let diagnostic_handler _ = ()
     20 
     21 (* Tiny unit test framework - really just to help find which line is busted *)
     22 let print_checkpoints = false
     23 
     24 let suite name f =
     25   if print_checkpoints then
     26     prerr_endline (name ^ ":");
     27   f ()
     28 
     29 
     30 (*===-- Linker -----------------------------------------------------------===*)
     31 
     32 let test_linker () =
     33   set_diagnostic_handler context (Some diagnostic_handler);
     34 
     35   let fty = function_type void_type [| |] in
     36 
     37   let make_module name =
     38     let m = create_module context name in
     39     let fn = define_function ("fn_" ^ name) fty m in
     40     ignore (build_ret_void (builder_at_end context (entry_block fn)));
     41     m
     42   in
     43 
     44   let m1 = make_module "one"
     45   and m2 = make_module "two" in
     46   link_modules' m1 m2;
     47   dispose_module m1;
     48 
     49   let m1 = make_module "one"
     50   and m2 = make_module "two" in
     51   link_modules' m1 m2;
     52   dispose_module m1;
     53 
     54   let m1 = make_module "one"
     55   and m2 = make_module "one" in
     56   try
     57     link_modules' m1 m2;
     58     failwith "must raise"
     59   with Error _ ->
     60     dispose_module m1
     61 
     62 (*===-- Driver ------------------------------------------------------------===*)
     63 
     64 let _ =
     65   suite "linker" test_linker
     66