From abd6db4a30359a595ed2d94db8a2ee329228c755 Mon Sep 17 00:00:00 2001
From: Toby Brain <tobias@modestmind.net>
Date: Wed, 23 Mar 2022 13:31:06 +1100
Subject: [PATCH] Close the listener after a single request

---
 tcp_server_test.go | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/tcp_server_test.go b/tcp_server_test.go
index 3a14c738..09254948 100644
--- a/tcp_server_test.go
+++ b/tcp_server_test.go
@@ -18,15 +18,12 @@ func WithListenServer(t *testing.T, test func(server string)) {
 	defer l.Close()
 
 	go func() {
-		for {
-			conn, err := l.Accept()
-			if err != nil {
-				fmt.Println("Failed to accept connection: ", err.Error())
-				continue
-			}
-
-			handleRequest(conn)
+		conn, err := l.Accept()
+		if err != nil {
+			t.Logf("Failed to accept connection: %s", err.Error())
 		}
+
+		handleRequest(conn)
 	}()
 
 	test(server)