TheRock/compiler/amd-llvm/llvm/test/Bindings/OCaml/linker.ml

(* RUN: rm -rf %t && mkdir -p %t && cp %s %t/linker.ml
 * RUN: %ocamlc -g -w +A -package llvm.linker -linkpkg %t/linker.ml -o %t/executable
 * RUN: %t/executable
 * RUN: %ocamlopt -g -w +A -package llvm.linker -linkpkg %t/linker.ml -o %t/executable
 * RUN: %t/executable
 * XFAIL: vg_leak
 *)

(* Note: It takes several seconds for ocamlopt to link an executable with
         libLLVMCore.a, so it's better to write a big test than a bunch of
         little ones. *)

open Llvm
open Llvm_linker

let context = create_context ()
let void_type = Llvm.void_type context

let diagnostic_handler _ = ()

(* Tiny unit test framework - really just to help find which line is busted *)
let print_checkpoints = false

let suite name f =
  if print_checkpoints then
    prerr_endline (name ^ ":");
  f ()


(*===-- Linker -----------------------------------------------------------===*)

let test_linker () =
  set_diagnostic_handler context (Some diagnostic_handler);

  let fty = function_type void_type [| |] in

  let make_module name =
    let m = create_module context name in
    let fn = define_function ("fn_" ^ name) fty m in
    ignore (build_ret_void (builder_at_end context (entry_block fn)));
    m
  in

  let m1 = make_module "one"
  and m2 = make_module "two" in
  link_modules m1 m2;
  dispose_module m1;

  let m1 = make_module "one"
  and m2 = make_module "two" in
  link_modules m1 m2;
  dispose_module m1;

  let m1 = make_module "one"
  and m2 = make_module "one" in
  try
    link_modules m1 m2;
    failwith "must raise"
  with Error _ ->
    dispose_module m1

(*===-- Driver ------------------------------------------------------------===*)

let _ =
  suite "linker" test_linker;
  dispose_context context