async def test_close_closed(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connection.close()
connector.close_connection = unittest.mock.Mock()
connection.close()
self.assertFalse(connector.close_connection.called)
async def test_close_closed(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connection.close()
connector.close_connection = unittest.mock.Mock()
connection.close()
self.assertFalse(connector.close_connection.called)
def test_repr_connected(self):
connection = self.new_connection()
transport = unittest.mock.Mock()
transport._sock.getsockname.return_value = ('127.0.0.1', 4242)
transport._sock.getpeername.return_value = ('10.5.0.45', 4242)
connection.protocol.connection_made(transport)
self.assertEqual('<Connection@{:x}: 127.0.0.1:4242 -> 10.5.0.45:4242>'
''.format(id(connection)),
repr(connection))
connection.close()
async def test_send(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connection.protocol.send = unittest.mock.Mock(
wraps=connection.protocol.send)
await connection.send(b'...', None)
self.assertTrue(connection.protocol.send.called)
connection.close()
async def test_recv(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connection.protocol.recv = unittest.mock.Mock(
wraps=connection.protocol.recv)
await connection.recv()
self.assertTrue(connection.protocol.recv.called)
connection.close()
async def test_send(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connection.protocol.send = unittest.mock.Mock(
wraps=connection.protocol.send)
await connection.send(b'...', None)
self.assertTrue(connection.protocol.send.called)
connection.close()
async def test_recv(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connection.protocol.recv = unittest.mock.Mock(
wraps=connection.protocol.recv)
await connection.recv()
self.assertTrue(connection.protocol.recv.called)
connection.close()
def test_repr_connected(self):
connection = self.new_connection()
transport = unittest.mock.Mock()
transport._sock.getsockname.return_value = ('127.0.0.1', 4242)
transport._sock.getpeername.return_value = ('10.5.0.45', 4242)
connection.protocol.connection_made(transport)
self.assertEqual(
'<Connection@{:x}: 127.0.0.1:4242 -> 10.5.0.45:4242>'
''.format(id(connection)), repr(connection))
connection.close()
async def test_close(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connector.close_connection = unittest.mock.Mock(
wraps=connector.close_connection)
protocol_close = connection.protocol.close = unittest.mock.Mock(
wraps=connection.protocol.close)
connection.close()
self.assertTrue(connector.close_connection.called)
self.assertTrue(protocol_close.called)
self.assertIsNone(connection.protocol)
self.assertTrue(connection.closed)
async def test_close(self):
connector = self.new_connector()
address = aioppspp.connection.Address('0.0.0.0', 0)
connection = await connector.connect(address)
connection.protocol.connection_made(unittest.mock.Mock())
connector.close_connection = unittest.mock.Mock(
wraps=connector.close_connection)
protocol_close = connection.protocol.close = unittest.mock.Mock(
wraps=connection.protocol.close)
connection.close()
self.assertTrue(connector.close_connection.called)
self.assertTrue(protocol_close.called)
self.assertIsNone(connection.protocol)
self.assertTrue(connection.closed)
def test_connected(self):
connection = self.new_connection()
connection.protocol.connection_made(unittest.mock.Mock())
self.assertFalse(connection.closed)
connection.close()
def test_connected(self):
connection = self.new_connection()
connection.protocol.connection_made(unittest.mock.Mock())
self.assertFalse(connection.closed)
connection.close()
