From 9a45ffd459cb8cfc39ad33d3b3ae24aa08203ab0 Mon Sep 17 00:00:00 2001
From: Dirk McCormick <dirkmdev@gmail.com>
Date: Wed, 8 Jul 2020 17:10:02 -0400
Subject: [PATCH] test: add basic paychmgr store test

---
 paychmgr/store_test.go | 87 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 paychmgr/store_test.go

diff --git a/paychmgr/store_test.go b/paychmgr/store_test.go
new file mode 100644
index 000000000..6ef407f4f
--- /dev/null
+++ b/paychmgr/store_test.go
@@ -0,0 +1,87 @@
+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
+	vouchers, 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
+	lane, 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
+}