def test_mask_payload(self): EXPECTED = ( # (Payload, Mask, Result) (b"Hello", b'\xb4\x8dA\xa2', b'\xfc\xe8-\xce\xdb'), (b"World", b'K\xb5n^', b'\x1c\xda\x1c2/'), ) for payload, mask, result in EXPECTED: self.assertEqual(FrameBuilder._mask_payload(payload, mask), result)