testing: Fix flaky test

bep · Apr 1, 2024 · a4d48d7 · a4d48d7
1 parent 474a323
commit a4d48d7
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/client.go b/client.go
@@ -227,7 +227,6 @@ func (c *ClientRaw) Close() error {
 	if c == nil {
 		return nil
 	}
-	defer close(c.Messages)
 
 	c.sendMu.Lock()
 	defer c.sendMu.Unlock()
@@ -238,10 +237,8 @@ func (c *ClientRaw) Close() error {
 		return ErrShutdown
 	}
 	c.closing = true
-
-	err := c.conn.Close()
-
-	return err
+	close(c.Messages)
+	return c.conn.Close()
 }
 
 // Execute sends body to the server and sends any messages to the messages channel.