}

    /**
     * Get the current key rotation count
     *
     * @return number of times keys have been rotated
     */
    public int getKeyRotationCount() {
        return rotationCount;
    }

    /**
     * Set the key rotation bytes limit
     *
     * @param limit number of bytes to encrypt before rotating keys
     */

        byte[] encrypted2 = assertDoesNotThrow(() -> context.encryptMessage(largeMessage, 2L));
        assertNotNull(encrypted2);

        // Verify rotation occurred by checking metrics
        assertTrue(context.getKeyRotationCount() > 0, "Key rotation should have occurred");

        // Can still encrypt after rotation
        byte[] plaintext3 = "Message after rotation".getBytes();