adding cli wallet tests
This commit is contained in:
TheMenko
parent
d77e59309d
commit
8a2d27309d
288
cli/wallet_test.go
Normal file
288
cli/wallet_test.go
Normal file
@ -0,0 +1,288 @@
|
||||
package cli
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/filecoin-project/go-address"
|
||||
"github.com/filecoin-project/go-state-types/abi"
|
||||
"github.com/filecoin-project/go-state-types/big"
|
||||
"github.com/filecoin-project/go-state-types/crypto"
|
||||
"github.com/filecoin-project/lotus/api"
|
||||
apitypes "github.com/filecoin-project/lotus/api/types"
|
||||
types "github.com/filecoin-project/lotus/chain/types"
|
||||
"github.com/golang/mock/gomock"
|
||||
"github.com/ipfs/go-cid"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestWalletNew(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletNew))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
keyType := types.KeyType("secp256k1")
|
||||
address := address.Address{}
|
||||
|
||||
mockApi.EXPECT().WalletNew(ctx, keyType).Return(address, nil)
|
||||
|
||||
err := app.Run([]string{"wallet", "new"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletList(t *testing.T) {
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
addresses := []address.Address{addr}
|
||||
assert.NoError(t, err)
|
||||
|
||||
cid := cid.Cid{}
|
||||
key := types.NewTipSetKey(cid)
|
||||
|
||||
actor := types.Actor{
|
||||
Code: cid,
|
||||
Head: cid,
|
||||
Nonce: 0,
|
||||
Balance: big.NewInt(100),
|
||||
}
|
||||
|
||||
t.Run("wallet-list-addr-only", func(t *testing.T) {
|
||||
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletList))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
gomock.InOrder(
|
||||
mockApi.EXPECT().WalletList(ctx).Return(addresses, nil),
|
||||
mockApi.EXPECT().WalletDefaultAddress(ctx).Return(addr, nil),
|
||||
mockApi.EXPECT().StateGetActor(ctx, addr, key).Return(&actor, nil),
|
||||
)
|
||||
|
||||
err := app.Run([]string{"wallet", "list", "addr-only"})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
t.Run("wallet-list-id", func(t *testing.T) {
|
||||
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletList))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
gomock.InOrder(
|
||||
mockApi.EXPECT().WalletList(ctx).Return(addresses, nil),
|
||||
mockApi.EXPECT().WalletDefaultAddress(ctx).Return(addr, nil),
|
||||
mockApi.EXPECT().StateGetActor(ctx, addr, key).Return(&actor, nil),
|
||||
mockApi.EXPECT().StateLookupID(ctx, addr, key).Return(addr, nil),
|
||||
)
|
||||
|
||||
err := app.Run([]string{"wallet", "list", "--id"})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
t.Run("wallet-list-market", func(t *testing.T) {
|
||||
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletList))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
balance := api.MarketBalance{
|
||||
Escrow: big.NewInt(1234),
|
||||
Locked: big.NewInt(123),
|
||||
}
|
||||
|
||||
gomock.InOrder(
|
||||
mockApi.EXPECT().WalletList(ctx).Return(addresses, nil),
|
||||
mockApi.EXPECT().WalletDefaultAddress(ctx).Return(addr, nil),
|
||||
mockApi.EXPECT().StateGetActor(ctx, addr, key).Return(&actor, nil),
|
||||
mockApi.EXPECT().StateMarketBalance(ctx, addr, key).Return(balance, nil),
|
||||
)
|
||||
|
||||
err := app.Run([]string{"wallet", "list", "--market"})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
}
|
||||
|
||||
func TestWalletBalance(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletBalance))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
balance := big.NewInt(1234)
|
||||
|
||||
mockApi.EXPECT().WalletBalance(ctx, addr).Return(balance, nil)
|
||||
|
||||
err = app.Run([]string{"wallet", "balance", "f01234"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletGetDefault(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletGetDefault))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr := address.Address{}
|
||||
|
||||
mockApi.EXPECT().WalletDefaultAddress(ctx).Return(addr, nil)
|
||||
|
||||
err := app.Run([]string{"wallet", "default"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletSetDefault(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletSetDefault))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
mockApi.EXPECT().WalletSetDefault(ctx, addr).Return(nil)
|
||||
|
||||
err = app.Run([]string{"wallet", "set-default", "f01234"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletExport(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletExport))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
keyInfo := types.KeyInfo{}
|
||||
|
||||
mockApi.EXPECT().WalletExport(ctx, addr).Return(&keyInfo, nil)
|
||||
|
||||
err = app.Run([]string{"wallet", "export", "f01234"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletSign(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletSign))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
msg := []byte{1}
|
||||
signature := crypto.Signature{}
|
||||
|
||||
mockApi.EXPECT().WalletSign(ctx, addr, msg).Return(&signature, nil)
|
||||
|
||||
err = app.Run([]string{"wallet", "sign", "f01234", "01"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletVerify(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletVerify))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
msg := []byte{1}
|
||||
signature := crypto.Signature{
|
||||
Type: crypto.SigTypeSecp256k1,
|
||||
Data: []byte{},
|
||||
}
|
||||
|
||||
mockApi.EXPECT().WalletVerify(ctx, addr, msg, &signature).Return(true, nil)
|
||||
|
||||
err = app.Run([]string{"wallet", "verify", "f01234", "01", "01"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletDelete(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletDelete))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
mockApi.EXPECT().WalletDelete(ctx, addr).Return(nil)
|
||||
|
||||
err = app.Run([]string{"wallet", "delete", "f01234"})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletMarketWithdraw(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletMarket))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
addr, err := address.NewIDAddress(1234)
|
||||
assert.NoError(t, err)
|
||||
|
||||
balance := api.MarketBalance{
|
||||
Escrow: big.NewInt(100),
|
||||
Locked: big.NewInt(10),
|
||||
}
|
||||
|
||||
cid := cid.Cid{}
|
||||
msgLookup := api.MsgLookup{}
|
||||
|
||||
var networkVers apitypes.NetworkVersion
|
||||
|
||||
gomock.InOrder(
|
||||
mockApi.EXPECT().StateMarketBalance(ctx, addr, types.TipSetKey{}).Return(balance, nil),
|
||||
// mock reserve to 10
|
||||
mockApi.EXPECT().MarketGetReserved(ctx, addr).Return(big.NewInt(10), nil),
|
||||
// available should be 80.. escrow - locked - reserve
|
||||
mockApi.EXPECT().MarketWithdraw(ctx, addr, addr, big.NewInt(80)).Return(cid, nil),
|
||||
mockApi.EXPECT().StateWaitMsg(ctx, cid, uint64(5), abi.ChainEpoch(int64(-1)), true).Return(&msgLookup, nil),
|
||||
mockApi.EXPECT().StateNetworkVersion(ctx, types.TipSetKey{}).Return(networkVers, nil),
|
||||
)
|
||||
|
||||
err = app.Run([]string{"wallet", "market", "withdraw", "--wallet", addr.String()})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestWalletMarketAdd(t *testing.T) {
|
||||
app, mockApi, _, done := NewMockAppWithFullAPI(t, WithCategory("wallet", walletMarket))
|
||||
defer done()
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
toAddr := address.Address{}
|
||||
defaultAddr := address.Address{}
|
||||
|
||||
cid := cid.Cid{}
|
||||
|
||||
gomock.InOrder(
|
||||
mockApi.EXPECT().WalletDefaultAddress(ctx).Return(defaultAddr, nil),
|
||||
mockApi.EXPECT().MarketAddBalance(ctx, defaultAddr, toAddr, big.NewInt(80)).Return(cid, nil),
|
||||
)
|
||||
|
||||
err := app.Run([]string{"wallet", "market", "add", "0.000000000000000080", "--address", toAddr.String()})
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user