akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

brontide: make establishTestConnection full async

Browse Source 
In this commit we modify the establishTestConnection() function that
each of the brontide unit tests utilize. Before this commit, we would
fully block on the Accept method of the listener. Since then it has
been observed, that at times if Accept blocks indefinitely, then the
entire test will fail after 10 minutes. To allow the test to return
early with a pertinent error, we’ll now make the entire test async, so
we can immediately return with an error if detected.
This commit is contained in:
Olaoluwa Osuntokun 2017-11-01 13:28:18 -07:00
parent b28d613b07
commit 1899847823
No known key found for this signature in database
GPG Key ID: 964EA263DD637C21
1 changed files with 26 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
brontide/noise_test.go
View File

@ -14,8 +14,8 @@ import (
)
func establishTestConnection() (net.Conn, net.Conn, error) {
// First, generate the long-term private keys both ends of the connection
// within our test.
// First, generate the long-term private keys both ends of the
// connection within our test.
localPriv, err := btcec.NewPrivateKey(btcec.S256())
if err != nil {
return nil, nil, err
@ -42,24 +42,38 @@ func establishTestConnection() (net.Conn, net.Conn, error) {
}
// Initiate a connection with a separate goroutine, and listen with our
// main one. If both errors are nil, then encryption+auth was succesful.
errChan := make(chan error)
connChan := make(chan net.Conn)
// main one. If both errors are nil, then encryption+auth was
// successful.
conErrChan := make(chan error, 1)
connChan := make(chan net.Conn, 1)
go func() {
conn, err := Dial(remotePriv, netAddr)
errChan <- err
conErrChan <- err
connChan <- conn
}()
localConn, listenErr := listener.Accept()
if listenErr != nil {
return nil, nil, listenErr
lisErrChan := make(chan error, 1)
lisChan := make(chan net.Conn, 1)
go func() {
localConn, listenErr := listener.Accept()
lisErrChan <- listenErr
lisChan <- localConn
}()
select {
case err := <-conErrChan:
if err != nil {
return nil, nil, err
}
case err := <-lisErrChan:
if err != nil {
return nil, nil, err
}
}
if dialErr := <-errChan; dialErr != nil {
return nil, nil, dialErr
}
localConn := <-lisChan
remoteConn := <-connChan
return localConn, remoteConn, nil