65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
|
package cli
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/go-address"
|
||
|
|
"github.com/filecoin-project/go-state-types/crypto"
|
||
|
|
"github.com/filecoin-project/lotus/api/mocks"
|
||
|
|
types "github.com/filecoin-project/lotus/chain/types"
|
||
|
|
gomock "github.com/golang/mock/gomock"
|
||
|
|
cid "github.com/ipfs/go-cid"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSendService(t *testing.T) {
|
||
|
|
mockCtrl := gomock.NewController(t)
|
||
|
|
defer mockCtrl.Finish()
|
||
|
|
|
||
|
|
mockApi := mocks.NewMockFullNode(mockCtrl)
|
||
|
|
|
||
|
|
srvcs := &ServicesImpl{
|
||
|
|
api: mockApi,
|
||
|
|
closer: func() {},
|
||
|
|
}
|
||
|
|
|
||
|
|
addrGen := address.NewForTestGetter()
|
||
|
|
a1 := addrGen()
|
||
|
|
a2 := addrGen()
|
||
|
|
|
||
|
|
const balance = 10000
|
||
|
|
|
||
|
|
params := SendParams{
|
||
|
|
From: a1,
|
||
|
|
To: a2,
|
||
|
|
Val: types.NewInt(balance - 100),
|
||
|
|
}
|
||
|
|
ctx, done := context.WithCancel(context.Background())
|
||
|
|
defer done()
|
||
|
|
|
||
|
|
msgCid := cid.Undef
|
||
|
|
gomock.InOrder(
|
||
|
|
mockApi.EXPECT().WalletBalance(ctx, params.From).Return(types.NewInt(balance), nil),
|
||
|
|
mockApi.EXPECT().MpoolPushMessage(ctx, gomock.Any(), nil).DoAndReturn(
|
||
|
|
func(_ context.Context, msg *types.Message, _ interface{}) (*types.SignedMessage, error) {
|
||
|
|
msgCid = msg.Cid()
|
||
|
|
assert.Equal(t, params.From, msg.From)
|
||
|
|
assert.Equal(t, params.To, msg.To)
|
||
|
|
assert.Equal(t, params.Val, msg.Value)
|
||
|
|
|
||
|
|
sm := types.SignedMessage{
|
||
|
|
Message: *msg,
|
||
|
|
Signature: crypto.Signature{1, []byte{1}},
|
||
|
|
}
|
||
|
|
msgCid = sm.Cid()
|
||
|
|
return &sm, nil
|
||
|
|
}),
|
||
|
|
)
|
||
|
|
|
||
|
|
c, err := srvcs.Send(ctx, params)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, msgCid, c)
|
||
|
|
|
||
|
|
}
|