From b769faaec5f7bbf7e0891fef73a95bb4ad58abcb Mon Sep 17 00:00:00 2001
From: David Zhao <david@davidzhao.com>
Date: Wed, 2 Jun 2021 10:20:11 -0700
Subject: [PATCH] minor: better EOF detection

---
 test/client/client.go | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/test/client/client.go b/test/client/client.go
index ee750f262..fe6b6cbcc 100644
--- a/test/client/client.go
+++ b/test/client/client.go
@@ -197,7 +197,9 @@ func (c *RTCClient) Run() error {
 	// run the session
 	for {
 		res, err := c.ReadResponse()
-		if err != nil {
+		if errors.Is(io.EOF, err) {
+			return nil
+		} else if err != nil {
 			return err
 		}
 		switch msg := res.Message.(type) {
@@ -287,11 +289,6 @@ func (c *RTCClient) Run() error {
 			c.lock.Unlock()
 		}
 	}
-
-	if err != io.EOF {
-		return err
-	}
-	return nil
 }
 
 func (c *RTCClient) WaitUntilConnected() error {