com.consol.citrus.message.builder.ReceiveMessageBuilderSupport Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2020 the original author or authors.
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.consol.citrus.message.builder;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

import com.consol.citrus.actions.ReceiveMessageAction;
import com.consol.citrus.message.MessageProcessor;
import com.consol.citrus.validation.HeaderValidator;
import com.consol.citrus.validation.MessageValidator;
import com.consol.citrus.validation.ValidationContextAdapter;
import com.consol.citrus.validation.ValidationProcessor;
import com.consol.citrus.validation.context.ValidationContext;
import com.consol.citrus.variable.VariableExtractor;

/**
 * @author Christoph Deppisch
 */
public class ReceiveMessageBuilderSupport, S extends ReceiveMessageBuilderSupport>
        extends MessageBuilderSupport {

    private final List controlMessageProcessors = new ArrayList<>();

    private boolean headerNameIgnoreCase = false;

    protected ReceiveMessageBuilderSupport(B delegate) {
        super(delegate);
    }

    /**
     * Adds a custom timeout to this message receiving action.
     *
     * @param receiveTimeout
     * @return The modified receive message action builder
     */
    public S timeout(final long receiveTimeout) {
        delegate.timeout(receiveTimeout);
        return self;
    }

    /**
     * Validate header names with case insensitive keys.
     *
     * @param value
     * @return The modified receive message action builder
     */
    public S headerNameIgnoreCase(final boolean value) {
        this.headerNameIgnoreCase = value;
        return self;
    }

    /**
     * Adds a validation context.
     * @param validationContext
     * @return The modified receive message action builder
     */
    public S validate(final ValidationContext.Builder validationContext) {
        delegate.validate(validationContext);
        return self;
    }

    /**
     * Adds a validation context.
     * @param validationContext
     * @return The modified receive message action builder
     */
    public S validate(final ValidationContext validationContext) {
        return validate((ValidationContext.Builder) () -> validationContext);
    }

    /**
     * Adds a validation context.
     * @param adapter
     * @return The modified receive message action builder
     */
    public S validate(final ValidationContextAdapter adapter) {
        return validate(adapter.asValidationContext());
    }

    /**
     * Sets validation contexts.
     * @param validationContexts
     * @return The modified receive message action builder
     */
    public S validate(final List> validationContexts) {
        delegate.validate(validationContexts);
        return self;
    }

    /**
     * Sets validation contexts.
     * @param validationContexts
     * @return The modified receive message action builder
     */
    public S validate(ValidationContext.Builder ... validationContexts) {
        return validate(Arrays.asList(validationContexts));
    }

    /**
     * Sets message selector string.
     *
     * @param messageSelector
     * @return The modified receive message action builder
     */
    public S selector(final String messageSelector) {
        delegate.selector(messageSelector);
        return self;
    }

    /**
     * Sets message selector elements.
     *
     * @param messageSelector
     * @return The modified receive message action builder
     */
    public S selector(final Map messageSelector) {
        delegate.selector(messageSelector);
        return self;
    }

    /**
     * Sets explicit message validators for this receive action.
     *
     * @param validator
     * @return The modified receive message action builder
     */
    public S validator(final MessageValidator validator) {
        delegate.validator(validator);
        return self;
    }

    /**
     * Sets explicit message validators for this receive action.
     *
     * @param validators
     * @return The modified receive message action builder
     */
    @SafeVarargs
    public final S validators(final MessageValidator... validators) {
        return validators(Arrays.asList(validators));
    }

    /**
     * Sets explicit message validators for this receive action.
     *
     * @param validators
     * @return The modified receive message action builder
     */
    public S validators(final List> validators) {
        delegate.validators(validators);
        return self;
    }

    /**
     * Sets explicit message validators by name.
     *
     * @param validatorNames
     * @return The modified receive message action builder
     */
    public S validator(final String... validatorNames) {
        delegate.validator(validatorNames);
        return self;
    }

    /**
     * Sets explicit header validator for this receive action.
     *
     * @param validators
     * @return The modified receive message action builder
     */
    public S validator(final HeaderValidator... validators) {
        delegate.validator(validators);
        return self;
    }

    /**
     * Adds validation processor to the receive action for validating
     * the received message with Java code.
     *
     * @param processor
     * @return The modified receive message action builder
     */
    public B validate(final ValidationProcessor processor) {
        return delegate.validate(processor);
    }

    @Override
    public S process(MessageProcessor processor) {
        if (processor instanceof VariableExtractor) {
            super.process(processor);
        } else {
            this.controlMessageProcessors.add(processor);
        }
        return self;
    }

    public List getControlMessageProcessors() {
        return controlMessageProcessors;
    }

    public boolean isHeaderNameIgnoreCase() {
        return headerNameIgnoreCase;
    }
}