vinchent/go-by-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wallet: add withdraw method

Browse Source
This commit is contained in:
vinchent 2024-09-11 11:22:04 +02:00
parent 36ef78fb92
commit 8ed0235074
2 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wallet/wallet.go
View File

@ -16,6 +16,10 @@ func (w *Wallet) Deposit(amount Bitcoin) {
w.balance += amount w.balance += amount
} }
func (w *Wallet) Withdraw(amount Bitcoin) {
w.balance -= amount
}
func (w *Wallet) Balance() Bitcoin { func (w *Wallet) Balance() Bitcoin {
return w.balance return w.balance
} }

18
wallet/wallet_test.go
View File

@ -3,14 +3,22 @@ package wallet
import "testing" import "testing"
func TestWallet(t *testing.T) { func TestWallet(t *testing.T) {
wallet := Wallet{} assertBalance := func(t testing.TB, wallet Wallet, want Bitcoin) {
t.Helper()
wallet.Deposit(10)
got := wallet.Balance() got := wallet.Balance()
want := Bitcoin(10)
if got != want { if got != want {
t.Errorf("got %q want %q", got, want) t.Errorf("got %q want %q", got, want)
} }
}
t.Run("deposit", func(t *testing.T) {
wallet := Wallet{}
wallet.Deposit(10)
assertBalance(t, wallet, Bitcoin(10))
})
t.Run("withdraw", func(t *testing.T) {
wallet := Wallet{balance: Bitcoin(20)}
wallet.Withdraw(10)
assertBalance(t, wallet, Bitcoin(10))
})
} }