org.jboss.windup.rules.apps.javaee.TechnologyUsageStatisticsModelExists Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of windup-rules-java-ee Show documentation
The newest version!
package org.jboss.windup.rules.apps.javaee;

import java.util.Set;

import org.apache.commons.lang3.StringUtils;
import org.jboss.windup.config.GraphRewrite;
import org.jboss.windup.config.condition.GraphCondition;
import org.jboss.windup.reporting.model.TechnologyUsageStatisticsModel;
import org.ocpsoft.rewrite.context.EvaluationContext;

/**
 * This is primarily used by tests to locate a {@link TechnologyUsageStatisticsModel} based upon the given parameters.
 *
 * @author Jesse Sightler
 */
public class TechnologyUsageStatisticsModelExists extends GraphCondition {
    private String technologyName;
    private Set expectedTags;
    private int expectedCount;

    public TechnologyUsageStatisticsModelExists() {
    }

    public TechnologyUsageStatisticsModelExists(String technologyName, int expectedCount, Set expectedTags) {
        this.technologyName = technologyName;
        this.expectedCount = expectedCount;
        this.expectedTags = expectedTags;
    }

    @Override
    public boolean evaluate(GraphRewrite event, EvaluationContext context) {
        TechnologyUsageStatisticsService service = new TechnologyUsageStatisticsService(event.getGraphContext());
        Iterable models = service.findAllByProperty(TechnologyUsageStatisticsModel.NAME, this.technologyName);
//        Iterable models = service.findAll();

        boolean result = false;
        for (TechnologyUsageStatisticsModel model : models) {
            if (matchesExpectations(model))
                result = true;
        }

        return result;
    }

    private boolean matchesExpectations(TechnologyUsageStatisticsModel model) {
        if (!StringUtils.equals(this.technologyName, model.getName()))
            return false;

        if (this.expectedCount != model.getOccurrenceCount())
            return false;

        // expectedTags could contain both "Java EE" and "Embedded"
        // while the model could just have one of them
        // so that's why is expectedTags that can contain model's tags
        return this.expectedTags.containsAll(model.getTags());
    }

    public String getTechnologyName() {
        return technologyName;
    }

    public Set getExpectedTags() {
        return expectedTags;
    }

    public int getExpectedCount() {
        return expectedCount;
    }
}