multi: add interface for subscribe client so it can be mocked
The current implementation of subscribe is difficult to mock because the queue that we send updates on in unexported, so you cannot create a subscribe.Client object and then add your own updates. While it is possible to run a subscribe server in tests, subscribe servers will shutdown before dispatching their udpates to all clients, which can be flakey (and is difficult to workaround). In this commit, we add a subscription interface so that these testing struggles can be addressed with a mock.
This commit is contained in:
4 files changed
+36
-8
No files matched your search
@@ -60,6 +60,7 @@ import (
|
||||
"github.com/lightningnetwork/lnd/routing"
|
||||
"github.com/lightningnetwork/lnd/routing/localchans"
|
||||
"github.com/lightningnetwork/lnd/routing/route"
|
||||
"github.com/lightningnetwork/lnd/subscribe"
|
||||
"github.com/lightningnetwork/lnd/sweep"
|
||||
"github.com/lightningnetwork/lnd/ticker"
|
||||
"github.com/lightningnetwork/lnd/tor"
|
||||
@@ -1203,9 +1204,13 @@ func newServer(cfg *Config, listenAddrs []net.Addr,
|
||||
|
||||
// Create a channel event store which monitors all open channels.
|
||||
s.chanEventStore = chanfitness.NewChannelEventStore(&chanfitness.Config{
|
||||
SubscribeChannelEvents: s.channelNotifier.SubscribeChannelEvents,
|
||||
SubscribePeerEvents: s.peerNotifier.SubscribePeerEvents,
|
||||
GetOpenChannels: s.remoteChanDB.FetchAllOpenChannels,
|
||||
SubscribeChannelEvents: func() (subscribe.Subscription, error) {
|
||||
return s.channelNotifier.SubscribeChannelEvents()
|
||||
},
|
||||
SubscribePeerEvents: func() (subscribe.Subscription, error) {
|
||||
return s.peerNotifier.SubscribePeerEvents()
|
||||
},
|
||||
GetOpenChannels: s.remoteChanDB.FetchAllOpenChannels,
|
||||
})
|
||||
|
||||
if cfg.WtClient.Active {
|
||||
|
||||
Reference in new issue
Block a user