Close the listener after a single request

everoute · Mar 23, 2022 · abd6db4 · abd6db4
1 parent 463724b
commit abd6db4
Showing 1 changed file with 5 additions and 8 deletions.
diff --git 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)