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_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()