From 3b9c77c4859a4218fc4ce54cdfb95ef79e9aae87 Mon Sep 17 00:00:00 2001
From: matograine <tom.ngr@zaclys.net>
Date: Tue, 24 Mar 2020 13:53:31 +0100
Subject: [PATCH] [test] #213: create test for generate_unlocks

---
 tests/test_unit_tx.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/tests/test_unit_tx.py b/tests/test_unit_tx.py
index 53944414..d35c2bcc 100644
--- a/tests/test_unit_tx.py
+++ b/tests/test_unit_tx.py
@@ -15,6 +15,7 @@ from silkaj.tx import (
     generate_and_send_transaction,
     display_sent_tx,
     check_transaction_values,
+    generate_unlocks,
 )
 from silkaj.tui import display_amount, display_pubkey
 from silkaj.money import UDValue
@@ -1334,3 +1335,35 @@ def test_check_transaction_values_errors(
                 != -1
             )
     assert pytest_exit.type == SystemExit
+
+
+# test generate_unlocks()
+@pytest.mark.parametrize(
+    "listinput, expected",
+    [
+        (
+            [
+                InputSource(
+                    amount=100,
+                    base=0,
+                    source="T",
+                    origin_id="1F3059ABF35D78DFB5AFFB3DEAB4F76878B04DB6A14757BBD6B600B1C19157E7",
+                    index=2,
+                ),
+                InputSource(
+                    amount=mock_ud_value,
+                    base=0,
+                    source="D",
+                    origin_id="2sq4w8yYVDWNxVWZqGWWDriFf5z7dn7iLahDCvEEotuY",
+                    index=6,
+                ),
+            ],
+            [
+                Unlock(index=0, parameters=[SIGParameter(0)]),
+                Unlock(index=1, parameters=[SIGParameter(0)]),
+            ],
+        ),
+    ],
+)
+def test_generate_unlocks(listinput, expected):
+    assert expected == generate_unlocks(listinput)
-- 
GitLab