// Return mocked transports
        doReturn(trans1).when(poolSpy).getSmbTransport(eq(ctx), eq(addr1), anyInt(), anyBoolean(), anyBoolean());
        doReturn(trans2).when(poolSpy).getSmbTransport(eq(ctx), eq(addr2), anyInt(), anyBoolean(), anyBoolean());

        // When: Get transport by name
        SmbTransportImpl result = poolSpy.getSmbTransport(ctx, "test.server", 445, false, false);

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.io.IOException;

        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockContext.getTransportPool()).thenReturn(mockTransportPool);
        when(mockTransportPool.getSmbTransport(any(), anyString(), anyInt(), anyBoolean(), anyBoolean())).thenReturn(mockTransport);
        when(mockTransport.unwrap(any())).thenReturn(mockTransport);
        when(mockTransport.getDfsReferrals(any(), anyString(), anyString(), anyString(), anyInt())).thenReturn(null);

        // Mock transport to throw IOException - simulating connection failure
        when(mockTransportPool.getSmbTransport(any(CIFSContext.class), anyString(), anyInt(), anyBoolean(), anyBoolean()))
                .thenThrow(new IOException("Connection failed"));

        // Instantiate the class under test
        dfsImpl = new DfsImpl(mockContext);
    }

    // Test for the constructor

            "true,true,false", "true,false,true", "false,true,true", "false,false,false" })
    void disconnect_flagCombinations(boolean hard, boolean inuse, boolean expected) throws Exception {
        when(transport.disconnect(anyBoolean(), anyBoolean())).thenAnswer(inv -> {
            boolean h = inv.getArgument(0);
            boolean u = inv.getArgument(1);
            // Arbitrary behavior mapping for test purposes
            return (h && !u) || (!h && u);

    void testEqualsAndHashCode() throws Exception {
        UniAddress a = mock(UniAddress.class);
        when(nsc.getAllByName(eq("server"), anyBoolean())).thenReturn(new Address[] { a });
        when(nsc.getAllByName(eq("SERVER"), anyBoolean())).thenReturn(new Address[] { a });
        SmbResourceLocatorImpl l1 = locator("smb://server/share/file");
        SmbResourceLocatorImpl l2 = locator("smb://SERVER/share/file");

            when(mockResource.openOutputStream()).thenReturn(mockOutputStream);
            when(mockResource.openOutputStream(anyBoolean())).thenReturn(mockOutputStream);
            when(mockResource.openOutputStream(anyBoolean(), anyInt())).thenReturn(mockOutputStream);
            when(mockResource.openOutputStream(anyBoolean(), anyInt(), anyInt(), anyInt())).thenReturn(mockOutputStream);

            // When

import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.lenient;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;

import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;