All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.citrusframework.config.xml.JavaActionParser Maven / Gradle / Ivy

/*
 * Copyright 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 org.citrusframework.config.xml;

import java.util.ArrayList;
import java.util.List;

import org.citrusframework.actions.JavaAction;
import org.citrusframework.config.util.BeanDefinitionParserUtils;
import org.citrusframework.util.StringUtils;
import org.springframework.beans.factory.BeanCreationException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.xml.BeanDefinitionParser;
import org.springframework.beans.factory.xml.ParserContext;
import org.springframework.util.xml.DomUtils;
import org.w3c.dom.Element;

import static java.lang.Boolean.parseBoolean;
import static java.lang.Double.parseDouble;
import static java.lang.Integer.parseInt;
import static java.lang.Long.parseLong;

/**
 * Bean definition parser for java action in test case.
 *
 */
public class JavaActionParser implements BeanDefinitionParser {

    @Override
	public BeanDefinition parse(Element element, ParserContext parserContext) {
        BeanDefinitionBuilder beanDefinition = BeanDefinitionBuilder.rootBeanDefinition(JavaActionFactoryBean.class);

        DescriptionElementParser.doParse(element, beanDefinition);

        BeanDefinitionParserUtils.setPropertyValue(beanDefinition, element.getAttribute("class"), "className");
        BeanDefinitionParserUtils.setPropertyReference(beanDefinition, element.getAttribute("ref"), "instance");

        Element constructorElement = DomUtils.getChildElementByTagName(element, "constructor");
        List arguments = new ArrayList<>();
        if (constructorElement != null) {
            List argumentList = DomUtils.getChildElementsByTagName(constructorElement, "argument");
            for (Element arg : argumentList) {
                arguments.add(resolveArgument(arg.getAttribute("type"), arg.getTextContent()));
            }
            beanDefinition.addPropertyValue("constructorArgs", arguments);
        }

        Element methodElement = DomUtils.getChildElementByTagName(element, "method");
        arguments = new ArrayList<>();
        if (methodElement != null) {
            String methodName = methodElement.getAttribute("name");
            beanDefinition.addPropertyValue("methodName", methodName);

            List argumentList = DomUtils.getChildElementsByTagName(methodElement, "argument");
            for (Element arg : argumentList) {
                arguments.add(resolveArgument(arg.getAttribute("type"), DomUtils.getTextValue(arg)));
            }
            beanDefinition.addPropertyValue("methodArgs", arguments);
        }

        return beanDefinition.getBeanDefinition();
    }

    private Object resolveArgument(String type, String value) {
        if (!StringUtils.hasText(type) || type.equals("String")) {
            return value;
        } else if (type.equals("String[]")) {
            return value.split(",");
        } else if (type.equals("boolean")) {
            return parseBoolean(value);
        }  else if (type.equals("int")) {
            return parseInt(value);
        } else if (type.equals("long")) {
            return parseLong(value);
        } else if (type.equals("double")) {
            return parseDouble(value);
        }

        throw new BeanCreationException("Found unsupported method argument type: '" + type + "'");
    }

    /**
     * Test action factory bean.
     */
    public static class JavaActionFactoryBean extends AbstractTestActionFactoryBean {

        private final JavaAction.Builder builder = new JavaAction.Builder();

        /**
         * Setter for class name
         * @param className
         */
        public void setClassName(String className) {
            builder.className(className);
        }

        /**
         * Setter for constructor args
         * @param constructorArgs
         */
        public void setConstructorArgs(List constructorArgs) {
            builder.constructorArgs(constructorArgs);
        }

        /**
         * Setter for method args
         * @param methodArgs
         */
        public void setMethodArgs(List methodArgs) {
            builder.methodArgs(methodArgs);
        }

        /**
         * Setter for method name
         * @param methodName
         */
        public void setMethodName(String methodName) {
            builder.method(methodName);
        }

        /**
         * Setter for object instance
         * @param instance
         */
        public void setInstance(Object instance) {
            builder.instance(instance);
        }

        @Override
        public JavaAction getObject() throws Exception {
            return builder.build();
        }

        @Override
        public Class getObjectType() {
            return JavaAction.class;
        }

        /**
         * Obtains the builder.
         * @return the builder implementation.
         */
        @Override
        public JavaAction.Builder getBuilder() {
            return builder;
        }
    }
}