private PrintWriter printWriter;
    private ByteArrayOutputStream outputStream;
    private ServletOutputStream servletOutputStream;

    @BeforeEach
    void setUp() throws Exception {
        // Setup response writers
        stringWriter = new StringWriter();
        printWriter = new PrintWriter(stringWriter);
        outputStream = new ByteArrayOutputStream();

    val message: String,
    val param: String?,
  ) {
    enum class Type {
      Handshake,
      Plaintext,
      Encrypted,
      Setup,
      Unknown,
    }

    val type: Type
      get() =
        when {
          message == "adding as trusted certificates" -> Type.Setup
          message == "Raw read" || message == "Raw write" -> Type.Encrypted

        // Test case: close() when opened and successful MsrpcSamrCloseHandle
        String server = "testServer";
        int access = 123;

        // Setup for successful constructor
        doNothing().when(mockHandle).sendrecv(any(MsrpcSamrConnect4.class));

        // Setup for successful close
        ArgumentCaptor<MsrpcSamrCloseHandle> closeCaptor = ArgumentCaptor.forClass(MsrpcSamrCloseHandle.class);
        doAnswer(invocation -> {

  private SettableFuture<String> future;
  private ListenableFutureTester tester;

  @Override
  protected void setUp() throws Exception {
    super.setUp();

    future = SettableFuture.create();
    tester = new ListenableFutureTester(future);
    tester.setUp();
  }

  public void testDefaultState() throws Exception {
    assertThrows(TimeoutException.class, () -> future.get(5, MILLISECONDS));
  }

            statusField.setAccessible(true);
            statusField.setInt(resp, status);
            Field availField = TransPeekNamedPipeResponse.class.getDeclaredField("available");
            availField.setAccessible(true);
            availField.setInt(resp, 123);
            return resp;

public class UpdateLabelJobTest extends UnitFessTestCase {

    private UpdateLabelJob updateLabelJob;

    @Override
    public void setUp() throws Exception {
        super.setUp();
        updateLabelJob = new UpdateLabelJob();

        // Setup mock components
        setupMockComponents();
    }

    private void setupMockComponents() {
        // Mock SearchEngineClient

      - name: 'Check out repository'
        uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
      - name: 'Set up JDKs'
        uses: actions/setup-java@dded0888837ed1f317902acf8a20df0ad188d165 # v5.0.0
        with:
          # For discussion, see the first setup-java block.
          # The publish-snapshot workflow doesn't run tests, so we don't have to care which version Maven would select for that step.
          java-version: 24

    @Mock
    private SmbFile mockFile1;

    @Mock
    private SmbFile mockFile2;

    private DirectoryLeaseManager directoryLeaseManager;
    private AutoCloseable mocks;

    @BeforeEach
    public void setUp() {
        mocks = MockitoAnnotations.openMocks(this);
        when(context.getConfig()).thenReturn(config);
        when(config.isDirectoryNotificationsEnabled()).thenReturn(false); // Disable for unit tests

    private PurgeLogJob purgeLogJob;

    @Override
    public void setUp() throws Exception {
        super.setUp();
        purgeLogJob = new PurgeLogJob();
    }

    // Test all services execute successfully
    public void test_execute_allSuccess() {
        // Setup tracking variables
        final boolean[] deleteCrawlingInfoCalled = { false };

  private static final int DELAY_MS = 50;
  private static final String RETURN_VALUE = "abc";

  private TimeLimiter timeLimiter;

  @Override
  protected void setUp() throws Exception {
    super.setUp();
    timeLimiter = new FakeTimeLimiter();
  }

  public void testCallWithTimeout_propagatesReturnValue() throws Exception {
    String result =