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 17e5ffc4fce0ac0058dfcd6f1fa3bae4aaf7652e..20218dfd5e1cc6ed328df19366d5ffd423d5c9d4 100644 (file)
--- 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