60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package balance
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/lmittmann/w3"
|
|
)
|
|
|
|
func TestBatchBalance_TokensBalance(t *testing.T) {
|
|
type args struct {
|
|
owner common.Address
|
|
tokens []common.Address
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want bool
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "A random members (min dust available) balances",
|
|
args: args{
|
|
owner: w3.A("0x4e956b5De3c33566c596754B4fa0ABd9F2789578"),
|
|
tokens: []common.Address{
|
|
w3.A("0xaB89822F31c2092861F713F6F34bd6877a8C1878"),
|
|
w3.A("0x982caeF20362ADEAC3f9a25E37E20E6787f27f85"),
|
|
w3.A("0x9ADd261033baA414c84FF84A4Fe396338C1ba13a"),
|
|
w3.A("0x7dF20b526318d37Cd7DA9518E51d4A51fec30c9A"),
|
|
},
|
|
},
|
|
want: true,
|
|
wantErr: false,
|
|
},
|
|
}
|
|
|
|
batchBalance, err := NewBatchBalance(conf.rpcProvider, w3.A(conf.batchContract))
|
|
|
|
if err != nil {
|
|
t.Fatalf("NewBatchBalance error = %v", err)
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got, err := batchBalance.TokensBalance(context.Background(), tt.args.owner, tt.args.tokens)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("BatchBalance.TokensBalance() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
|
|
if got[2].Int64() > 0 {
|
|
t.Errorf("TokenBalance = %d, want %d", got[2].Int64(), 0)
|
|
}
|
|
|
|
})
|
|
}
|
|
}
|