fix: Close TURN connections to resolve flake (#1079)

This commit is contained in:
Kyle Carberry
2022-04-19 11:14:55 -05:00
committed by GitHub
parent 89aa39b5c8
commit 04985a1754

View File

@ -44,6 +44,9 @@ func TestTURNConn(t *testing.T) {
Logger: logger.Named("client"),
})
require.NoError(t, err)
defer func() {
_ = client.Close()
}()
serverDialer, serverTURN := net.Pipe()
turnServer.Accept(serverTURN, &net.TCPAddr{
@ -62,6 +65,9 @@ func TestTURNConn(t *testing.T) {
Logger: logger.Named("server"),
})
require.NoError(t, err)
defer func() {
_ = server.Close()
}()
exchange(t, client, server)
_, err = client.Ping()
@ -71,12 +77,7 @@ func TestTURNConn(t *testing.T) {
func exchange(t *testing.T, client, server *peer.Conn) {
var wg sync.WaitGroup
wg.Add(2)
t.Cleanup(func() {
_ = client.Close()
_ = server.Close()
wg.Wait()
})
t.Cleanup(wg.Wait)
go func() {
defer wg.Done()
for {