From dca65ede97d521bb4483080b7ec2fbb20047f62f Mon Sep 17 00:00:00 2001 From: Jonathan Zernik Date: Sat, 23 Oct 2021 10:38:44 -0500 Subject: [PATCH] Add test for yield received payments from settle index (#1703) --- tests/db/test_squeak_db.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/db/test_squeak_db.py b/tests/db/test_squeak_db.py index 6aa3ca15..361e8cb4 100644 --- a/tests/db/test_squeak_db.py +++ b/tests/db/test_squeak_db.py @@ -1460,3 +1460,21 @@ def test_get_received_payments(squeak_db, inserted_received_payment_ids): ) assert len(received_payments) == len(inserted_received_payment_ids) + + +def test_yield_received_payments_from_settle_index(squeak_db, inserted_received_payment_ids): + payments_iter = squeak_db.yield_received_payments_from_index( + start_index=0, + ) + + assert len(list(payments_iter)) == len(inserted_received_payment_ids) + + +def test_yield_received_payments_from_other_settle_index(squeak_db, inserted_received_payment_ids): + index = 43 + payments_iter = squeak_db.yield_received_payments_from_index( + start_index=index, + ) + + assert len(list(payments_iter)) == len( + inserted_received_payment_ids) - index