testBuffer[offset + 9] = 0x01; // Sub-authority count
        for (int i = 0; i < 6; i++) {
            testBuffer[offset + 10 + i] = 0x00;
        }
        testBuffer[offset + 15] = 0x01;
        for (int i = 0; i < 4; i++) {
            testBuffer[offset + 16 + i] = 0x00;
        }

        // Test decode with offset
        int size = ace.decode(testBuffer, offset, testBuffer.length - offset);

        private byte[] data;
        private int decodeCallCount = 0;
        private boolean throwOnDecode = false;

        public TestCreateContextResponse(byte[] name) {
            this.name = name;
        }

        @Override
        public byte[] getName() {
            return name;
        }

        @Override
        public int decode(byte[] buffer, int bufferIndex, int len) throws SMBProtocolDecodingException {

            for (int i = 0; i < 100; i++) {
                int bytesRead = response.readBytesWireFormat(buffer, 0);
                // Then
                assertEquals(4, bytesRead, "Failed at iteration " + i);
            }

            // When - perform multiple write operations
            for (int i = 0; i < 100; i++) {
                int bytesWritten = response.writeBytesWireFormat(buffer, 0);

        byte[] buffer = new byte[256];
        int result = request.readBytesWireFormat(buffer, 0);

        assertEquals(0, result);
    }

    @ParameterizedTest
    @ValueSource(ints = { 0, 10, 50, 100 })
    @DisplayName("writeBytesWireFormat should work at different offsets")
    void testWriteBytesWireFormatAtDifferentOffsets(int offset) {
        byte[] buffer = new byte[512];

        assertTrue(namesObj instanceof lsarpc.LsarTransNameArray);
    }

    @ParameterizedTest
    @ValueSource(ints = { 1, 5, 10, 20, 50 })
    void constructor_shouldHandleVariousSidArraySizes(int size) {
        // Arrange
        testSids = new jcifs.SID[size];
        for (int i = 0; i < size; i++) {
            testSids[i] = mock(jcifs.SID.class);

    void testWriteSetupWireFormatWithVariousFids(int fid, byte expectedLow, byte expectedHigh) {
        // Arrange
        transPeekNamedPipe = new TransPeekNamedPipe(mockConfig, TEST_PIPE_NAME, fid);
        byte[] buffer = new byte[10];
        int offset = 0;

        // Act
        int bytesWritten = transPeekNamedPipe.writeSetupWireFormat(buffer, offset);

        // Assert

        @ParameterizedTest
        @ValueSource(ints = { NegTokenInit.DELEGATION, NegTokenInit.MUTUAL_AUTHENTICATION, NegTokenInit.REPLAY_DETECTION,
                NegTokenInit.SEQUENCE_CHECKING, NegTokenInit.ANONYMITY, NegTokenInit.CONFIDENTIALITY, NegTokenInit.INTEGRITY })
        @DisplayName("Individual flag values are correctly set and retrieved")
        void testIndividualFlagValues(int flag) {
            NegTokenInit init = new NegTokenInit();

    @ValueSource(ints = { 1024, 2048, 4096, 8192, 16384, 32768, 65535 })
    void testMaxDataCountValues(int maxDataCount) throws Exception {
        netShareEnum = new NetShareEnum(realConfig);

        // Set maxDataCount using reflection
        setFieldValue(netShareEnum, "maxDataCount", maxDataCount);

        byte[] dst = new byte[256];

    }

    @ParameterizedTest
    @ValueSource(ints = { NtStatus.NT_STATUS_SUCCESS, NtStatus.NT_STATUS_NONE_MAPPED, 0x00000107 })
    @DisplayName("resolveSids(DcerpcHandle,...) accepts success/none/some-not-mapped codes")
    void resolveSids_allowsCertainRetvals_noThrow(int ret) throws Exception {
        SIDCacheImpl cache = new SIDCacheImpl(mock(CIFSContext.class));

        void testEncodeVariousHashAlgoCounts(int algoCount) {
            int[] hashAlgos = new int[algoCount];
            for (int i = 0; i < algoCount; i++) {
                hashAlgos[i] = i + 1;
            }
            byte[] salt = { 0x01, 0x02 };
            PreauthIntegrityNegotiateContext context = new PreauthIntegrityNegotiateContext(mockConfig, hashAlgos, salt);

            int bytesWritten = context.encode(buffer, 0);