com.tngtech.archunit.lang.conditions.ContainsOnlyCondition Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2019 TNG Technology Consulting GmbH
 *
 * 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.tngtech.archunit.lang.conditions;

import java.util.Collection;

import com.tngtech.archunit.lang.ArchCondition;
import com.tngtech.archunit.lang.CollectsLines;
import com.tngtech.archunit.lang.ConditionEvent;
import com.tngtech.archunit.lang.ConditionEvents;
import com.tngtech.archunit.lang.conditions.ContainAnyCondition.AnyConditionEvent;

class ContainsOnlyCondition extends ArchCondition> {
    private final ArchCondition condition;

    ContainsOnlyCondition(ArchCondition condition) {
        super("contain only elements that " + condition.getDescription());
        this.condition = condition;
    }

    @Override
    public void check(Collection collection, ConditionEvents events) {
        ConditionEvents subEvents = new ConditionEvents();
        for (T item : collection) {
            condition.check(item, subEvents);
        }
        if (!subEvents.isEmpty()) {
            events.add(new OnlyConditionEvent(collection, subEvents));
        }
    }

    @Override
    public String toString() {
        return getClass().getSimpleName() + "{condition=" + condition + "}";
    }

    static class OnlyConditionEvent implements ConditionEvent {
        private final Collection correspondingObjects;
        private final Collection allowed;
        private final Collection violating;

        private OnlyConditionEvent(Collection correspondingObjects, ConditionEvents events) {
            this(correspondingObjects, events.getAllowed(), events.getViolating());
        }

        OnlyConditionEvent(Collection correspondingObjects,
                Collection allowed,
                Collection violating) {
            this.correspondingObjects = correspondingObjects;
            this.allowed = allowed;
            this.violating = violating;
        }

        @Override
        public boolean isViolation() {
            return !violating.isEmpty();
        }

        @Override
        public void addInvertedTo(ConditionEvents events) {
            events.add(new AnyConditionEvent(correspondingObjects, violating, allowed));
        }

        @Override
        public void describeTo(CollectsLines messages) {
            // NOTE: Do not join the lines here, because at the moment the reported number of violations equals the number of failure messages.
            //       Thus a joined message counts as one violation, in the "only" case, each violation stands by itself though
            //       (as opposed to the "any" case, where only the whole set of violations in combination causes an "any" violation)
            for (ConditionEvent event : violating) {
                event.describeTo(messages);
            }
        }

        @Override
        public void handleWith(Handler handler) {
            for (ConditionEvent event : violating) {
                event.handleWith(handler);
            }
        }

        @Override
        public String toString() {
            return getClass().getSimpleName() + "{" +
                    "correspondingObjects=" + correspondingObjects +
                    ", allowed=" + allowed +
                    ", violating=" + violating +
                    '}';
        }
    }
}