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_flush(self):
sock = MockSocket(bytearray())
msgSock = MessageSocket(sock, None)
for res in msgSock.flush():
if res in (0, 1):
self.assertTrue(False, "Blocking flush")
else:
break
self.assertEqual(len(sock.sent), 0)
for res in msgSock.flush():
if res in (0, 1):
self.assertTrue(False, "Blocking flush")
else:
break
self.assertEqual(len(sock.sent), 0)
def test_queueMessage_with_conflicting_types(self):
sock = MockSocket(bytearray())
msgSock = MessageSocket(sock, None)
msgSock.version = (3, 3)
msg = Message(ContentType.handshake, bytearray(b'\xaa\xaa\xaa'))
for res in msgSock.queueMessage(msg):
if res in (0, 1):
self.assertTrue(False, "Blocking queue")
else:
break
self.assertEqual(len(sock.sent), 0)
msg = Message(ContentType.alert, bytearray(b'\x02\x01'))
for res in msgSock.queueMessage(msg):
if res in (0, 1):
self.assertTrue(False, "Blocking queue")
else:
break
self.assertEqual(len(sock.sent), 1)
self.assertEqual(bytearray(
b'\x16' +
b'\x03\x03' +
b'\x00\x03' +
b'\xaa'*3), sock.sent[0])
for res in msgSock.flush():
if res in (0, 1):
self.assertTrue(False, "Blocking flush")
else:
break
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_queueMessage(self):
sock = MockSocket(bytearray())
msgSocket = MessageSocket(sock, None)
msg = Message(ContentType.alert, bytearray(b'\xff\xbb'))
for res in msgSocket.queueMessage(msg):
if res in (0, 1):
self.assertTrue(False, "Blocking queue")
else:
break
self.assertEqual(len(sock.sent), 0)
msg = Message(ContentType.alert, bytearray(b'\xff\xaa'))
for res in msgSocket.queueMessage(msg):
if res in (0, 1):
self.assertTrue(False, "Blocking queue")
else:
break
self.assertEqual(len(sock.sent), 0)
for res in msgSocket.flush():
if res in (0, 1):
self.assertTrue(False, "Blocking flush")
else:
break
self.assertEqual(len(sock.sent), 1)
self.assertEqual(sock.sent[0], bytearray(
b'\x15' +
b'\x00\x00' +
b'\x00\x04' +
b'\xff\xbb' +
b'\xff\xaa'))