mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-02-23 14:40:30 +01:00
Code Issues Projects Releases Wiki Activity

lntest: extend RestartNode to also unlock node if password is present

Browse source
This commit is contained in:
Olaoluwa Osuntokun 2019-03-10 16:32:44 -07:00
parent e929142ff7
commit b419179c86
No known key found for this signature in database
GPG key ID: CE58F7F8E20FD9A2
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lntest/harness.go
View file

@ -576,7 +576,9 @@ func (n *NetworkHarness) DisconnectNodes(ctx context.Context, a, b *HarnessNode)
// This method can be useful when testing edge cases such as a node broadcast // This method can be useful when testing edge cases such as a node broadcast
// and invalidated prior state, or persistent state recovery, simulating node // and invalidated prior state, or persistent state recovery, simulating node
// crashes, etc. // crashes, etc.
func (n *NetworkHarness) RestartNode(node *HarnessNode, callback func() error) error { func (n *NetworkHarness) RestartNode(node *HarnessNode,
callback func() error) error {
if err := node.stop(); err != nil { if err := node.stop(); err != nil {
return err return err
} }
@ -587,7 +589,19 @@ func (n *NetworkHarness) RestartNode(node *HarnessNode, callback func() error) e
} }
} }
return node.start(n.lndErrorChan) if err := node.start(n.lndErrorChan); err != nil {
return err
}
// If the node doesn't have a password set, then we can exit here as we
// don't need to unlock it.
if len(node.cfg.Password) == 0 {
return nil
}
// Otherwise, we'll unlock the wallet, then complete the final steps
// for the node initialization process.
return node.Unlock(context.Background(), node.cfg.Password)
} }
// SuspendNode stops the given node and returns a callback that can be used to // SuspendNode stops the given node and returns a callback that can be used to