htlcswitch: add tests to ensure we don't forward to inactive links

This commit is contained in:
Olaoluwa Osuntokun
2017-12-10 16:19:40 -08:00
parent c3d345b575
commit 8da46e0b5a
2 changed files with 119 additions and 9 deletions

View File

@@ -397,16 +397,19 @@ type mockChannelLink struct {
peer Peer
packets chan *htlcPacket
eligible bool
}
func newMockChannelLink(chanID lnwire.ChannelID, shortChanID lnwire.ShortChannelID,
peer Peer) *mockChannelLink {
peer Peer, eligible bool) *mockChannelLink {
return &mockChannelLink{
chanID: chanID,
shortChanID: shortChanID,
packets: make(chan *htlcPacket, 1),
peer: peer,
eligible: eligible,
}
}
@@ -430,7 +433,7 @@ func (f *mockChannelLink) Bandwidth() lnwire.MilliSatoshi { return 99999999
func (f *mockChannelLink) Peer() Peer { return f.peer }
func (f *mockChannelLink) Start() error { return nil }
func (f *mockChannelLink) Stop() {}
func (f *mockChannelLink) EligibleToForward() bool { return true }
func (f *mockChannelLink) EligibleToForward() bool { return f.eligible }
var _ ChannelLink = (*mockChannelLink)(nil)