From f455fa237672a65c4bea4936cb81b3db5e4ec854 Mon Sep 17 00:00:00 2001
From: Lesenechal Remi <lesenechal.remi@gmail.com>
Date: Mon, 25 May 2020 19:30:25 +0200
Subject: [PATCH] add a test case for double fold env recomputing

---
 src/bin/expect_tests/michelson_converter.ml   | 22 ++++++
 .../contracts/double_fold_converter.religo    | 72 +++++++++++++++++++
 2 files changed, 94 insertions(+)
 create mode 100644 src/test/contracts/double_fold_converter.religo

diff --git a/src/bin/expect_tests/michelson_converter.ml b/src/bin/expect_tests/michelson_converter.ml
index f1437f44f..cc055ee24 100644
--- a/src/bin/expect_tests/michelson_converter.ml
+++ b/src/bin/expect_tests/michelson_converter.ml
@@ -184,6 +184,28 @@ let%expect_test _ =
 
 let%expect_test _ =
   run_ligo_good [ "compile-contract" ; (contract "michelson_converter_mixed_pair_or.mligo") ; "main2" ] ;
+  [%expect {|
+    { parameter
+        (or (pair %option1 (string %bar) (nat %baz)) (pair %option2 (string %bar) (nat %baz))) ;
+      storage nat ;
+      code { DUP ;
+             CAR ;
+             IF_LEFT
+               { DUP ; LEFT (pair (string %bar) (nat %baz)) ; DIP { DROP } }
+               { DUP ; RIGHT (pair (string %bar) (nat %baz)) ; DIP { DROP } } ;
+             DUP ;
+             IF_LEFT
+               { DUP ; LEFT (pair (string %bar) (nat %baz)) ; DIP { DROP } }
+               { DUP ; RIGHT (pair (string %bar) (nat %baz)) ; DIP { DROP } } ;
+             DIP { DROP } ;
+             DUP ;
+             IF_LEFT
+               { DUP ; CDR ; NIL operation ; PAIR ; DIP { DROP } }
+               { DUP ; CDR ; NIL operation ; PAIR ; DIP { DROP } } ;
+             DIP { DROP 2 } } } |}]
+
+let%expect_test _ =
+  run_ligo_good [ "compile-contract" ; (contract "double_fold_converter.religo") ; "main" ] ;
   [%expect {|
     { parameter
         (or (pair %option1 (string %bar) (nat %baz)) (pair %option2 (string %bar) (nat %baz))) ;
diff --git a/src/test/contracts/double_fold_converter.religo b/src/test/contracts/double_fold_converter.religo
new file mode 100644
index 000000000..01245222b
--- /dev/null
+++ b/src/test/contracts/double_fold_converter.religo
@@ -0,0 +1,72 @@
+type tokenId = nat;
+type tokenOwner = address;
+type tokenAmount = nat;
+type transferContents = {
+    to_: tokenOwner,
+    token_id: tokenId,
+    amount: tokenAmount
+};
+type transfer = {
+    from_: tokenOwner,
+    txs: list(transferContents)
+};
+type transferContentsMichelson = michelson_pair_right_comb(transferContents);
+type transferAuxiliary = {
+    from_: tokenOwner,
+    txs: list(transferContentsMichelson)
+};
+type transferMichelson = michelson_pair_right_comb(transferAuxiliary);
+type transferParameter = list(transferMichelson);
+type parameter = 
+| Transfer(transferParameter)
+type storage = big_map(tokenId, tokenOwner);
+type entrypointParameter = (parameter, storage);
+type entrypointReturn = (list(operation), storage);
+let errorTokenUndefined = "TOKEN_UNDEFINED";
+let errorNotOwner = "NOT_OWNER";
+let errorInsufficientBalance = "INSUFFICIENT_BALANCE";
+type transferContentsIteratorAccumulator = (storage, tokenOwner);
+let transferContentsIterator = ((accumulator, transferContentsMichelson): (transferContentsIteratorAccumulator, transferContentsMichelson)): transferContentsIteratorAccumulator => {
+    let (storage, from_) = accumulator;
+    let transferContents: transferContents = Layout.convert_from_right_comb(transferContentsMichelson);
+    let tokenOwner: option(tokenOwner) = Map.find_opt(transferContents.token_id, storage);
+    let tokenOwner = switch (tokenOwner) {
+        | None => (failwith(errorTokenUndefined): tokenOwner)
+        | Some(tokenOwner) => if (tokenOwner == from_) {
+                tokenOwner
+            } else {
+                (failwith(errorInsufficientBalance): tokenOwner);
+            }
+    };
+    let storage = Map.update(
+        transferContents.token_id,
+        Some(transferContents.to_),
+        storage
+    );
+    (storage, from_)
+};
+let allowOnlyOwnTransfer = (from: tokenOwner): unit => {
+    if (from != Tezos.sender) {
+        failwith(errorNotOwner)
+    } else { (); }
+}
+let transferIterator = ((storage, transferMichelson): (storage, transferMichelson)): storage => {
+    let transferAuxiliary2: transferAuxiliary = Layout.convert_from_right_comb(transferMichelson);
+    let from_: tokenOwner = transferAuxiliary2.from_;
+    allowOnlyOwnTransfer(from_);
+    let (storage, _) = List.fold(
+        transferContentsIterator, 
+        transferAuxiliary2.txs,
+        (storage, from_)
+    );
+    storage
+};
+let transfer = ((transferParameter, storage): (transferParameter, storage)): entrypointReturn => {
+    let storage = List.fold(transferIterator, transferParameter, storage);
+    (([]: list(operation)), storage);
+};
+let main = ((parameter, storage): entrypointParameter): entrypointReturn => {
+    switch (parameter) {
+        | Transfer(transferParameter) => transfer((transferParameter, storage))
+    }
+}
\ No newline at end of file