when(mockContext.getConfig()).thenReturn(mockConfig);
        request = new Smb2SessionSetupRequest(mockContext, TEST_SECURITY_MODE, TEST_CAPABILITIES, TEST_PREVIOUS_SESSION_ID, TEST_TOKEN);
    }

    @Test
    @DisplayName("Should create request with correct parameters")
    void testConstructorSetsCorrectParameters() throws Exception {
        // Given & When
        int securityMode = 0x03;

    void setUp() {
        lenient().when(mockFileEntry1.getName()).thenReturn("file1.txt");
        lenient().when(mockFileEntry2.getName()).thenReturn("file2.txt");
        lenient().when(mockFileEntry3.getName()).thenReturn("file3.txt");

        lenient().when(mockResource1.getName()).thenReturn("file1.txt");
        lenient().when(mockResource2.getName()).thenReturn("file2.txt");
        lenient().when(mockResource3.getName()).thenReturn("file3.txt");

    @BeforeEach
    void setUp() throws Exception {
        when(ctx.getConfig()).thenReturn(cfg);
        when(cfg.isSigningEnforced()).thenReturn(false);
        when(cfg.getSessionTimeout()).thenReturn(30_000);
        when(cfg.getResponseTimeout()).thenReturn(5_000);
        when(address.getHostAddress()).thenReturn("127.0.0.1");
        when(address.getHostName()).thenReturn("localhost");

        testFileId = new byte[16];
        Arrays.fill(testFileId, (byte) 0xAB);

        // Use lenient stubbing for default configuration values
        lenient().when(mockConfig.getNotifyBufferSize()).thenReturn(8192);
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);

        request = new Smb2ChangeNotifyRequest(mockConfig, testFileId);
    }

    @Test

    /**
     * Test for close() method throwing CIFSException.
     * Verifies that the method correctly throws a CIFSException when configured to do so.
     * @throws CIFSException
     */
    @Test
    void testClose_throwsCIFSException() throws CIFSException {
        doThrow(new CIFSException("Test Exception")).when(smbTreeHandle).close();
        assertThrows(CIFSException.class, () -> smbTreeHandle.close(), "close() should throw CIFSException");

        void testConstructorZero() {
            // Given/When: Creating NdrShort with zero
            NdrShort ndrShort = new NdrShort(0);

            // Then: Value should be zero
            assertEquals(0, ndrShort.value);
        }

        @Test
        @DisplayName("Should handle maximum byte value correctly")
        void testConstructorMaxByte() {
            // Given/When: Creating NdrShort with max byte value

    void hasCapability_variousCaps(int cap) throws SmbException {
        when(transport.hasCapability(anyInt())).thenAnswer(inv -> ((int) inv.getArgument(0)) >= 0);

        boolean result = transport.hasCapability(cap);

        assertEquals(cap >= 0, result);
        verify(transport, times(1)).hasCapability(cap);
    }

    // Invalid input: simulate underlying SmbException when capability check fails
    @Test

        when(sh.isSMB2()).thenReturn(true);
        when(dh.isSMB2()).thenReturn(true);
        when(sh.isSameTree(dh)).thenReturn(true);

        // Source open returns a handle that reports size 0
        SmbFileHandleImpl sfd = mock(SmbFileHandleImpl.class);
        when(sfd.getInitialSize()).thenReturn(0L);

            rpc.retval = 0;
            return null;
        }).when(mockDcerpcHandle).sendrecv(any(MsrpcLsarOpenPolicy2.class));

        LsaPolicyHandle handle = new LsaPolicyHandle(mockDcerpcHandle, "server", 123);

        // Now, mock the behavior for MsrpcLsarClose to throw IOException
        doThrow(new IOException("Close network error")).when(mockDcerpcHandle).sendrecv(any(MsrpcLsarClose.class));

        // Act & Assert

        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);
        lenient().when(mockConfig.isWitnessServiceDiscovery()).thenReturn(true);
    }