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')