private Configuration mockConfig;

    private Smb2QueryDirectoryResponse response;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockConfig.isUseUnicode()).thenReturn(true);
        when(mockConfig.getMaximumBufferSize()).thenReturn(65536);
    }

    @Test
    @DisplayName("Test constructor initializes with config and expectInfoClass")
    void testConstructor() {

        when(handle.isValid()).thenReturn(true);
        when(handle.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(false);
        when(tree.getConfig()).thenReturn(mock(Configuration.class));
        when(tree.hasCapability(SmbConstants.CAP_NT_SMBS)).thenReturn(true);
        when(handle.getFid()).thenReturn(fid);

    # /- - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    # o classificationUndefinedHandlingType: (NotRequired - Default LOGGING)
    #  The handling type when undefined classification is found.
    #
    #   EXCEPTION - throws exception when found
    #   LOGGING   - logging only when found (exception if ReplaceSchema)
    #   ALLOWED   - no action
    #
    #; classificationUndefinedHandlingType = LOGGING
    # - - - - - - - - - -/

    @DisplayName("Should set file ID correctly in Smb2CloseRequest")
    void testSetFileIdInCloseRequest() {
        // Given
        Smb2CloseRequest request = new Smb2CloseRequest(mockConfig, emptyFileId);

        // When
        request.setFileId(testFileId);

        // Then - verify through internal state (would need getter or reflection in real scenario)

import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;

    /** The key of the configuration. e.g. Referrer-Policy:strict-origin-when-cross-origin */
    String API_JSON_RESPONSE_HEADERS = "api.json.response.headers";

    /** The key of the configuration. e.g. false */
    String API_JSON_RESPONSE_EXCEPTION_INCLUDED = "api.json.response.exception.included";

    /** The key of the configuration. e.g. Referrer-Policy:strict-origin-when-cross-origin */

            int result = 0;
            when(mockBuffer.dec_ndr_short()).thenReturn(4096, 4096, 10, result, 0);
            when(mockBuffer.dec_ndr_long()).thenReturn(12345);
            when(mockBuffer.dec_ndr_small()).thenReturn(1);
            lenient().doNothing().when(mockBuffer).advance(anyInt());
            lenient().when(mockBuffer.align(anyInt())).thenReturn(0);

            // When
            bind.decode_out(mockBuffer);

    /**
     * Ensure encode throws NPE when passed a null buffer.
     */
    @Test
    @DisplayName("Encode with null buffer throws NullPointerException")
    public void testEncodeNullBuffer() throws NdrException {
        NdrHyper hyper = new NdrHyper(5L);
        assertThrows(NullPointerException.class, () -> hyper.encode(null));
    }

    /**
     * Ensure decode throws NPE when passed a null buffer.
     */
    @Test

     * Verifies that unwrap can return itself when the type matches.
     */
    @Test
    void testUnwrap_withSameType() {
        when(smbTree.unwrap(SmbTree.class)).thenReturn(smbTree);

        SmbTree result = smbTree.unwrap(SmbTree.class);

        assertNotNull(result, "Unwrapped tree should not be null");
        assertSame(smbTree, result, "Should return itself when unwrapping to SmbTree");
    }

    /**

import static org.junit.jupiter.api.Assertions.assertSame;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.security.Principal;

import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.junit.jupiter.params.ParameterizedTest;