// Setup mock configuration with lenient stubbing
        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(handle.getTransportContext()).thenCallRealMethod();
        lenient().when(handle.getServer()).thenCallRealMethod();
        lenient().when(handle.getServerWithDfs()).thenCallRealMethod();
        lenient().when(handle.getSessionKey()).thenCallRealMethod();
        lenient().when(handle.getBinding()).thenReturn(mockDcerpcBinding);

        lenient().when(request.getSession()).thenReturn(session);
        lenient().when(request.getSession(false)).thenReturn(session);
        lenient().when(request.getSession(true)).thenReturn(session);

        // Setup default session attribute behavior
        lenient().when(session.getAttribute(anyString())).thenReturn(null);

        // Setup SmbFile mock

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

        // Common happy-path defaults
        // Use lenient() to avoid strict stubbing issues with Mockito
        lenient().when(tree.acquire()).thenReturn(tree);
        lenient().when(tree.getConfig()).thenReturn(config);
        lenient().when(config.getListCount()).thenReturn(10);
        lenient().when(config.getListSize()).thenReturn(4096);

        lenient().when(parent.getLocator()).thenReturn(locator);

        when(cifsContext.getNameServiceClient()).thenReturn(nameServiceClient);
        lenient().when(cifsContext.getConfig()).thenReturn(configuration);
        lenient().when(configuration.getDefaultDomain()).thenReturn("TEST_DOMAIN");
        lenient().when(nameServiceClient.getByName(anyString(), anyBoolean())).thenReturn(mock(Address.class));
        lenient().when(transportPool.getChallenge(any(), any())).thenReturn(new byte[8]);
    }

    @Mock
    SmbTreeHandleImpl tree;

    private void stubValidTree(long treeId, boolean connected, boolean smb2) {
        lenient().when(tree.acquire()).thenReturn(tree);
        lenient().when(tree.getTreeId()).thenReturn(treeId);
        lenient().when(tree.isConnected()).thenReturn(connected);
        lenient().when(tree.isSMB2()).thenReturn(smb2);
    }

    @Test

    private SmbFileOutputStream outputStream;

    @BeforeEach
    void setUp() {
        // Common setup for most tests
        lenient().when(mockFileHandle.getTree()).thenReturn(mockTreeHandle);
        lenient().when(mockTreeHandle.getConfig()).thenReturn(mockConfig);
    }

    @Test
    void testWriteSingleByte() throws IOException, CIFSException {
        // Given

        // Setup mock configuration with lenient stubbing
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.isUseWitness()).thenReturn(true);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);

        FileEntry fe1 = mock(FileEntry.class);
        lenient().when(fe1.getName()).thenReturn("file1");
        lenient().when(fe1.getFileIndex()).thenReturn(1);

        FileEntry fe2 = mock(FileEntry.class);
        lenient().when(fe2.getName()).thenReturn("file2");
        when(fe2.getFileIndex()).thenReturn(2);

        FileEntry fe3 = mock(FileEntry.class);
        lenient().when(fe3.getName()).thenReturn("file3");