diff options
Diffstat (limited to 'internal/casino/manager_test.go')
| -rw-r--r-- | internal/casino/manager_test.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/internal/casino/manager_test.go b/internal/casino/manager_test.go index 9b5d35a..a832d53 100644 --- a/internal/casino/manager_test.go +++ b/internal/casino/manager_test.go @@ -45,6 +45,14 @@ func (w *testWallet) ResolveStake(stakeID string) { w.open = kept } +// ResolveStakesByIDs consumes the listed stakes in one step, mirroring the +// production wallet. +func (w *testWallet) ResolveStakesByIDs(stakeIDs ...string) { + for _, id := range stakeIDs { + w.ResolveStake(id) + } +} + // ResolveStakes mirrors the production wallet: every outstanding stake is // settled (no longer refundable). func (w *testWallet) ResolveStakes() { w.open = nil } |