private static void rethrow(ExecutionException e) throws ExecutionException {
    Throwable wrapper = e.getCause();
    if (wrapper instanceof WrapperException) {
      Throwable cause = wrapper.getCause();
      if (cause instanceof RuntimeException) {
        throw (RuntimeException) cause;
      } else if (cause instanceof Error) {
        throw (Error) cause;
      }
    }
    throw e;
  }

  @Override

public class PrefixQueryCommandTest extends QueryTestBase {
    private PrefixQueryCommand queryCommand;
    private QueryFieldConfig queryFieldConfig;

    @Override
    protected void setUpChild() throws Exception {
        // Get the queryProcessor and queryFieldConfig that were registered in parent class
        queryProcessor = ComponentUtil.getComponent("queryProcessor");

        public ModelSource resolveModel(Parent parent) throws UnresolvableModelException {
            return null;
        }

        @Override
        public ModelSource resolveModel(Dependency dependency) throws UnresolvableModelException {
            return null;
        }

        @Override
        public void addRepository(Repository repository) throws InvalidRepositoryException {}

        @Override

        runner.ensureYellow();
    }

    @AfterClass
    public static void afterClass() throws Exception {
        if (runner != null) {
            runner.close();
            runner.clean();
        }
    }

    @Before
    public void before() throws Exception {
        runner.admin().indices().prepareDelete("_all").execute().actionGet();
        runner.refresh();

            return true;
        }

        @Override
        public void process(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
                throws IOException, ServletException {
            throw new ServletException("Test error");
        }
    }

    private static class StatusCodeWebApiManager implements WebApiManager {
        private final int statusCode;

        }
    }

    @Test
    void testReadDataWireFormatWithEmptyData() throws SMBProtocolDecodingException {
        byte[] buffer = new byte[100];
        int result = response.readDataWireFormat(buffer, 0, 0);
        assertEquals(0, result);
    }

    @Test
    void testReadDataWireFormatAtBufferBoundary() throws SMBProtocolDecodingException {
        // Test with data exactly matching output buffer size

     * @return the Type 3 message to send to the server
     * @throws GeneralSecurityException if a cryptographic error occurs
     * @throws CIFSException if a CIFS protocol error occurs
     */
    protected Type3Message createType3Message(final Type2Message msg2) throws GeneralSecurityException, CIFSException {
        if (this.auth instanceof NtlmNtHashAuthenticator) {

    void connectWrapException_wrapsToSmbException(SmbResourceLocatorImpl loc, Exception thrown) throws Exception {
        // Arrange: spy on instance to force connect() failure with specific exception
        CIFSContext ctx = mock(CIFSContext.class);
        SmbTreeConnectionTrace trace = Mockito.spy(new SmbTreeConnectionTrace(ctx));
        doThrow(thrown).when(trace).connect(loc);

    }
  }

  @Throws(IOException::class)
  fun flush() {
    withLock {
      if (closed) throw IOException("closed")
      sink.flush()
    }
  }

  @Throws(IOException::class)
  fun rstStream(
    streamId: Int,
    errorCode: ErrorCode,
  ) {
    withLock {
      if (closed) throw IOException("closed")
      require(errorCode.httpCode != -1)

        assertDoesNotThrow(() -> cache.resolveSids(handle, policy, sids));
    }

    @Test
    @DisplayName("resolveSids(DcerpcHandle,...) throws SmbException on unexpected retval")
    void resolveSids_throwsOnUnexpectedRetval() throws Exception {
        SIDCacheImpl cache = new SIDCacheImpl(mock(CIFSContext.class));
        DcerpcHandle handle = mock(DcerpcHandle.class);