mirror of https://github.com/BTCPrivate/lnd.git
routing: add RestartRouter method to testCtx
This commit is contained in:
parent
cdf2f43432
commit
3be905109c
|
@ -31,6 +31,37 @@ type testCtx struct {
|
||||||
chainView *mockChainView
|
chainView *mockChainView
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *testCtx) RestartRouter() error {
|
||||||
|
// First, we'll reset the chainView's state as it doesn't persist the
|
||||||
|
// filter between restarts.
|
||||||
|
c.chainView.Reset()
|
||||||
|
|
||||||
|
// With the chainView reset, we'll now re-create the router itself, and
|
||||||
|
// start it.
|
||||||
|
router, err := New(Config{
|
||||||
|
Graph: c.graph,
|
||||||
|
Chain: c.chain,
|
||||||
|
ChainView: c.chainView,
|
||||||
|
SendToSwitch: func(_ *btcec.PublicKey,
|
||||||
|
_ *lnwire.UpdateAddHTLC, _ *sphinx.Circuit) ([32]byte, error) {
|
||||||
|
return [32]byte{}, nil
|
||||||
|
},
|
||||||
|
ChannelPruneExpiry: time.Hour * 24,
|
||||||
|
GraphPruneInterval: time.Hour * 2,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("unable to create router %v", err)
|
||||||
|
}
|
||||||
|
if err := router.Start(); err != nil {
|
||||||
|
return fmt.Errorf("unable to start router: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Finally, we'll swap out the pointer in the testCtx with this fresh
|
||||||
|
// instance of the router.
|
||||||
|
c.router = router
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func createTestCtx(startingHeight uint32, testGraph ...string) (*testCtx, func(), error) {
|
func createTestCtx(startingHeight uint32, testGraph ...string) (*testCtx, func(), error) {
|
||||||
var (
|
var (
|
||||||
graph *channeldb.ChannelGraph
|
graph *channeldb.ChannelGraph
|
||||||
|
|
Loading…
Reference in New Issue