assertNotNull(retrieved, "Should retrieve stored key");
        assertArrayEquals(testKey, retrieved.getEncoded(), "Retrieved key should match");
    }

    @Test
    public void testGetRawKey() {
        String sessionId = "test-session-2";
        keyManager.storeSessionKey(sessionId, testKey, "AES");

        byte[] rawKey = keyManager.getRawKey(sessionId);