diff --git a/tests/test_unit_tx.py b/tests/test_unit_tx.py index 539444146a332363b32447e5ace8afe8f58e628d..d35c2bcc9f75a932e325710a47523bdb281f5ccf 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)