lotus/paychmgr/store_test.go
2020-07-09 18:49:43 -04:00

88 lines
2.2 KiB
Go

package paychmgr
import (
"testing"
"github.com/filecoin-project/go-address"
tutils "github.com/filecoin-project/specs-actors/support/testing"
ds "github.com/ipfs/go-datastore"
ds_sync "github.com/ipfs/go-datastore/sync"
"github.com/stretchr/testify/require"
)
func TestStore(t *testing.T) {
store := NewStore(ds_sync.MutexWrap(ds.NewMapDatastore()))
addrs, err := store.ListChannels()
require.NoError(t, err)
require.Len(t, addrs, 0)
ci := &ChannelInfo{
Channel: tutils.NewIDAddr(t, 100),
Control: tutils.NewIDAddr(t, 101),
Target: tutils.NewIDAddr(t, 102),
Direction: DirOutbound,
Vouchers: []*VoucherInfo{{Voucher: nil, Proof: []byte{}}},
}
ci2 := &ChannelInfo{
Channel: tutils.NewIDAddr(t, 200),
Control: tutils.NewIDAddr(t, 201),
Target: tutils.NewIDAddr(t, 202),
Direction: DirOutbound,
Vouchers: []*VoucherInfo{{Voucher: nil, Proof: []byte{}}},
}
// Track the channel
err = store.TrackChannel(ci)
require.NoError(t, err)
// Tracking same channel again should error
err = store.TrackChannel(ci)
require.Error(t, err)
// Track another channel
err = store.TrackChannel(ci2)
require.NoError(t, err)
// List channels should include all channels
addrs, err = store.ListChannels()
require.NoError(t, err)
require.Len(t, addrs, 2)
require.Contains(t, addrsStrings(addrs), "t0100")
require.Contains(t, addrsStrings(addrs), "t0200")
// Request vouchers for channel
vouchers, err := store.VouchersForPaych(ci.Channel)
require.NoError(t, err)
require.Len(t, vouchers, 1)
// Requesting voucher for non-existent channel should error
_, err = store.VouchersForPaych(tutils.NewIDAddr(t, 300))
require.Equal(t, err, ErrChannelNotTracked)
// Allocate lane for channel
lane, err := store.AllocateLane(ci.Channel)
require.NoError(t, err)
require.Equal(t, lane, uint64(0))
// Allocate next lane for channel
lane, err = store.AllocateLane(ci.Channel)
require.NoError(t, err)
require.Equal(t, lane, uint64(1))
// Allocate next lane for non-existent channel should error
_, err = store.AllocateLane(tutils.NewIDAddr(t, 300))
require.Equal(t, err, ErrChannelNotTracked)
}
func addrsStrings(addrs []address.Address) []string {
str := make([]string, len(addrs))
for i, a := range addrs {
str[i] = a.String()
}
return str
}