diff --git a/mocket/mocket.py b/mocket/mocket.py index 9f33d158..93530c42 100644 --- a/mocket/mocket.py +++ b/mocket/mocket.py @@ -561,7 +561,7 @@ def data(self): return self response_index = 0 - request_cls = str + request_cls = bytes response_cls = Response responses = None _served = None diff --git a/tests/main/test_mocket.py b/tests/main/test_mocket.py index f0068590..67ad337f 100644 --- a/tests/main/test_mocket.py +++ b/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"]))