keyManager.storeSessionKey("session-" + i, key, "AES");
        }

        assertEquals(5, keyManager.getKeyCount(), "Should have 5 keys");

        keyManager.clearAllKeys();

        assertEquals(0, keyManager.getKeyCount(), "Should have no keys after clear");

        // Verify all keys are removed
        for (int i = 0; i < 5; i++) {

            }
        }
    }

    /**
     * Get the number of stored keys
     *
     * @return number of keys in memory
     */
    public int getKeyCount() {
        return sessionKeys.size();
    }

    /**
     * Check if a session key exists
     *
     * @param sessionId session identifier
     * @return true if key exists
     */