// Set up mock provider to return TCP memory regions
        when(mockProvider.registerMemory(any(ByteBuffer.class), any())).thenAnswer(invocation -> {
            ByteBuffer buffer = invocation.getArgument(0);
            EnumSet<RdmaAccess> access = invocation.getArgument(1);
            return new TcpMemoryRegion(buffer, access);
        });

        bufferManager = new RdmaBufferManager(mockProvider);
    }

        when(mockClearScrollBuilder.addScrollId(any(String.class))).thenReturn(mockClearScrollBuilder);
        doAnswer(invocation -> {
            ActionListener<ClearScrollResponse> listener = invocation.getArgument(0);
            listener.onResponse(mock(ClearScrollResponse.class));
            return null;
        }).when(mockClearScrollBuilder).execute(any(ActionListener.class));

        // Mock transport.connect() to succeed
        doNothing().when(transport).connect();

        // Setup response for tree connect
        doAnswer(invocation -> {
            ServerMessageBlock request = invocation.getArgument(0);
            ServerMessageBlock response = invocation.getArgument(1);
            if (request instanceof SmbComTreeConnectAndX && response instanceof SmbComTreeConnectAndXResponse) {

  private final int bufferSize;

  /** Number of bytes processed per process() invocation. */
  private final int chunkSize;

  /**
   * Constructor for use by subclasses. This hasher instance will process chunks of the specified
   * size.
   *
   * @param chunkSize the number of bytes available per {@link #process(ByteBuffer)} invocation;
   *     must be at least 4
   */

        if (lmCompatibility == 3 || lmCompatibility == 4 || lmCompatibility == 5) {
            // For LM compatibility 3-5, only 16 bytes are used
            doAnswer(invocation -> {
                byte[] dest = invocation.getArgument(2);
                int offset = invocation.getArgument(3);
                System.arraycopy(new byte[16], 0, dest, offset, 16);
                return null;

        // Only mark the sessionSetup response as received, not the actual request/response
        doAnswer(invocation -> {
            ServerMessageBlock request = invocation.getArgument(0);
            ServerMessageBlock response = invocation.getArgument(1);
            // Only mark session setup responses as received, not our test request

        lenient().when(servletConfig.getInitParameterNames()).thenReturn(Collections.enumeration(initParams.keySet()));
        lenient().when(servletConfig.getInitParameter(anyString())).thenAnswer(invocation -> initParams.get(invocation.getArgument(0)));

        // Mock HTTP method for request - this is required for HttpServlet.service()
        lenient().when(request.getMethod()).thenReturn("GET");

            when(mockMd5.digest()).thenReturn(innerDigest);
            when(mockMd5.digest(buffer, 0, expectedOutput.length)).thenAnswer(invocation -> {
                byte[] b = invocation.getArgument(0);
                int off = invocation.getArgument(1);
                int l = invocation.getArgument(2);
                System.arraycopy(expectedOutput, 0, b, off, l);
                return l;
            });

  protected AbstractService() {}

  /**
   * This method is called by {@link #startAsync} to initiate service startup. The invocation of
   * this method should cause a call to {@link #notifyStarted()}, either during this method's run,
   * or after it has returned. If startup fails, the invocation should cause a call to {@link
   * #notifyFailed(Throwable)} instead.
   *

  protected AbstractService() {}

  /**
   * This method is called by {@link #startAsync} to initiate service startup. The invocation of
   * this method should cause a call to {@link #notifyStarted()}, either during this method's run,
   * or after it has returned. If startup fails, the invocation should cause a call to {@link
   * #notifyFailed(Throwable)} instead.
   *