when(tree.acquire()).thenReturn(tree);

        // Setup for ensureOpen
        when(tree.isSMB2()).thenReturn(true);
        when(pipe.openUnshared(anyString(), anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fh);
        when(fh.acquire()).thenReturn(fh);
        lenient().when(fh.isValid()).thenReturn(true);
        lenient().when(fh.getTree()).thenReturn(tree);

        when(tree.getSendBufferSize()).thenReturn(1024);
        when(tree.hasCapability(anyInt())).thenAnswer(inv -> ntSmbsCap && inv.getArgument(0).equals(SmbConstants.CAP_NT_SMBS));
        when(tree.areSignaturesActive()).thenReturn(false);
        when(tree.isSMB2()).thenReturn(smb2);

        when(file.openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fh);
        when(fh.acquire()).thenReturn(fh);

            doReturn(mockTreeHandle).when(smbFile).ensureTreeConnected();
            when(mockTreeHandle.getConfig()).thenReturn(mockConfig);
            doReturn(mockFileHandle).when(smbFile).openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt());
        }

        @Test
        void testGetInputStream() throws IOException {
            // Act
            var inputStream = smbFile.getInputStream();

            // Assert

            CreateContextResponse response = mock(CreateContextResponse.class);
            byte[] buffer = new byte[100];
            when(response.decode(any(byte[].class), anyInt(), anyInt())).thenReturn(10);

            int result = response.decode(buffer, 0, 10);
            assertEquals(10, result);
            verify(response, times(1)).decode(buffer, 0, 10);
        }

        @Test

        void shouldHandleMockedDecodeMethod() throws jcifs.internal.SMBProtocolDecodingException {
            // Given
            byte[] buffer = new byte[100];
            when(mockAllocInfo.decode(any(byte[].class), anyInt(), anyInt())).thenReturn(42);

            // When
            int result = mockAllocInfo.decode(buffer, 10, 50);

            // Then
            assertEquals(42, result);

import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.io.IOException;

        when(mockInputData.encode(any(byte[].class), anyInt())).thenReturn(100);
        request.setInputData(mockInputData);
        byte[] buffer = new byte[1024];

        int written = request.writeBytesWireFormat(buffer, 0);

        assertTrue(written > 57);
        verify(mockInputData).encode(any(byte[].class), anyInt());
    }

    @Test

            lenient().doNothing().when(mockBuffer).advance(anyInt());
            lenient().doNothing().when(mockBuffer).enc_ndr_long(anyInt());
            lenient().doNothing().when(mockBuffer).enc_ndr_short(anyInt());
            lenient().doNothing().when(mockBuffer).enc_ndr_small(anyInt());
            lenient().doNothing().when(mockBuffer).setIndex(anyInt());

            message.encode(mockBuffer);

import static org.junit.jupiter.api.Assertions.assertNull;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.io.IOException;

import org.junit.jupiter.api.BeforeEach;

            response.setAsync(true);

            boolean result = response.verifySignature(buffer, 0, 100);

            assertTrue(result);
            verify(mockDigest, never()).verify(any(), anyInt(), anyInt(), anyInt(), any());
        }

        @Test
        @DisplayName("Should skip verification when digest is null")
        void testVerifySignatureNoDigest() {
            byte[] buffer = new byte[100];