}

    /**
     * Gets the context handle returned by the server.
     *
     * @return the context handle
     */
    public byte[] getContextHandle() {
        return contextHandle != null ? contextHandle.clone() : null;
    }

    /**
     * Sets the context handle.
     *
     * @param contextHandle the context handle
     */
    public void setContextHandle(byte[] contextHandle) {

        @Test
        @DisplayName("Should handle self unwrap")
        void shouldHandleSelfUnwrap() {
            SmbSession mockSession = mock(SmbSession.class);
            when(mockSession.unwrap(SmbSession.class)).thenReturn(mockSession);

            SmbSession result = mockSession.unwrap(SmbSession.class);
            assertSame(mockSession, result, "unwrap should handle self unwrapping");
        }
    }

    protected DcerpcSecurityProvider securityProvider = null;
    private static int call_id = 1;

    /**
     * Gets a DCERPC handle for the specified URL and authentication
     * @param url the DCERPC URL to connect to
     * @param auth the NTLM authentication credentials
     * @return a DCERPC handle for the connection
     * @throws UnknownHostException if the host cannot be resolved

        this.flags = flags;
    }

    /**
     * Gets the context handle.
     *
     * @return the context handle
     */
    public byte[] getContextHandle() {
        return contextHandle;
    }

    /**
     * Sets the context handle.
     *
     * @param contextHandle the context handle
     */
    public void setContextHandle(byte[] contextHandle) {

        void testConstructorAndOpnum() {
            // When: Creating close handle message
            samr.SamrCloseHandle message = new samr.SamrCloseHandle(mockPolicyHandle);

            // Then: Should have correct opnum and handle
            assertEquals(0x01, message.getOpnum());
            assertEquals(mockPolicyHandle, message.handle);
        }

        @Test

     * @throws CIFSException if an error occurs connecting to the tree
     */
    SmbTreeHandleInternal ensureTreeConnected() throws CIFSException;

    /**
     * Ensures that the file handle is open and returns it.
     *
     * @return file handle
     * @throws CIFSException if an error occurs opening the file
     */
    SmbFileHandle ensureOpen() throws CIFSException;

    /**
     * Receive data from the pipe
     *

            assertEquals(expectedFlags, flags);
        }

        @Test
        @DisplayName("getDefaultFlags should handle Type1Message with request target")
        void testGetDefaultFlags_CIFSContext_Type1Message_Unicode_RequestTarget_DomainPresent() {
            // Given
            CIFSContext mockContext = createMockContext();

            assertThrows(NullPointerException.class, () -> {
                buffer.size();
            });
        }

        @ParameterizedTest
        @DisplayName("Should handle various referral levels")
        @ValueSource(ints = { 0, 1, 2, 3, 4, 5, 255, 256, 32767, 65535 })
        void testConstructorWithVariousReferralLevels(int maxReferralLevel) {
            String path = "\\\\test";

        // When
        int bytesWritten = response.writeBytesWireFormat(buffer, offset);

        // Then
        assertEquals(0, bytesWritten);
    }

    @Test
    @DisplayName("Should handle error response structure size 9")
    void testReadBytesWireFormatErrorResponse() throws Exception {
        // Given
        byte[] buffer = new byte[256];
        int bufferIndex = 0;

                log.warn("File handle was not properly closed, performing emergency cleanup: " + this);
                if (this.creationBacktrace != null) {
                    log.warn(Arrays.toString(this.creationBacktrace));
                }
                emergencyCloseHandle();
            }
        } catch (Exception e) {
            log.error("Error during file handle finalization", e);
        } finally {