for (int i = 0; i < 3; i++) {
            assertEquals(1, keyManager.getKeyVersion("session-" + i), "Key version should be 1");
        }
    }

    @Test
    public void testCleanupArchivedKeys() throws GeneralSecurityException {
        String sessionId = "cleanup-test";
        keyManager.storeSessionKey(sessionId, testKey, "AES");

        // Create multiple versions