MocketEntry.request_class str vs bytes (#177)

* Change request_cls to prevent unintended bytes -> str conversion (#176)
* Use type from `.compat`.

Co-authored-by: Michael Lazar <[email protected]>

Author: mindflayer

mocket/mocket.py

@@ -548,9 +548,7 @@ def get_truesocket_recording_dir(cls):
     @classmethod
     def assert_fail_if_entries_not_served(cls):
         """Mocket checks that all entries have been served at least once."""
-        if not all(
-            entry._served for entry in itertools.chain(*cls._entries.values())
-        ):
+        if not all(entry._served for entry in itertools.chain(*cls._entries.values())):
             raise AssertionError("Some Mocket entries have not been served")
 
 
@@ -561,7 +559,7 @@ def data(self):
             return self
 
     response_index = 0
-    request_cls = str
+    request_cls = byte_type
     response_cls = Response
     responses = None
     _served = None

tests/main/test_mocket.py

@@ -94,6 +94,18 @@ def test_raise_exception(self):
         with self.assertRaises(IOError):
             entry.get_response()
 
+    def test_collect_last_request(self):
+        addr = ("localhost", 80)
+
+        entry = MocketEntry(addr, True)
+        Mocket.register(entry)
+        with Mocketizer():
+            _so = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            _so.connect(addr)
+            _so.sendall(b"data\r\n")
+            _so.close()
+            self.assertEqual(Mocket.last_request(), b"data\r\n")
+
     def test_subsequent_recv_requests_have_correct_length(self):
         addr = ("localhost", 80)
         Mocket.register(MocketEntry(addr, [b"Long payload", b"Short"]))