highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-05-14 13:50:04 +02:00
Code Issues Packages Projects Releases Wiki Activity

lntest: create persistent connection

Browse Source 
This commit adds a ConnectNodesPerm function to the itest NetworkHarness
so that persistent connections between nodes can be mocked.
This commit is contained in:
Elle Mouton 2021-07-17 12:01:39 +02:00
parent 9a97577c00
commit 5177ec2f84
No known key found for this signature in database
GPG Key ID: D7D916376026F177
1 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lntest/harness.go
View File

@ -670,13 +670,31 @@ func (n *NetworkHarness) EnsureConnected(t *testing.T, a, b *HarnessNode) {
) )
} }
// ConnectNodes establishes an encrypted+authenticated p2p connection from node // ConnectNodes attempts to create a connection between nodes a and b.
func (n *NetworkHarness) ConnectNodes(t *testing.T, a, b *HarnessNode) {
n.connectNodes(t, a, b, false)
}
// ConnectNodesPerm attempts to connect nodes a and b and sets node b as
// a peer that node a should persistently attempt to reconnect to if they
// become disconnected.
func (n *NetworkHarness) ConnectNodesPerm(t *testing.T,
a, b *HarnessNode) {
n.connectNodes(t, a, b, true)
}
// connectNodes establishes an encrypted+authenticated p2p connection from node
// a towards node b. The function will return a non-nil error if the connection // a towards node b. The function will return a non-nil error if the connection
// was unable to be established. // was unable to be established. If the perm parameter is set to true then
// node a will persistently attempt to reconnect to node b if they get
// disconnected.
// //
// NOTE: This function may block for up to 15-seconds as it will not return // NOTE: This function may block for up to 15-seconds as it will not return
// until the new connection is detected as being known to both nodes. // until the new connection is detected as being known to both nodes.
func (n *NetworkHarness) ConnectNodes(t *testing.T, a, b *HarnessNode) { func (n *NetworkHarness) connectNodes(t *testing.T, a, b *HarnessNode,
perm bool) {
ctxb := context.Background() ctxb := context.Background()
ctx, cancel := context.WithTimeout(ctxb, DefaultTimeout) ctx, cancel := context.WithTimeout(ctxb, DefaultTimeout)
defer cancel() defer cancel()
@ -692,6 +710,7 @@ func (n *NetworkHarness) ConnectNodes(t *testing.T, a, b *HarnessNode) {
Pubkey: bobInfo.IdentityPubkey, Pubkey: bobInfo.IdentityPubkey,
Host: b.Cfg.P2PAddr(), Host: b.Cfg.P2PAddr(),
}, },
Perm: perm,
} }
err = n.connect(ctx, req, a) err = n.connect(ctx, req, a)