more test

Author: gares

tests/sources/dune

@@ -16,4 +16,5 @@
 (executable (name sepcomp_tyid)  (modules sepcomp_tyid)  (libraries sepcomp))
 (executable (name sepcomp_tyid2)  (modules sepcomp_tyid2)  (libraries sepcomp))
 (executable (name sepcomp_hover)  (modules sepcomp_hover)  (libraries sepcomp))
-(executable (name sepcomp_extend_sig)  (modules sepcomp_extend_sig)  (libraries sepcomp))
+(executable (name sepcomp_extend_sig)  (modules sepcomp_extend_sig)  (libraries sepcomp))
+(executable (name sepcomp_extend_sig2)  (modules sepcomp_extend_sig2)  (libraries sepcomp))

tests/sources/sepcomp_extend_sig2.ml

@@ -0,0 +1,32 @@
+let u = {|
+
+pred p i:int.
+p 2 :- !, fail.
+
+|}
+;;
+
+let v = {|
+
+p _.
+main :- p 2.
+
+|}
+;;
+
+let () =
+  let open Sepcomp.Sepcomp_template in
+  let elpi = init () in
+  let flags = Elpi.API.Compile.default_flags in
+  let base = Elpi.API.Compile.empty_base ~elpi in
+  let _, u = cc ~elpi ~flags ~base 0 u in
+  let su = signature_of u in
+  let base = extend_signature ~flags ~base su in
+  let base = extend ~flags ~base u in
+  let base, _ = cc ~elpi ~flags ~base 0 v in
+  let q = query ~elpi base in
+  assert(try_exec q = false);
+  let base, _ = cc ~elpi ~flags ~base 0 "fail." in
+  let q = query ~elpi base in
+  exec q
+

tests/sources/sepcomp_template.ml

@@ -32,6 +32,13 @@ let exec q =
     | Execute.Success _ -> exit 0
     | Execute.NoMoreSteps -> assert false
 
+let try_exec q =
+    let exe = Compile.optimize q in
+    match Execute.once exe with
+    | Execute.Failure -> false
+    | Execute.Success _ -> true
+    | Execute.NoMoreSteps -> assert false
+    
 let main us =
   let elpi = init () in
   let flags = Compile.default_flags in

tests/suite/elpi_api.ml

@@ -72,3 +72,9 @@ let () = declare "sepcomp_extend_sig"
   ~expectation:Test.Success
   ()
 
+let () = declare "sepcomp_extend_sig2"
+  ~source_dune:"sepcomp_extend_sig2.exe"
+  ~description:"extend unit with signature then code"
+  ~expectation:Test.Success
+  ()
+