org.citrusframework.xml.actions.CreateVariables Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of citrus-xml Show documentation
Citrus XML integration
The newest version!
/*
 * 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.xml.actions;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

import jakarta.xml.bind.annotation.XmlAccessType;
import jakarta.xml.bind.annotation.XmlAccessorType;
import jakarta.xml.bind.annotation.XmlAttribute;
import jakarta.xml.bind.annotation.XmlElement;
import jakarta.xml.bind.annotation.XmlRootElement;
import jakarta.xml.bind.annotation.XmlType;
import jakarta.xml.bind.annotation.XmlValue;
import org.citrusframework.TestActionBuilder;
import org.citrusframework.actions.CreateVariablesAction;
import org.citrusframework.exceptions.CitrusRuntimeException;
import org.citrusframework.util.FileUtils;

@XmlRootElement(name = "create-variables")
public class CreateVariables implements TestActionBuilder {

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

    @XmlElement(name = "variable", required = true)
    protected List variables = new ArrayList<>();

    @Override
    public CreateVariablesAction build() {
        for (Variable variable : variables) {
            if (variable.getVariableValue() != null) {
                if (variable.getVariableValue().getScript() != null) {
                    if (variable.getVariableValue().getScript().getFile() != null) {
                        try {
                            builder.variable(variable.getName(),
                                    FileUtils.readToString(FileUtils.getFileResource(variable.getVariableValue().getScript().getFile())));
                        } catch (IOException e) {
                            throw new CitrusRuntimeException("Failed to read variable script file resource", e);
                        }
                    } else {
                        builder.variable(variable.getName(), String.format("script:<%s>%s",
                                Optional.ofNullable(variable.getVariableValue().getScript().getType()).orElse("groovy"),
                                variable.getVariableValue().getScript().getValue()));
                    }
                } else {
                    builder.variable(variable.getName(), variable.getVariableValue().getData());
                }
            } else {
                builder.variable(variable.getName(), variable.getValue());
            }
        }

        return builder.build();
    }

    @XmlElement
    public CreateVariables setDescription(String value) {
        builder.description(value);
        return this;
    }

    public CreateVariables setVariables(List variables) {
        this.variables = variables;
        return this;
    }

    @XmlAccessorType(XmlAccessType.FIELD)
    @XmlType(name = "", propOrder = {
            "variableValue"
    })
    public static class Variable {

        @XmlElement(name = "value")
        protected Value variableValue;
        @XmlAttribute(name = "name", required = true)
        protected String name;
        @XmlAttribute(name = "value")
        protected String value;

        public Variable.Value getVariableValue() {
            return variableValue;
        }

        public void setVariableValue(Variable.Value value) {
            this.variableValue = value;
        }

        public String getName() {
            return name;
        }

        public void setName(String value) {
            this.name = value;
        }

        public String getValue() {
            return value;
        }

        public void setValue(String value) {
            this.value = value;
        }

        @XmlAccessorType(XmlAccessType.FIELD)
        @XmlType(name = "", propOrder = {
                "script",
                "data"
        })
        public static class Value {

            @XmlElement
            protected Script script;
            @XmlElement
            protected String data;

            public Script getScript() {
                return script;
            }

            public void setScript(Script value) {
                this.script = value;
            }

            public String getData() {
                return data;
            }

            public void setData(String value) {
                this.data = value;
            }
        }

        @XmlAccessorType(XmlAccessType.FIELD)
        @XmlType(name = "ScriptType", propOrder = {
                "value"
        })
        public static class Script {

            @XmlValue
            protected String value;
            @XmlAttribute(name = "type", required = true)
            protected String type;
            @XmlAttribute(name = "file")
            protected String file;

            public String getValue() {
                return value;
            }

            public void setValue(String value) {
                this.value = value;
            }

            public String getType() {
                return type;
            }

            public void setType(String value) {
                this.type = value;
            }

            public String getFile() {
                return file;
            }

            public void setFile(String value) {
                this.file = value;
            }
        }
    }
}