def test_flushBlocking(self):
sock = MockSocket(bytearray())
msgSock = MessageSocket(sock, None)
msgSock.flushBlocking()
self.assertEqual(len(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])
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])
