def test_queueMessage_with_conflicting_types_and_blocking_socket(self):
sock = MockSocket(bytearray(), blockEveryOther=True)
sock.blockWrite = True
msgSock = MessageSocket(sock, None)
msgSock.version = (3, 3)
msg = Message(ContentType.handshake, bytearray(b'\xaa\xaa\xaa'))
blocked = False
for res in msgSock.queueMessage(msg):
if res in (0, 1):
blocked = True
else:
break
# no write so no blocking
self.assertFalse(blocked)
self.assertEqual(len(sock.sent), 0)
msg = Message(ContentType.alert, bytearray(b'\x02\x01'))
blocked = False
for res in msgSock.queueMessage(msg):
if res in (0, 1):
blocked = True
else:
break
# blocked once, so one write
self.assertTrue(blocked)
self.assertEqual(len(sock.sent), 1)
self.assertEqual(bytearray(
b'\x16' +
b'\x03\x03' +
b'\x00\x03' +
b'\xaa'*3), sock.sent[0])
sock.blockWrite = True
blocked = False
for res in msgSock.flush():
if res in (0, 1):
blocked = True
else:
break
# blocked once, so one write
self.assertTrue(blocked)
self.assertEqual(len(sock.sent), 2)
self.assertEqual(bytearray(
b'\x15' +
b'\x03\x03' +
b'\x00\x02' +
b'\x02\x01'), sock.sent[1])
def test_flushBlocking_with_data(self):
sock = MockSocket(bytearray(), blockEveryOther=True)
sock.blockWrite = True
msgSock = MessageSocket(sock, None)
msgSock.version = (3, 3)
msg = Message(ContentType.handshake, bytearray(b'\xaa\xaa\xaa'))
msgSock.queueMessageBlocking(msg)
self.assertEqual(len(sock.sent), 0)
msgSock.flushBlocking()
self.assertEqual(len(sock.sent), 1)
self.assertEqual(
bytearray(b'\x16' + b'\x03\x03' + b'\x00\x03' + b'\xaa' * 3),
sock.sent[0])
def test_flushBlocking_with_data(self):
sock = MockSocket(bytearray(), blockEveryOther=True)
sock.blockWrite = True
msgSock = MessageSocket(sock, None)
msgSock.version = (3, 3)
msg = Message(ContentType.handshake, bytearray(b'\xaa\xaa\xaa'))
msgSock.queueMessageBlocking(msg)
self.assertEqual(len(sock.sent), 0)
msgSock.flushBlocking()
self.assertEqual(len(sock.sent), 1)
self.assertEqual(bytearray(
b'\x16' +
b'\x03\x03' +
b'\x00\x03' +
b'\xaa'*3), sock.sent[0])
