com.lyncode.pal.render.model.TestCaseModel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pal-core Show documentation

The newest version!

package com.lyncode.pal.render.model;

import com.google.common.base.Function;
import com.google.common.collect.Collections2;
import com.lyncode.pal.model.PalTestCase;
import com.lyncode.pal.model.PalTestScenario;
import com.lyncode.pal.model.Status;
import com.lyncode.pal.utils.SubtleWordUtils;

import javax.annotation.Nullable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;

public class TestCaseModel {
    private final PalTestCase input;

    public TestCaseModel(PalTestCase input) {
        this.input = input;
    }

    public int count (String status) {
        return input.count(Status.valueOf(status));
    }

    public String name () {
        String simpleName = input.type().getSimpleName();
        if (simpleName.endsWith("Test"))
            simpleName = simpleName.substring(0, simpleName.length() - 4);
        return SubtleWordUtils.sentencify(simpleName, true);
    }

    public boolean hasErrors () {
        return input.count(Status.Failed) > 0;
    }

    public String url () {
        return input.type().getName() + ".html";
    }

    public Collection scenarios () {
        ArrayList models = new ArrayList(Collections2.transform(input.scenarios(), new Function() {
            @Nullable
            @Override
            public ScenarioModel apply(@Nullable PalTestScenario input) {
                return new ScenarioModel(input);
            }
        }));
        Collections.sort(models);
        return models;
    }
}