From 3873404e9acf94d4bd16e3722485529a6192b424 Mon Sep 17 00:00:00 2001
From: Kyle Carberry <kyle@coder.com>
Date: Tue, 19 Apr 2022 16:03:01 +0000
Subject: [PATCH] fix: Close TURN connections to resolve flake

---
 coderd/turnconn/turnconn_test.go | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/coderd/turnconn/turnconn_test.go b/coderd/turnconn/turnconn_test.go
index 346bfb2d420c1..6a8d0411cb7b1 100644
--- a/coderd/turnconn/turnconn_test.go
+++ b/coderd/turnconn/turnconn_test.go
@@ -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 {