def test_io_receive_raises_socket_timeout(self):
connection = FakeConnection()
io = IO(connection.parameters)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.timeout('timeout')
io._receive()
self.assertIsNone(connection.check_for_errors())
def test_io_receive_does_not_raise_on_block(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.error(EWOULDBLOCK)
io._receive()
self.assertIsNone(connection.check_for_errors())
def test_io_receive_does_not_raise_on_block(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.error(EWOULDBLOCK)
io._receive()
self.assertIsNone(connection.check_for_errors())
def test_io_receive_raises_ssl_want_read_error(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = compatibility.SSLWantReadError()
io._receive()
self.assertIsNone(connection.check_for_errors())
def test_io_receive_raises_socket_error(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.error('travis-ci')
io._receive()
self.assertRaisesRegexp(AMQPConnectionError, 'travis-ci',
connection.check_for_errors)
def test_io_receive_raises_socket_error(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.error('travis-ci')
io._receive()
self.assertRaisesRegexp(
AMQPConnectionError,
'travis-ci',
connection.check_for_errors
)
def test_io_simple_receive_when_socket_not_set(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
self.assertFalse(io.use_ssl)
self.assertEqual(io._receive(), bytes())
self.assertRaisesRegexp(AMQPConnectionError, 'connection/socket error',
connection.check_for_errors)
def test_io_simple_receive_when_socket_not_set(self):
connection = FakeConnection()
io = IO(connection.parameters, exceptions=connection.exceptions)
self.assertFalse(io.use_ssl)
self.assertEqual(io._receive(), bytes())
self.assertRaisesRegexp(
AMQPConnectionError,
'connection/socket error',
connection.check_for_errors
)
def test_io_receive_raises_socket_timeout(self):
connection = FakeConnection()
io = IO(connection.parameters)
io.socket = mock.Mock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.timeout('timeout')
io._receive()
