func TestGetBalance(t *testing.T) {
a := account.ExchangeAccount{
Balance: map[coin.Type]uint64{
coin.Bitcoin: 90000,
coin.Skycoin: 450000,
},
}
if a.GetBalance(coin.Bitcoin) != 90000 {
t.Error("get bitcoin balance failed")
return
}
if a.GetBalance(coin.Skycoin) != 450000 {
t.Error("get skycoin balance failed")
return
}
}
func TestDecreaseBalance(t *testing.T) {
var btcInit uint64 = 90000
var skyInit uint64 = 450000
testData := map[coin.Type][]struct {
V uint64
Expect uint64
}{
coin.Bitcoin: {
{10000, 80000},
{20000, 70000},
{1000, 89000},
{100, 89900},
},
coin.Skycoin: {
{10000, 440000},
{30000, 420000},
{50000, 400000},
},
}
for cp, tds := range testData {
for _, d := range tds {
a := account.ExchangeAccount{
Balance: map[coin.Type]uint64{
coin.Bitcoin: btcInit,
coin.Skycoin: skyInit,
},
}
if err := a.DecreaseBalance(cp, d.V); err != nil {
t.Error(err)
return
}
b := a.GetBalance(cp)
if b != d.Expect {
t.Errorf("decrease %s balance failed, v:%d, expect:%d", cp, b, d.Expect)
return
}
}
}
}