}

    @Test
    void testGetException() {
        DefaultProblem problem = new DefaultProblem(null, null, null, -1, -1, null);
        assertNull(problem.getException());

        Exception e = new Exception();
        problem = new DefaultProblem(null, null, null, -1, -1, e);
        assertSame(e, problem.getException());
    }

    @Test
    void testGetSource() {

     * @param params The map of parameters.
     */
    @Override
    public void setInitParameterMap(final Map<String, Object> params) {
        this.params.putAll(params);
    }

    /**
     * Initializes the parameter map of the wrapped factory with the parameters stored in this wrapper.
     */
    public void initParameterMap() {
        factory.setInitParameterMap(params);
    }

    /**

    
    public ChannelInfo selectChannel(SMBMessage message) {
        return loadBalancer.selectChannel(message);
    }
    
    public void handleChannelFailure(ChannelInfo channel, Exception error) {
        failover.handleFailure(channel, error);
    }
    
    private void performHealthCheck() {
        for (ChannelInfo channel : channels.values()) {
            if (channel.getIdleTime() > 60000) {  // 1 minute idle

        runner.ensureYellow();
    }

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

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

            return false;
        }
    }

    @Test
    public void testUnicodeConfiguration() throws Exception {
        // Test Unicode configuration is enabled by default
        assertTrue(cfg.isUseUnicode(), "Unicode should be enabled by default");
    }

    @Test
    public void testAsciiConfiguration() throws Exception {
        Configuration asciiCfg = new OffUnicodeConfig();

        void nullBuffer_throwsNPE() {
            // Arrange
            Smb2LogoffResponse resp = newResponse();

            // Act & Assert
            assertThrows(NullPointerException.class, () -> resp.readBytesWireFormat(null, 0));
        }

        @Test
        @DisplayName("Insufficient buffer causes ArrayIndexOutOfBoundsException")
        void shortBuffer_throwsAIOOBE() {

    @Mock
    private SmbTree tree;

    @BeforeEach
    void resetMocks() {
        Mockito.reset(session, transport, tree);
    }

    // Happy path: isInUse returns the mocked state
    @ParameterizedTest
    @DisplayName("isInUse reflects current mocked state")
    @ValueSource(booleans = { true, false })
    void isInUse_variants(boolean inUse) {
        when(session.isInUse()).thenReturn(inUse);

  @CollectionFeature.Require(SUPPORTS_ITERATOR_REMOVE)
  public void testRemoveIf_alwaysFalse() {
    assertFalse("removeIf(x -> false) should return false", collection.removeIf(x -> false));
    expectUnchanged();
  }

  @CollectionFeature.Require(SUPPORTS_ITERATOR_REMOVE)
  @CollectionSize.Require(absent = ZERO)
  public void testRemoveIf_sometimesTrue() {
    assertTrue(

        assertTrue("Last lines should be kept", output.contains("line15"));
    }

    public void test_run_maxBufferSizeConstant() {
        assertEquals(1000, InputStreamThread.MAX_BUFFER_SIZE);
    }

    public void test_run_withMaxBufferSize() throws InterruptedException {
        StringBuilder inputBuilder = new StringBuilder();
        for (int i = 1; i <= 1005; i++) {

    @Mock
    private SmbTransportInternal mockTransport;

    @BeforeEach
    void setUp() {
        dfsResolver = new DfsImpl(mockContext);
        when(mockContext.getConfig()).thenReturn(mockConfig);
    }

    // Test for isTrustedDomain method
    @Test
    void testIsTrustedDomain_DfsDisabled() throws CIFSException {
        when(mockConfig.isDfsDisabled()).thenReturn(true);