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

wallet: use self defined type Bitcoin

Browse Source
This commit is contained in:
vinchent 2024-09-11 11:15:24 +02:00
parent 8644bf547b
commit 71bc024fe2
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
wallet/wallet.go
View File

@ -1,13 +1,15 @@
package wallet package wallet
type Bitcoin int
type Wallet struct { type Wallet struct {
balance int balance Bitcoin
} }
func (w *Wallet) Deposit(units int) { func (w *Wallet) Deposit(amount Bitcoin) {
w.balance += units w.balance += amount
} }
func (w *Wallet) Balance() int { func (w *Wallet) Balance() Bitcoin {
return w.balance return w.balance
} }

2
wallet/wallet_test.go
View File

@ -8,7 +8,7 @@ func TestWallet(t *testing.T) {
wallet.Deposit(10) wallet.Deposit(10)
got := wallet.Balance() got := wallet.Balance()
want := 10 want := Bitcoin(10)
if got != want { if got != want {
t.Errorf("got %d want %d", got, want) t.Errorf("got %d want %d", got, want)