def test_io_receive_raises_socket_error(self):
connection = FakeConnection()
io = IO(connection.parameters)
io._exceptions = []
io.socket = MagicMock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.error('error')
io._receive()
self.assertIsInstance(io._exceptions[0], AMQPConnectionError)
def test_io_receive_raises_socket_error(self):
connection = FakeConnection()
io = IO(connection.parameters)
io._exceptions = []
io.socket = MagicMock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.error('error')
io._receive()
self.assertIsInstance(io._exceptions[0], AMQPConnectionError)
def test_io_simple_receive(self):
connection = FakeConnection()
io = IO(connection.parameters)
io.socket = MagicMock(name='socket', spec=socket.socket)
io.socket.recv.return_value = '12345'
self.assertEqual(io._receive(), '12345')
def test_io_simple_ssl_receive(self):
connection = FakeConnection()
connection.parameters['ssl'] = True
io = IO(connection.parameters)
self.assertTrue(io.use_ssl)
if hasattr(ssl, 'SSLObject'):
io.socket = Mock(name='socket', spec=ssl.SSLObject)
elif hasattr(ssl, 'SSLSocket'):
io.socket = Mock(name='socket', spec=ssl.SSLSocket)
io.socket.read.return_value = '12345'
self.assertEqual(io._receive(), '12345')
def test_io_simple_ssl_receive(self):
connection = FakeConnection()
connection.parameters['ssl'] = True
io = IO(connection.parameters)
self.assertTrue(io.use_ssl)
if hasattr(ssl, 'SSLObject'):
io.socket = Mock(name='socket', spec=ssl.SSLObject)
elif hasattr(ssl, 'SSLSocket'):
io.socket = Mock(name='socket', spec=ssl.SSLSocket)
io.socket.read.return_value = '12345'
self.assertEqual(io._receive(), '12345')
def test_io_receive_raises_socket_timeout(self):
connection = FakeConnection()
io = IO(connection.parameters)
io.socket = MagicMock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.timeout('timeout')
io._receive()
def test_io_receive_raises_socket_timeout(self):
connection = FakeConnection()
io = IO(connection.parameters)
io.socket = MagicMock(name='socket', spec=socket.socket)
io.socket.recv.side_effect = socket.timeout('timeout')
io._receive()
