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 run(self):
for chunk_type, payload in NailgunProtocol.iter_chunks(
self._socket, return_bytes=True):
if chunk_type == ChunkType.STDIN:
self._write_handle.write(payload)
self._write_handle.flush()
elif chunk_type == ChunkType.STDIN_EOF:
self._write_handle.close()
break
else:
self._try_close()
raise NailgunProtocol.ProtocolError(
'received unexpected chunk {} -> {}: closing.'.format(
chunk_type, payload))
def run(self):
try:
for chunk_type, payload in NailgunProtocol.iter_chunks(self._socket, return_bytes=True):
if self.is_stopped:
return
if chunk_type == ChunkType.STDIN:
self._write_handle.write(payload)
self._write_handle.flush()
elif chunk_type == ChunkType.STDIN_EOF:
return
else:
raise NailgunProtocol.ProtocolError(
'received unexpected chunk {} -> {}'.format(chunk_type, payload)
)
finally:
self._write_handle.close()
