From eea8e30e87c79ba29f852295cfd9b17ca891cc0f Mon Sep 17 00:00:00 2001
From: John David Pressman <jd@jdpressman.com>
Date: Fri, 17 Jan 2020 01:53:22 -0800
Subject: [PATCH] Add CameLIGO CHECK_SIGNATURE test to suite

---
 src/test/contracts/check_signature.mligo |  2 ++
 src/test/integration_tests.ml            | 14 ++++++++++++++
 2 files changed, 16 insertions(+)
 create mode 100644 src/test/contracts/check_signature.mligo

diff --git a/src/test/contracts/check_signature.mligo b/src/test/contracts/check_signature.mligo
new file mode 100644
index 000000000..ecd56eb4b
--- /dev/null
+++ b/src/test/contracts/check_signature.mligo
@@ -0,0 +1,2 @@
+let check_signature (pk, signed, msg: key * signature * bytes) : bool =
+  Crypto.check pk signed msg
diff --git a/src/test/integration_tests.ml b/src/test/integration_tests.ml
index 3daec1887..1eec89aaa 100644
--- a/src/test/integration_tests.ml
+++ b/src/test/integration_tests.ml
@@ -1936,6 +1936,19 @@ let check_signature () : unit result =
   let%bind () = expect_eq program "check_signature" make_input make_expected in
   ok ()
 
+let check_signature_mligo () : unit result =
+  let open Tezos_crypto in
+  let (_, raw_pk, sk) = Signature.generate_key () in
+  let pk_str = Signature.Public_key.to_b58check raw_pk in
+  let signed = Signature.sign sk (Bytes.of_string "hello world") in
+  let%bind program = mtype_file "./contracts/check_signature.mligo" in
+  let make_input = e_tuple [e_key pk_str ;
+                            e_signature (Signature.to_b58check signed) ;
+                            e_bytes_ofbytes (Bytes.of_string "hello world")] in
+  let make_expected = e_bool true in
+  let%bind () = expect_eq program "check_signature" make_input make_expected in
+  ok ()
+
 let curry () : unit result =
   let%bind program = mtype_file "./contracts/curry.mligo" in
   let%bind () =
@@ -2073,6 +2086,7 @@ let main = test_suite "Integration (End to End)" [
     test "key hash (mligo)" key_hash_mligo ;
     test "key hash (religo)" key_hash_religo ;
     test "check signature" check_signature ;
+    test "check signature (mligo)" check_signature_mligo ;
     test "chain id" chain_id ;
     test "type alias" type_alias ;
     test "function" function_ ;