def test_webframe_to_bytes_non_fragmented_with_mask(self): # generating a mask is usually random which is not good for testing WebSocketFrame.generate_mask = Mock(return_value=525161) f = WebSocketFrame(message="TEST DATA", headers=WebSocketFrameHeaders( mask=WebSocketFrame.generate_mask(), mask_flag=1)) self.assertEqual(f.to_bytes(), b'\x81\x89\x00\x08\x03iTMP= LB=A')
def test_webframe_to_bytes_non_fragmented_no_mask(self): f = WebSocketFrame(message="TEST DATA") bts = f.to_bytes() self.assertEqual(bts, b'\x81\tTEST DATA')