def test_writer_send(self): transport = BytesIO() ww = WebSocketWriter(transport) ww.send(b'Hello') transport.seek(0) self.assertEqual(transport.read(), b'\x82\x05Hello') transport = BytesIO() ww = WebSocketWriter(transport) ww.send('пы') transport.seek(0) self.assertEqual(transport.read(), b'\x81\x04\xd0\xbf\xd1\x8b')
def test_writer_close(self): transport = BytesIO() ww = WebSocketWriter(transport) ww.close() transport.seek(0) self.assertEqual(transport.read(), FrameBuilder.close(masked=False))