com.consol.citrus.dsl.builder.AbstractTestContainerBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of citrus-java-dsl Show documentation
There is a newer version: 3.4.1
/*
 * Copyright 2006-2015 the original author or authors.
 *
 * Licensed 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.dsl.builder;

import com.consol.citrus.TestAction;
import com.consol.citrus.container.TestActionContainer;
import com.consol.citrus.dsl.actions.DelegatingTestAction;
import com.consol.citrus.dsl.design.TestDesigner;
import com.consol.citrus.dsl.runner.TestRunner;

import java.util.List;

/**
 * Abstract container builder takes care on calling the container runner when actions are placed in the container.
 * @author Christoph Deppisch
 * @since 2.3
 */
public abstract class AbstractTestContainerBuilder extends AbstractTestActionBuilder implements TestActionContainerBuilder {

    /** The test runner */
    protected TestRunner runner;

    /** The test designer */
    protected TestDesigner designer;

    /** The action container */
    protected final T container;

    /**
     * Default constructor with test runner and test action.
     * @param runner
     * @param container
     */
    public AbstractTestContainerBuilder(TestRunner runner, T container) {
        super(container);
        this.runner = runner;
        this.container = container;
    }

    /**
     * Default constructor.
     * @param designer
     * @param container
     */
    public AbstractTestContainerBuilder(TestDesigner designer, T container) {
        super(container);
        this.designer = designer;
        this.container = container;
    }

    /**
     * Delegates container execution to container runner or fills container with actions that were not added before
     * when using anonymous test action implementations for instance.
     * @param actions
     * @return
     */
    public T actions(TestAction ... actions) {
        for (int i = 0; i < actions.length; i++) {
            TestAction currentAction = getAction(actions[i]);

            if (currentAction instanceof com.consol.citrus.dsl.runner.ApplyTestBehaviorAction ||
                    currentAction instanceof com.consol.citrus.dsl.design.ApplyTestBehaviorAction) {
                continue;
            } else if (container.getActions().size() == i) {
                container.addTestAction(currentAction);
            } else if (container.getActions().get(i) instanceof DelegatingTestAction) {
                if (currentAction instanceof DelegatingTestAction &&
                        !((DelegatingTestAction) currentAction).getDelegate().equals(((DelegatingTestAction)container.getActions().get(i)).getDelegate())) {
                    container.getActions().add(i, ((DelegatingTestAction) currentAction).getDelegate());
                } else if (!(currentAction instanceof DelegatingTestAction) &&
                        !currentAction.equals(((DelegatingTestAction)container.getActions().get(i)).getDelegate())) {
                    container.getActions().add(i, currentAction);
                }
            } else if (!container.getActions().get(i).equals(currentAction)) {
                container.getActions().add(i, currentAction);
            }
        }

        if (runner != null) {
            return runner.run(container);
        } else {
            designer.action(container);
            return container;
        }
    }

    /**
     * Get action, either through action builder build method or action itself.
     * @param action
     * @return
     */
    private TestAction getAction(TestAction action) {
        if (action instanceof TestActionBuilder) {
            return ((TestActionBuilder) action).build();
        } else {
            return action;
        }
    }

    @Override
    public List getActions() {
        return super.build().getActions();
    }
}