@@ -199,17 +199,17 @@ def test_handle_read__incomplete(self):
199199 Verify that handle_read() processes incomplete messages properly.
200200 """
201201 self .obj_ut .process_msg = Mock ()
202- self .assertEqual (self .obj_ut ._iobuf .getvalue (), '' ) # buf starts empty
202+ self .assertEqual (self .obj_ut ._iobuf .getvalue (), b '' ) # buf starts empty
203203 # incomplete header
204- self .obj_ut ._iobuf .write ('\x84 \x00 \x00 \x00 \x00 ' )
204+ self .obj_ut ._iobuf .write (b '\x84 \x00 \x00 \x00 \x00 ' )
205205 self .obj_ut .handle_read ()
206- self .assertEqual (self .obj_ut ._iobuf .getvalue (), '\x84 \x00 \x00 \x00 \x00 ' )
206+ self .assertEqual (self .obj_ut ._iobuf .getvalue (), b '\x84 \x00 \x00 \x00 \x00 ' )
207207
208208 # full header, but incomplete body
209- self .obj_ut ._iobuf .write ('\x00 \x00 \x00 \x15 ' )
209+ self .obj_ut ._iobuf .write (b '\x00 \x00 \x00 \x15 ' )
210210 self .obj_ut .handle_read ()
211211 self .assertEqual (self .obj_ut ._iobuf .getvalue (),
212- '\x84 \x00 \x00 \x00 \x00 \x00 \x00 \x00 \x15 ' )
212+ b '\x84 \x00 \x00 \x00 \x00 \x00 \x00 \x00 \x15 ' )
213213 self .assertEqual (self .obj_ut ._current_frame .end_pos , 30 )
214214
215215 # verify we never attempted to process the incomplete message
@@ -220,14 +220,14 @@ def test_handle_read__fullmessage(self):
220220 Verify that handle_read() processes complete messages properly.
221221 """
222222 self .obj_ut .process_msg = Mock ()
223- self .assertEqual (self .obj_ut ._iobuf .getvalue (), '' ) # buf starts empty
223+ self .assertEqual (self .obj_ut ._iobuf .getvalue (), b '' ) # buf starts empty
224224
225225 # write a complete message, plus 'NEXT' (to simulate next message)
226226 # assumes protocol v3+ as default Connection.protocol_version
227- body = 'this is the drum roll'
228- extra = 'NEXT'
227+ body = b 'this is the drum roll'
228+ extra = b 'NEXT'
229229 self .obj_ut ._iobuf .write (
230- '\x84 \x01 \x00 \x02 \x03 \x00 \x00 \x00 \x15 ' + body + extra )
230+ b '\x84 \x01 \x00 \x02 \x03 \x00 \x00 \x00 \x15 ' + body + extra )
231231 self .obj_ut .handle_read ()
232232 self .assertEqual (self .obj_ut ._iobuf .getvalue (), extra )
233233 self .obj_ut .process_msg .assert_called_with (
0 commit comments