org.citrusframework.xml.container.Assert 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.container;

import jakarta.xml.bind.annotation.XmlAttribute;
import jakarta.xml.bind.annotation.XmlElement;
import jakarta.xml.bind.annotation.XmlRootElement;

import org.citrusframework.TestActionBuilder;
import org.citrusframework.spi.ReferenceResolver;
import org.citrusframework.spi.ReferenceResolverAware;
import org.citrusframework.xml.TestActions;

@XmlRootElement(name = "assert")
public class Assert implements TestActionBuilder, ReferenceResolverAware {

    private final org.citrusframework.container.Assert.Builder builder = new org.citrusframework.container.Assert.Builder();

    private ReferenceResolver referenceResolver;

    @Override
    public org.citrusframework.container.Assert build() {
        builder.getActions().stream()
                .filter(action -> action instanceof ReferenceResolverAware)
                .forEach(action -> ((ReferenceResolverAware) action).setReferenceResolver(referenceResolver));

        return builder.build();
    }

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

    @XmlAttribute
    public Assert setException(String type) {
        builder.exception(type);
        return this;
    }

    @XmlAttribute
    public Assert setMessage(String message) {
        builder.message(message);
        return this;
    }

    @XmlElement(required = true)
    public Assert setWhen(TestActions actions) {
        builder.actions(actions.getActionBuilders().toArray(TestActionBuilder[]::new));
        return this;
    }

    @Override
    public void setReferenceResolver(ReferenceResolver referenceResolver) {
        this.referenceResolver = referenceResolver;
    }
}