From f87ef68faeed587992f94f34e40d25b1c58ecb9f Mon Sep 17 00:00:00 2001
From: "Roger A. Light" <roger@atchoo.org>
Date: Wed, 16 Sep 2020 13:46:40 +0100
Subject: [PATCH] Detect closed socket in tests.

---
 test/mosq_test.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/mosq_test.py b/test/mosq_test.py
index 0dcb82e2..91bba79d 100644
--- a/test/mosq_test.py
+++ b/test/mosq_test.py
@@ -119,7 +119,10 @@ def receive_unordered(sock, recv1_packet, recv2_packet, error_string):
     expected2 = recv2_packet + recv1_packet
     recvd = b''
     while len(recvd) < len(expected1):
-        recvd += sock.recv(1)
+        r = sock.recv(1)
+        if len(r) == 0:
+            raise ValueError(error_string)
+        recvd += r
 
     if recvd == expected1 or recvd == expected2:
         return