def test_write(self):
msg = b'hello'
t = SocketTransport(loop, self.ssock, self.read, self.close)
t.write(msg)
loop.start(pyev.EVRUN_NOWAIT)
rmsg = self.csock.recv(len(msg))
self.assertEqual(rmsg, msg)
def __init__(self, loop, sock, address, protocol, server):
"""Create a server connection."""
self.loop = loop
self.sock = sock
self.address = address
self.protocol = protocol
self.server = server
self.transport = SocketTransport(self.loop, self.sock,
self.protocol.data, self.closed)
def test_close_buffered_write(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
count = 0
msg = b'hello'
while (t.write != t.buffered_write):
count += 1
t.write(msg)
t.write(msg)
t.close()
self.assertRaises(ConnectionClosed, t.write, msg)
def test_buffered_write(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
count = 0
msg = b'hello'
while (t.write != t.buffered_write):
count += 1
t.write(msg)
t.write(msg)
self.csock.recv(count * len(msg))
loop.start(pyev.EVRUN_NOWAIT)
self.assertTrue(t.write == t.unbuffered_write)
def __init__(self, loop, sock, addr, protocol, client):
"""Create a client connection."""
self.loop = loop
self.sock = sock
self.addr = addr
self.protocol = protocol
self.client = client
logger.debug('making transport')
self.transport = SocketTransport(self.loop, self.sock,
self.protocol.data, self.closed)
logger.debug('protocol.make_connection')
self.protocol.make_connection(self.transport, self.addr)
logger.debug('transport.start()')
self.transport.start()
logger.debug('transport started')
def test_stop(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
t.stop()
def test_closed_stop(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
t.close()
self.assertRaises(ConnectionClosed, t.stop)
def test_closed_write(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
t.close()
self.assertRaises(ConnectionClosed, t.write, ("hello"))
def test_close(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
t.close()
self.assertTrue(t.closed)
self.assertTrue(isinstance(self.reason, ConnectionClosed))
def test_creation(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
def test_error(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
self.csock.close()
t.write(b'hello')
loop.start(pyev.EVRUN_NOWAIT)
self.assertTrue(self.reason is not None)
def test_read(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
t.start()
self.csock.send(b'hello')
loop.start(pyev.EVRUN_NOWAIT)
self.assertEqual(self.data, b'hello')
def test_overflow_write(self):
t = SocketTransport(loop, self.ssock, self.read, self.close)
self.assertRaises(BufferOverflowError, t.write,
bytes([1 for x in range(0, 1024 * 1024)]))
