rawKey[0] = (byte) ~rawKey[0];
        byte[] rawKey2 = keyManager.getRawKey(sessionId);
        assertArrayEquals(testKey, rawKey2, "Should still match original");
    }

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

        assertTrue(keyManager.hasSessionKey(sessionId), "Key should exist");