byte[] dst = new byte[42];

        // When
        int result = trans2SetFileInformation.writeDataWireFormat(dst, 0);

        // Then
        assertEquals(40, result); // Returns 40 bytes written
    }

    @Test
    void testReadSetupWireFormat() {
        // When
        int result = trans2SetFileInformation.readSetupWireFormat(new byte[0], 0, 0);

        // Then
        assertEquals(0, result);
    }

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

        // Act
        handle.close();

        // Assert
        // Verify sendrecv was called for open and then for close
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcSamrOpenDomain.class));

        toolchainModels.add(ToolchainModel.newBuilder().type("unknown").build());
        when(session.getRequest()).thenReturn(executionRequest);
        Session sessionv4 = mock(Session.class);
        when(session.getSession()).thenReturn(sessionv4);
        when(sessionv4.getToolchains()).thenReturn(toolchainModels);

        @DisplayName("Should handle multiple invocations")
        void shouldHandleMultipleInvocations() {
            // Given
            when(mockAllocInfo.getCapacity()).thenReturn(1000L, 2000L, 3000L);
            when(mockAllocInfo.getFree()).thenReturn(500L, 1000L, 1500L);

            // When & Then
            assertEquals(1000L, mockAllocInfo.getCapacity());
            assertEquals(500L, mockAllocInfo.getFree());

     */
    @Test
    void testGetKey_KeyExists() throws LoginException {
        when(key1.getKeyType()).thenReturn(KEY_TYPE_1);
        when(key2.getKeyType()).thenReturn(KEY_TYPE_2);
        Set<Object> privateCredentials = new HashSet<>();
        privateCredentials.add(key1);
        privateCredentials.add(key2);
        when(subject.getPrivateCredentials()).thenReturn(privateCredentials);

        when(smbFileHandle.isValid()).thenReturn(true);
        assertTrue(smbFileHandle.isValid(), "isValid should return true when the handle is valid.");
        verify(smbFileHandle, times(1)).isValid();
    }

    /**
     * Test isValid() when the handle is invalid.
     */
    @Test
    void testIsValid_whenInvalid() {
        when(smbFileHandle.isValid()).thenReturn(false);

        CIFSContext mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
        Smb2TreeDisconnectRequest request = new Smb2TreeDisconnectRequest(mockConfig);

        // When
        Smb2TreeDisconnectResponse response = request.createResponse(mockContext, request);

        // Then
        assertNotNull(response);

     */
    @Test
    void testMethodInteraction() {
        CustomSmbTree customTree = mock(CustomSmbTree.class);
        when(smbTree.unwrap(CustomSmbTree.class)).thenReturn(customTree);
        doNothing().when(smbTree).close();
        doNothing().when(customTree).close();

        // Unwrap then close both
        CustomSmbTree unwrapped = smbTree.unwrap(CustomSmbTree.class);
        smbTree.close();
        unwrapped.close();

            buffer.putInt(512); // Bytes per sector
            byte[] bufferArray = buffer.array();

            // When
            int bytesConsumed = fileFsSizeInfo.decode(bufferArray, 0, bufferArray.length);

            // Then
            assertEquals(24, bytesConsumed);
            assertEquals(1048576L * 8 * 512, fileFsSizeInfo.getCapacity());

    @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");