byte[] rawKey = keyManager.getRawKey(sessionId);
        assertNotNull(rawKey, "Should retrieve raw key");
        assertArrayEquals(testKey, rawKey, "Raw key should match");

        // Verify we get a copy, not the original
        rawKey[0] = (byte) ~rawKey[0];
        byte[] rawKey2 = keyManager.getRawKey(sessionId);
        assertArrayEquals(testKey, rawKey2, "Should still match original");
    }

    @Test

        assertEquals(12345L, size);

        // Verify precise interactions
        verify(mockEntry, times(1)).getName();
        verify(mockEntry, times(2)).getType();
        verify(mockEntry, times(1)).getAttributes();
        verify(mockEntry, times(1)).createTime();
        verify(mockEntry, times(1)).lastModified();
        verify(mockEntry, times(1)).lastAccess();
        verify(mockEntry, times(1)).length();

        CIFSContext wrappedContext = context.withCredentials(newCreds);

        assertNotNull(wrappedContext);
        assertTrue(wrappedContext instanceof CIFSContextCredentialWrapper);
        // Verify that the new context uses the provided credentials
        assertEquals(newCreds, wrappedContext.getCredentials());
    }

    @Test
    void testWithAnonymousCredentials() {

        byte[] dest = new byte[5];

        int encodedLen = encodable.encode(dest, 0);

        // Verify encoded length
        assertEquals(3, encodedLen, "Encoded length should be equal to the specified length");

        // Verify content
        assertArrayEquals(new byte[] { 0x02, 0x03, 0x04, 0x00, 0x00 }, dest, "Encoded bytes should match the expected subset");
    }

    @Test

        // Create buffer for encoding
        byte[] buffer = new byte[16];
        int bytesWritten = info.encode(buffer, 0);

        // Verify bytes written
        assertEquals(8, bytesWritten);

        // Verify encoded value
        long decodedValue = SMBUtil.readInt8(buffer, 0);
        assertEquals(endOfFile, decodedValue);
    }

    @Test

        NdrSmall ndrSmall = new NdrSmall(0);

        // Call the decode method
        ndrSmall.decode(mockNdrBuffer);

        // Verify that dec_ndr_small was called
        verify(mockNdrBuffer, times(1)).dec_ndr_small();
        // Verify that the value field was updated correctly
        assertEquals(decodedValue, ndrSmall.value, "Value should be updated after decoding.");

        ndrLong.decode(mockNdrBuffer);

        // Verify that dec_ndr_long was called exactly once
        verify(mockNdrBuffer, times(1)).dec_ndr_long();
        // Verify that the value field was updated correctly
        assertEquals(decodedValue, ndrLong.value, "Decode should correctly assign the value returned by dec_ndr_long.");
        // Verify that no other interactions occurred with the mock buffer

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

        verify(smbTree).unwrap(CustomSmbTree.class);
        verify(smbTree).close();
        verify(customTree).close();
    }

    /**
     * Custom SmbTree interface for testing unwrap functionality.
     */
    interface CustomSmbTree extends SmbTree {

        // Verify bytes written
        assertEquals(24, written);

        // Verify structure size (should be 24)
        assertEquals(24, SMBUtil.readInt2(buffer, offset));

        // Verify Reserved1 (2 bytes at offset+2, should be 0)
        assertEquals(0, SMBUtil.readInt2(buffer, offset + 2));

        // Verify Reserved2 (4 bytes at offset+4, should be 0)

        // Decode
        int bytesDecoded = smbInfoAllocation.decode(buffer, 0, buffer.length);

        // Verify - actual implementation reads 20 bytes (4 + 4 + 4 + 4 + 4)
        // bytesPerSect is read as Int2 but advances by 4 bytes (padding)
        assertEquals(20, bytesDecoded);

        // Verify capacity calculation: alloc * sectPerAlloc * bytesPerSect
        long expectedCapacity = alloc * sectPerAlloc * bytesPerSect;