multi: patch unit tests for handling pending channels

lnpeer/mock_peer.go

@@ -0,0 +1,82 @@
+package lnpeer
+
+import (
+	"net"
+
+	"github.com/btcsuite/btcd/btcec/v2"
+	"github.com/btcsuite/btcd/wire"
+	"github.com/lightningnetwork/lnd/channeldb"
+	"github.com/lightningnetwork/lnd/lnwire"
+	"github.com/stretchr/testify/mock"
+)
+
+// MockPeer implements the `lnpeer.Peer` interface.
+type MockPeer struct {
+	mock.Mock
+}
+
+// Compile time assertion that MockPeer implements lnpeer.Peer.
+var _ Peer = (*MockPeer)(nil)
+
+func (m *MockPeer) SendMessage(sync bool, msgs ...lnwire.Message) error {
+	args := m.Called(sync, msgs)
+	return args.Error(0)
+}
+
+func (m *MockPeer) SendMessageLazy(sync bool, msgs ...lnwire.Message) error {
+	args := m.Called(sync, msgs)
+	return args.Error(0)
+}
+
+func (m *MockPeer) AddNewChannel(channel *channeldb.OpenChannel,
+	cancel <-chan struct{}) error {
+
+	args := m.Called(channel, cancel)
+	return args.Error(0)
+}
+
+func (m *MockPeer) AddPendingChannel(cid lnwire.ChannelID,
+	cancel <-chan struct{}) error {
+
+	args := m.Called(cid, cancel)
+	return args.Error(0)
+}
+
+func (m *MockPeer) RemovePendingChannel(cid lnwire.ChannelID) error {
+	args := m.Called(cid)
+	return args.Error(0)
+}
+
+func (m *MockPeer) WipeChannel(op *wire.OutPoint) {
+	m.Called(op)
+}
+
+func (m *MockPeer) PubKey() [33]byte {
+	args := m.Called()
+	return args.Get(0).([33]byte)
+}
+
+func (m *MockPeer) IdentityKey() *btcec.PublicKey {
+	args := m.Called()
+	return args.Get(0).(*btcec.PublicKey)
+}
+
+func (m *MockPeer) Address() net.Addr {
+	args := m.Called()
+	return args.Get(0).(net.Addr)
+}
+
+func (m *MockPeer) QuitSignal() <-chan struct{} {
+	args := m.Called()
+	return args.Get(0).(<-chan struct{})
+}
+
+func (m *MockPeer) LocalFeatures() *lnwire.FeatureVector {
+	args := m.Called()
+	return args.Get(0).(*lnwire.FeatureVector)
+}
+
+func (m *MockPeer) RemoteFeatures() *lnwire.FeatureVector {
+	args := m.Called()
+	return args.Get(0).(*lnwire.FeatureVector)
+}