// Mock the request buffer
        DfsReferralRequestBuffer mockRequest = mock(DfsReferralRequestBuffer.class);
        when(mockRequest.encode(any(byte[].class), anyInt())).thenReturn(10);

        // Inject mock request
        Field requestField = Trans2GetDfsReferral.class.getDeclaredField("request");
        requestField.setAccessible(true);

    @Mock
    private Configuration mockConfig;

    private SmbComSeekResponse response;

    @BeforeEach
    void setUp() {
        // Setup mock configuration to avoid NPE
        when(mockConfig.getPid()).thenReturn(12345);
        response = new SmbComSeekResponse(mockConfig);
    }

    /* ------------------------------------------------------------------ */

    void getConfig_delegatesToContext() {
        // Arrange
        CIFSContext ctx = mock(CIFSContext.class);
        Configuration cfg = mock(Configuration.class);
        when(ctx.getConfig()).thenReturn(cfg);
        SmbTreeConnectionTrace trace = new SmbTreeConnectionTrace(ctx);

        // Act
        Configuration result = trace.getConfig();

        // Assert
        assertSame(cfg, result);

        };

        URL url = new URL("http://custom.protocol/path");

        // Configure the mock factory to return our handler for the 'http' protocol.
        when(mockFactory.createURLStreamHandler("http")).thenReturn(mockStreamHandler);

        Handler.setURLStreamHandlerFactory(mockFactory);

        URLConnection connection = handler.openConnection(url);

    private Smb2FlushRequest request;

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        // Create a test file ID (16 bytes)
        testFileId = new byte[16];
        for (int i = 0; i < 16; i++) {
            testFileId[i] = (byte) (i + 1);
        }

        assertEquals(0, read);
    }

    @Test
    void testToString() {
        // Test toString method
        when(mockConfig.getListSize()).thenReturn(50);

        trans2FindNext2 = new Trans2FindNext2(mockConfig, TEST_SID, TEST_RESUME_KEY, TEST_FILENAME, TEST_BATCH_COUNT, TEST_BATCH_SIZE);

        String result = trans2FindNext2.toString();

        // Setup session.transport field (accessed directly in SmbTree)
        session.transport = transport;
        // Setup session to return transport via method call
        when(session.transport()).thenReturn(transport);
    }

    @Test
    void testConstructor() {
        // Test constructor with a specific service
        SmbTree tree = new SmbTree(session, "testShare", "testService");

        void testMockImplementation() {
            NegotiateContextRequest mockRequest = mock(NegotiateContextRequest.class);
            when(mockRequest.getContextType()).thenReturn(0xFF);

            assertEquals(0xFF, mockRequest.getContextType());
            verify(mockRequest).getContextType();
        }
    }

    @Nested
    @DisplayName("Edge Cases and Error Conditions")

    private static final int ALL_SECURITY_INFORMATION = 0x0000000F;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockConfig.getTransactionBufferSize()).thenReturn(65535);
    }

    @Test
    @DisplayName("Test constructor initialization with basic parameters")
    void testConstructorBasic() {
        int fid = 0x1234;
        int securityInfo = OWNER_SECURITY_INFORMATION;

        // Use real configuration for most tests
        realConfig = new BaseConfiguration(false);
        // Mock configuration setup
        when(mockConfig.getTransactionBufferSize()).thenReturn(65535);
    }

    @Test
    @DisplayName("Test constructor with domain and server types")
    void testConstructor() {
        String domain = "WORKGROUP";
        int serverTypes = NetServerEnum2.SV_TYPE_ALL;