diff options
Diffstat (limited to 'internal/casino/manager_test.go')
| -rw-r--r-- | internal/casino/manager_test.go | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/internal/casino/manager_test.go b/internal/casino/manager_test.go index fa26aa1..b037efa 100644 --- a/internal/casino/manager_test.go +++ b/internal/casino/manager_test.go @@ -18,7 +18,7 @@ func (w *testWallet) Wager(amount int) (Wager, error) { w.chips -= chips credits := amount - chips w.credits -= credits - wager := Wager{Total: amount, Chips: chips, Credits: credits} + wager := Wager{Total: amount, Chips: chips, Credits: credits, ID: string(rune('a' + len(w.open)))} w.open = append(w.open, wager) return wager, nil } @@ -28,6 +28,24 @@ func (w *testWallet) PayCredits(amount int) error { return nil } +// PayStakeCover routes through PayCredits so a wallet rejecting payouts +// (failingPayoutWallet) rejects stake-covering settlements identically. +func (w *testWallet) PayStakeCover(amount int, _ []string) error { + return w.PayCredits(amount) +} + +// ResolveStake settles one outstanding stake by handle, mirroring the +// production wallet. +func (w *testWallet) ResolveStake(stakeID string) { + kept := w.open[:0] + for _, wager := range w.open { + if wager.ID != stakeID { + kept = append(kept, wager) + } + } + w.open = kept +} + // ResolveStakes mirrors the production wallet: every outstanding stake is // settled (no longer refundable). func (w *testWallet) ResolveStakes() { w.open = nil } |