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
type Bitcoin int
type Wallet struct {
balance int
balance Bitcoin
}
func (w *Wallet) Deposit(units int) {
w.balance += units
func (w *Wallet) Deposit(amount Bitcoin) {
w.balance += amount
}
func (w *Wallet) Balance() int {
func (w *Wallet) Balance() Bitcoin {
return w.balance
}

2
wallet/wallet_test.go
View File

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