cic-go/batch_balance/token_balances_test.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)
			}

		})
	}
}