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()
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)
def test_connected(self): connection = self.new_connection() connection.protocol.connection_made(unittest.mock.Mock()) self.assertFalse(connection.closed) connection.close()