From: Tomas Krizek <tomas.krizek@nic.cz>
Date: Thu, 15 Nov 2018 12:33:20 +0000 (+0100)
Subject: pytests: fix utils::recv_answer to read only one DNS message at a time
X-Git-Tag: v3.2.0~18^2~35
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=e03194fc63a32b2916dd40e4039cc613398a3419;p=thirdparty%2Fknot-resolver.git

pytests: fix utils::recv_answer to read only one DNS message at a time
---

diff --git a/tests/pytests/utils.py b/tests/pytests/utils.py
index 17e5ffc4f..20218dfd5 100644
--- a/tests/pytests/utils.py
+++ b/tests/pytests/utils.py
@@ -22,11 +22,11 @@ def receive_answer(sock):
     answer_received_len = 0
     data_answer = b''
     while answer_received_len < answer_total_len:
-        data_chunk = sock.recv(2048)
+        data_chunk = sock.recv(answer_total_len - answer_received_len)
         if not data_chunk:
             return None
-        data_answer = data_answer + data_chunk
-        answer_received_len = answer_received_len + len(data_answer)
+        data_answer += data_chunk
+        answer_received_len += len(data_answer)
 
     return data_answer