}

    @Test
    @DisplayName("Should create lease key from byte array")
    void testLeaseKeyFromBytes() {
        byte[] testBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

        Smb2LeaseKey key = new Smb2LeaseKey(testBytes);

        assertArrayEquals(testBytes, key.getKey());
        assertFalse(key.isZero());
    }

    @Test

        assertEquals(0x08, AvPair.MsvAvSingleHost, "MsvAvSingleHost constant should be 0x08.");
        assertEquals(0x09, AvPair.MsvAvTargetName, "MsvAvTargetName constant should be 0x09.");
        assertEquals(0x0A, AvPair.MsvAvChannelBindings, "MsvAvChannelBindings constant should be 0x0A.");
    }

    @Test
    void testRawDataImmutability() {

    /**
     * Single host type
     */
    public static final int MsvAvSingleHost = 0x08;

    /**
     * Target name type
     */
    public static final int MsvAvTargetName = 0x09;

    /**
     * Channel bindings type
     */
    public static final int MsvAvChannelBindings = 0x0A;

    private final int type;
    private final byte[] raw;

    /**

     */
    @Test
    void testAvSingleHostCustomDataMachineIdConstructor_LongerInputs() {
        byte[] customData = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A }; // Longer than 8
        byte[] machineId = new byte[40]; // Longer than 32
        for (int i = 0; i < 40; i++) {
            machineId[i] = (byte) i;
        }

    private Smb2LeaseKey testKey;
    private int testState;

    @BeforeEach
    void setUp() {
        byte[] keyBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
        testKey = new Smb2LeaseKey(keyBytes);
        testState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;
        leaseContext = new LeaseV1CreateContextRequest(testKey, testState);
    }

        case 0x02:
            return "This folder and subfolders";
        case 0x0A:
            return "Subfolders only";
        case 0x01:
            return "This folder and files";
        case 0x09:
            return "Files only";
        }
        return "Invalid";
    }

    @Override
    public int getAccessMask() {
        return this.access;
    }

    @Override

            when(mockNdrBuffer.dec_ndr_small()).thenReturn((int) (byte) 0x55, (int) (byte) 0x66, // clock seq bytes
                    (int) (byte) 0x07, (int) (byte) 0x08, (int) (byte) 0x09, (int) (byte) 0x0A, (int) (byte) 0x0B, (int) (byte) 0x0C // node bytes
            );
            when(mockNdrBuffer.derive(anyInt())).thenReturn(mockNdrBuffer);

            // When: Decoding the policy handle

    @CsvSource({ "0x00, This folder only", "'0x03', 'This folder, subfolders and files'", "0x0B, Subfolders and files only",
            "0x02, This folder and subfolders", "0x0A, Subfolders only", "0x01, This folder and files", "0x09, Files only",
            "0x08, Invalid" })
    void testGetApplyToText(String flagsHex, String expectedText) {
        ace.flags = Integer.parseInt(flagsHex.substring(2), 16);

            return Stream.of(Arguments.of((byte) 0x00, "SMB2_OPLOCK_LEVEL_NONE"), Arguments.of((byte) 0x01, "SMB2_OPLOCK_LEVEL_II"),
                    Arguments.of((byte) 0x08, "SMB2_OPLOCK_LEVEL_EXCLUSIVE"), Arguments.of((byte) 0x09, "SMB2_OPLOCK_LEVEL_BATCH"),
                    Arguments.of((byte) 0xFF, "SMB2_OPLOCK_LEVEL_LEASE"));
        }
    }

        case 0x02:
            return "This folder and subfolders";
        case 0x0A:
            return "Subfolders only";
        case 0x01:
            return "This folder and files";
        case 0x09:
            return "Files only";
        }
        return "Invalid";
    }

    /**
     * Returns the access mask accociated with this ACE. Use the