}

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

        SecretKey retrieved = keyManager.getSessionKey(sessionId);
        assertNotNull(retrieved, "Should retrieve stored key");
        assertArrayEquals(testKey, retrieved.getEncoded(), "Retrieved key should match");
    }

    @Test
    public void testGetRawKey() {

     * Gets the session key from the underlying SMB session.
     *
     * @return session key of the underlying smb session
     * @throws CIFSException if an error occurs retrieving the session key
     */
    byte[] getSessionKey() throws CIFSException;

    /**
     * Gets the input stream for reading from this pipe.
     *
     * @return this pipe's input stream
     * @throws CIFSException if an error occurs getting the input stream

        // After close, report stale
        handle.close();
        assertTrue(handle.isStale(), "Closed handle reports stale");
    }

    @Test
    @DisplayName("getSessionKey propagates CIFSException from session acquisition")
    void getSessionKey_exception_propagates() throws Exception {
        SmbPipeHandleImpl handle = newHandleWithBasicStubs(0, "\\\\pipe\\\\sess");

        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        try {
            // Combine session key with channel-specific data
            byte[] sessionKey = getSessionKey();
            if (sessionKey != null) {
                baos.write(sessionKey);
            }
            baos.write(channel.getLocalInterface().getAddress().getAddress());

        multiConfig = new PropertyConfiguration(props);

        // Mock context and session for ChannelManager
        when(mockContext.getConfig()).thenReturn(multiConfig);
        when(mockSession.getSessionKey()).thenReturn(new byte[16]);

        channelManager = new ChannelManager(mockContext, mockSession);
    }

    @Test
    void testMultiChannelInitialization() throws Exception {

    }

    /**
     * Retrieve a session key
     *
     * @param sessionId unique session identifier
     * @return the secret key, or null if not found
     */
    public SecretKey getSessionKey(String sessionId) {
        checkNotClosed();

        SecretKey key = sessionKeys.get(sessionId);

        // Try to load from KeyStore if not in memory
        if (key == null && keyStore != null) {

            MIEName server = new MIEName(mech, ticket.getServer().getName());
            if (src.equals(client) && targ.equals(server)) {
                return ticket.getSessionKey();
            }
        }
        return null;
    }

    /**
     * {@inheritDoc}
     *
     * @see java.lang.Object#toString()
     */
    @Override

     * @return session key of the underlying smb session
     * @throws CIFSException if unable to retrieve the session key
     */
    public abstract byte[] getSessionKey() throws CIFSException;

    @Override
    public String toString() {
        return this.binding.toString();
    }

    /**
     * Sends a DCERPC fragment to the remote endpoint

        // Combine session key with channel-specific data
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        
        try {
            baos.write(session.getSessionKey());
            baos.write(channel.getLocalInterface().getAddress().getAddress());
            baos.write(channel.getRemoteInterface().getAddress().getAddress());

            log.error("Failed to perform emergency cleanup", e);
        }
    }

    /**
     * @return the sessionKey
     * @throws CIFSException
     */
    @Override
    public byte[] getSessionKey() throws CIFSException {
        if (this.sessionKey == null) {
            throw new CIFSException("No session key available");
        }
        return this.sessionKey;
    }

    @Override