lenient().when(registerResponse.isSuccess()).thenReturn(true);
        lenient().when(registerResponse.getRegistrationId()).thenReturn("test-reg-123");

        // Setup successful unregistration
        WitnessUnregisterResponse unregisterResponse = mock(WitnessUnregisterResponse.class);
        lenient().when(unregisterResponse.isSuccess()).thenReturn(true);

            // Create response from RPC message results
            WitnessRegisterResponse response = new WitnessRegisterResponse();
            response.setReturnCode(message.getReturnCode());

            if (message.isSuccess()) {
                // Generate registration ID from context handle
                byte[] contextHandle = message.getContextHandle();

                request.setFlags(registration.getFlags());

                WitnessRegisterResponse response = rpcClient.register(request);

                if (response != null && response.isSuccess()) {
                    registration.setState(WitnessRegistrationState.REGISTERED);
                    registration.setContextHandle(response.getContextHandle());

                request.setFlags(registration.getFlags());
                
                WitnessRegisterResponse response = rpcClient.register(request);
                
                if (response.isSuccess()) {
                    registration.setState(WitnessRegistrationState.REGISTERED);
                    registrations.put(registration.getRegistrationId(), registration);

    }

    private int returnCode;
    private String error;

    /**
     * Checks if the unregistration was successful.
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return returnCode == 0;
    }

    /**
     * Gets a human-readable error description.
     *
     * @return the error description
     */
    public String getError() {

    /**
     * Check if negotiation was successful
     *
     * @return true if successful, false otherwise
     */
    public boolean isSuccess() {
        return status == 0;
    }

    /**
     * Get negotiation status
     *
     * @return status code (0 = success)
     */
    public int getStatus() {
        return status;
    }

    /**
     * Set negotiation status
     *

    private long recommendedHeartbeatInterval;
    private String error;

    /**
     * Checks if the heartbeat was successful.
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return returnCode == 0;
    }

    /**
     * Gets the sequence number.
     *
     * @return the sequence number
     */
    public long getSequenceNumber() {
        return sequenceNumber;

    private byte[] contextHandle;
    private String error;

    /**
     * Checks if the registration was successful.
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return returnCode == 0;
    }

    /**
     * Gets a human-readable error description.
     *
     * @return the error description
     */
    public String getError() {

     */
    public boolean isSuccess() {
        return returnCode == ERROR_SUCCESS;
    }

    /**
     * Gets the error message for the current return code.
     *
     * @return a human-readable error message
     */
    public String getErrorMessage() {
        switch (returnCode) {
        case ERROR_SUCCESS:
            return "Success";
        case ERROR_INVALID_PARAMETER:

     * The response sequence number should match the request sequence number.
     *
     * @return true if the response is valid
     */
    public boolean isValidResponse() {
        return isSuccess() && responseSequenceNumber == sequenceNumber;
    }

    @Override
    protected void encodeWitnessParameters(NdrBuffer buf) throws NdrException {
        // Encode input parameters for WitnessHeartbeat