From 14b9f289b62b396da92bddc81b35609254cee9a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Moreau?= <cem.moreau@gmail.com> Date: Mon, 30 Dec 2024 19:32:59 +0100 Subject: [PATCH] Resolve #275 "[test] UD out of reeval window" (nodes/rust/duniter-v2s!301) * refac: comments + renaming `ud_index` which is confusiong * test: unclaimed UDs out of UD reeval window --- .../src/compute_claim_uds.rs | 29 +++++++++++++++---- 1 file changed, 24 insertions(+), 5 deletions(-) diff --git a/pallets/universal-dividend/src/compute_claim_uds.rs b/pallets/universal-dividend/src/compute_claim_uds.rs index 7d58edcea..1c2cbe8b7 100644 --- a/pallets/universal-dividend/src/compute_claim_uds.rs +++ b/pallets/universal-dividend/src/compute_claim_uds.rs @@ -24,18 +24,22 @@ pub(super) fn compute_claim_uds<Balance: AtLeast32BitUnsigned>( ) -> (UdIndex, Balance) { let mut total_amount = Zero::zero(); let mut total_count = 0; - for (ud_index, ud_amount) in past_reevals.rev() { - if ud_index <= first_ud_index { + // We start in reverse order, i.e. the most recent reeval first + for (reeval_index, ud_amount) in past_reevals.rev() { + // Therefore, if our first UD is above the current reeval index, we have reached our final useful reeval and must break + if reeval_index <= first_ud_index { let count = current_ud_index - first_ud_index; total_amount += Balance::from(count) * ud_amount; total_count += count; // First unclaimed UD is reached; stop counting now. break; - } else { - let count = current_ud_index - ud_index; + } + // Otherwise, we consume the full reeval contained UDs + else { + let count = current_ud_index - reeval_index; total_amount += Balance::from(count) * ud_amount; total_count += count; - current_ud_index = ud_index; + current_ud_index = reeval_index; } } @@ -109,4 +113,19 @@ mod tests { (2, 110_000) ); } + + #[test] + fn very_old_unclaimed_ud_out_of_reevals() { + let past_reevals = vec![ + // (3, 100 as Balance), "old" reeval which has gone out of reevals window. + (4, 1_000 as Balance), + (5, 10_000 as Balance), + (6, 100_000 as Balance), + ]; + // All the UDs out of the reeval window must produce 0 money units + assert_eq!( + compute_claim_uds(7, 1, past_reevals.into_iter()), + (3, 111_000) + ); + } } -- GitLab