com.lyncode.pal.model.PalTestGroup 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.model;

import com.lyncode.jtwig.JtwigModelMap;
import com.lyncode.pal.render.model.GroupModel;
import com.lyncode.pal.utils.JtwigUtils;
import org.apache.commons.io.FileUtils;

import java.io.File;
import java.util.*;

public class PalTestGroup implements Comparable {
    public static Collection fromTestCases(Collection testCases) {
        Map> packages = new HashMap>();
        for (PalTestCase testCase : testCases) {
            String name = testCase.getPackage().getName();
            if (!packages.containsKey(name))
                packages.put(name, new ArrayList());
            packages.get(name).add(testCase);
        }

        Collection groups = new PriorityQueue();
        for (String aPackage : packages.keySet()) {
            groups.add(new PalTestGroup(aPackage, packages.get(aPackage)));
        }

        return groups;
    }

    private final String aPackage;
    private final List testCases;

    public PalTestGroup(String aPackage, List palTestCases) {
        this.aPackage = aPackage;
        this.testCases = palTestCases;
    }

    @Override
    public int compareTo(PalTestGroup o) {
        return aPackage.compareTo(o.aPackage);
    }

    public int count(Status status) {
        int sum = 0;
        for (PalTestCase testCase : testCases) {
            sum += testCase.count(status);
        }
        return sum;
    }

    public void render() {
        File outputFile = new File(FileUtils.getTempDirectory(), String.format("%s.html", getPackage()));
        String templateLocation = "/pal/templates/pages/group.twig.html";

        try {
            JtwigUtils.renderTo(outputFile, templateLocation,
                    new JtwigModelMap()
                            .withModelAttribute("group", new GroupModel(this))
            );
            for (PalTestCase testCase : testCases) {
                testCase.render();
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public List testCases() {
        return testCases;
    }

    public String getPackage() {
        return aPackage;
    }
}