class TestNailgunClient(unittest.TestCase): def setUp(self): self.nailgun_client = NailgunClient() @mock.patch('pants.java.nailgun_client.RecvBufferedSocket', **PATCH_OPTS) def test_try_connect(self, mock_socket_cls): mock_socket = mock.Mock() mock_socket_cls.return_value = mock_socket self.assertEquals(self.nailgun_client.try_connect(), mock_socket) self.assertEquals(mock_socket_cls.call_count, 1) mock_socket.connect.assert_called_once_with( (NailgunClient.DEFAULT_NG_HOST, NailgunClient.DEFAULT_NG_PORT) ) @mock.patch('pants.java.nailgun_client.RecvBufferedSocket', **PATCH_OPTS) def test_try_connect_socket_error(self, mock_socket_cls): mock_socket = mock.Mock() mock_socket.connect.side_effect = socket.error() mock_socket_cls.return_value = mock_socket with self.assertRaises(NailgunClient.NailgunConnectionError): self.nailgun_client.try_connect() @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute(self, mock_session, mock_try_connect): self.nailgun_client.execute('test') self.assertEquals(mock_try_connect.call_count, 1) self.assertEquals(mock_session.call_count, 1) @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute_propagates_connection_error_on_connect(self, mock_session, mock_try_connect): mock_try_connect.side_effect = NailgunClient.NailgunConnectionError('oops') with self.assertRaises(NailgunClient.NailgunConnectionError): self.nailgun_client.execute('test') @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute_socketerror_on_execute(self, mock_session, mock_try_connect): mock_session.return_value.execute.side_effect = socket.error('oops') with self.assertRaises(NailgunClient.NailgunError): self.nailgun_client.execute('test') @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute_protocolerror_on_execute(self, mock_session, mock_try_connect): mock_session.return_value.ProtocolError = NailgunProtocol.ProtocolError mock_session.return_value.execute.side_effect = NailgunProtocol.ProtocolError('oops') with self.assertRaises(NailgunClient.NailgunError): self.nailgun_client.execute('test') def test_repr(self): self.assertIsNotNone(repr(self.nailgun_client)) @mock.patch('os.kill', **PATCH_OPTS) def test_send_control_c(self, mock_kill): self.nailgun_client._session = mock.Mock(remote_pid=31337) self.nailgun_client.send_control_c() mock_kill.assert_called_once_with(31337, signal.SIGINT) @mock.patch('os.kill', **PATCH_OPTS) def test_send_control_c_noop_none(self, mock_kill): self.nailgun_client._session = None self.nailgun_client.send_control_c() mock_kill.assert_not_called() @mock.patch('os.kill', **PATCH_OPTS) def test_send_control_c_noop_nopid(self, mock_kill): self.nailgun_client._session = mock.Mock(remote_pid=None) self.nailgun_client.send_control_c() mock_kill.assert_not_called()
class TestNailgunClient(unittest.TestCase): def setUp(self): self.nailgun_client = NailgunClient() @mock.patch('pants.java.nailgun_client.RecvBufferedSocket', **PATCH_OPTS) def test_try_connect(self, mock_socket_cls): mock_socket = mock.Mock() mock_socket_cls.return_value = mock_socket self.assertEquals(self.nailgun_client.try_connect(), mock_socket) self.assertEquals(mock_socket_cls.call_count, 1) mock_socket.connect.assert_called_once_with( (NailgunClient.DEFAULT_NG_HOST, NailgunClient.DEFAULT_NG_PORT)) @mock.patch('pants.java.nailgun_client.RecvBufferedSocket', **PATCH_OPTS) def test_try_connect_socket_error(self, mock_socket_cls): mock_socket = mock.Mock() mock_socket.connect.side_effect = socket.error() mock_socket_cls.return_value = mock_socket with self.assertRaises(NailgunClient.NailgunConnectionError): self.nailgun_client.try_connect() @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute(self, mock_session, mock_try_connect): self.nailgun_client.execute('test') self.assertEquals(mock_try_connect.call_count, 1) self.assertEquals(mock_session.call_count, 1) @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute_propagates_connection_error_on_connect( self, mock_session, mock_try_connect): mock_try_connect.side_effect = NailgunClient.NailgunConnectionError( '127.0.0.1:31337', 31337, Exception('oops'), None) with self.assertRaises(NailgunClient.NailgunConnectionError): self.nailgun_client.execute('test') @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute_socketerror_on_execute(self, mock_session, mock_try_connect): mock_session.return_value.execute.side_effect = socket.error('oops') with self.assertRaises(NailgunClient.NailgunError): self.nailgun_client.execute('test') @mock.patch.object(NailgunClient, 'try_connect', **PATCH_OPTS) @mock.patch('pants.java.nailgun_client.NailgunClientSession', **PATCH_OPTS) def test_execute_protocolerror_on_execute(self, mock_session, mock_try_connect): mock_session.return_value.ProtocolError = NailgunProtocol.ProtocolError mock_session.return_value.execute.side_effect = NailgunProtocol.ProtocolError( 'oops') with self.assertRaises(NailgunClient.NailgunError): self.nailgun_client.execute('test') def test_repr(self): self.assertIsNotNone(repr(self.nailgun_client)) @mock.patch('os.kill', **PATCH_OPTS) def test_send_control_c(self, mock_kill): self.nailgun_client._session = mock.Mock(remote_pid=31337) self.nailgun_client.send_control_c() mock_kill.assert_called_once_with(31337, signal.SIGINT) @mock.patch('os.kill', **PATCH_OPTS) def test_send_control_c_noop_none(self, mock_kill): self.nailgun_client._session = None self.nailgun_client.send_control_c() mock_kill.assert_not_called() @mock.patch('os.kill', **PATCH_OPTS) def test_send_control_c_noop_nopid(self, mock_kill): self.nailgun_client._session = mock.Mock(remote_pid=None) self.nailgun_client.send_control_c() mock_kill.assert_not_called()