Improve error message on replay data mismatch (#3386)

EnricoMi · web-flow · commit 927cf227ab83 · 2025-09-26T11:48:03.000+02:00
Follow-up on #3385.
diff --git a/tests/Connection.py b/tests/Connection.py
@@ -91,7 +91,8 @@ def testRecordAndReplay(replaying_connection_class, protocol, response_body, exp
     connection.getresponse.return_value = response
 
     # write mock response to buffer
-    RecordingMockConnection.setOpenFile(lambda slf, mode: file)
+    rdf = Framework.ReplayDataFile("string", file)
+    RecordingMockConnection.setOpenFile(lambda slf, mode: rdf)
     recording_connection = RecordingMockConnection(protocol, host, None, lambda *args, **kwds: connection)
     recording_connection.request(verb, url, None, headers)
     recording_connection.getresponse()
diff --git a/tests/Framework.py b/tests/Framework.py
@@ -61,7 +61,6 @@
 
 import base64
 import contextlib
-import io
 import json
 import os
 import traceback
@@ -140,7 +139,7 @@ def setOpenFile(func):
     def __init__(self, protocol, host, port, *args, **kwds):
         self.__file = self.__openFile("w")
         # write operations make the assumption that the file is not in binary mode
-        assert isinstance(self.__file, io.TextIOBase)
+        assert isinstance(self.__file, ReplayDataFile)
         self.__protocol = protocol
         self.__host = host
         self.__port = port
@@ -353,6 +352,9 @@ def __init__(self, filename: str, file):
     def __repr__(self) -> str:
         return f"{self.__filename}:{self.__line}"
 
+    def write(self, string: str):
+        self.__file.write(string)
+
     def readline(self) -> str:
         self.__line += 1
         line = self.__file.readline()
