From e2b21dae9f8ef38a838fbf6258fbeec8839e0f47 Mon Sep 17 00:00:00 2001 From: Andy Wang Date: Sun, 20 Oct 2019 15:21:31 +0100 Subject: [PATCH] Fix tests --- internal/multiplex/switchboard_test.go | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/internal/multiplex/switchboard_test.go b/internal/multiplex/switchboard_test.go index 32434fe..494ea2e 100644 --- a/internal/multiplex/switchboard_test.go +++ b/internal/multiplex/switchboard_test.go @@ -9,9 +9,17 @@ import ( ) func TestSwitchboard_Send(t *testing.T) { + getHole := func() net.Conn { + l, _ := net.Listen("tcp", "127.0.0.1:0") + go func() { + net.Dial("tcp", l.Addr().String()) + }() + hole, _ := l.Accept() + return hole + } doTest := func(seshConfig *SessionConfig) { sesh := MakeSession(0, seshConfig) - hole0 := newBlackHole() + hole0 := getHole() sesh.sb.addConn(hole0) connId, err := sesh.sb.assignRandomConn() if err != nil { @@ -26,7 +34,7 @@ func TestSwitchboard_Send(t *testing.T) { return } - hole1 := newBlackHole() + hole1 := getHole() sesh.sb.addConn(hole1) connId, err = sesh.sb.assignRandomConn() if err != nil { @@ -39,8 +47,6 @@ func TestSwitchboard_Send(t *testing.T) { return } - hole0.Close() - connId, err = sesh.sb.assignRandomConn() if err != nil { t.Error("failed to get a random conn", err)