TestCollection.java

package hudson.plugins.javatest_report;

import org.kohsuke.stapler.StaplerRequest;
import org.kohsuke.stapler.StaplerResponse;

import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

/**
 * {@link TestObject} that is a collection of other {@link TestObject}s.
 *
 * @param <C>
 *      Type of the child objects in this collection.
 * @param <S>
 *      The derived type of {@link TestCollection} (the same design pattern as you seen in {@link Enum})
 *
 * @author Kohsuke Kawaguchi
 * @author Rama Pulavarthi
 * @author Vladimir Ralev
 */
public abstract class TestCollection<
    S extends TestCollection<S,C>,
    C extends TestObject<C>> extends TestObject<S> {

    /**
     * All {@link Test}s keyed by their ID.
     */
    private final Map<String,C> tests = new TreeMap<String,C>();
    /**
     * All Failed Tests keyed by their ID.
     */
    private final Map<String,C> failedTests = new TreeMap<String,C>();
    private final Map<String,C> skippedTests = new TreeMap<String,C>();
    
    private final Map<String,Package> packages = new TreeMap<String,Package>();
    
    private int totalCount;
    private int failCount;
    private int skippedCount;

    public Collection<C> getChildren() {
        return tests.values();
    }

    public Collection<C> getFailedTests() {
        return failedTests.values();
    }
    
    public Collection<C> getSkippedTests() {
        return skippedTests.values();
    }

    public int getTotalCount() {
        return totalCount;
    }

    public int getFailCount() {
        return failCount;
    }
    
    public int getSkippedCount() {
        return skippedCount;
    }
    /**
     * Returns the caption of the children. Used in the view.
     */
    public abstract String getChildTitle();

    /**
     * Gets a {@link Test} by its id.
     */
    public C get(String id) {
        return tests.get(id);
    }

    /**
     * Adds a new child {@link TestObject} to this.
     * <p>
     * For Digester.
     */
    public void add(C t) {
        tests.put(t.getId(),t);
        if(t.getStatus() == Status.SKIP)
            skippedTests.put(t.getId(),t);
        else if(t.getStatus() != Status.PASS)
           failedTests.put(t.getId(),t);
        if(t.getStatus() != Status.SKIP)
           totalCount += t.getTotalCount();
        failCount += t.getFailCount();
        skippedCount += t.getSkippedCount();
        t.parent = this;
        if( (t.getStatus() == Status.SKIP) ||
        		!(this.getClass().isAssignableFrom(Suite.class))) return;
        String fqcn = t.getName();
        int packagePosition = fqcn.lastIndexOf("/");
        if(packagePosition>0 && packagePosition<fqcn.length())
        {
        	Test test = new Test();
        	test.setId(t.getId());
        	test.setName(t.getName());
        	test.setStatusString(t.getStatus().toString());
        	test.setDescription(t.getDescription());
        	test.addAttribute("logfile", t.getStatusMessage());
        	String packageName = fqcn.substring(0, packagePosition).replaceAll("/",".");
        	
        	String testName = fqcn.substring(packagePosition + 1, fqcn.length());
        	Package c = (Package) packages.get(packageName);
        	if(c != null)
        	{
        		c.add(test);
        	}
        	else
        	{
        		Package newSuite = new Package();
        		newSuite.setId(packageName);
        		newSuite.setName(packageName);
        		newSuite.add(test);
        		newSuite.parent = this;
        		packages.put(packageName, newSuite);
        	}
        }
    }
    
    public String[] getPackages()
    {
    	if(packages == null) 
    	{
    		System.out.println("null"+this.getClass()+this.toString());
    		return null;
    	}
    	Set<String> set = packages.keySet();
    	String[] strs = new String[set.size()];
    	strs = set.toArray(strs);
    	return strs;
    	
    }
    
    public Package getPackageTests(String packageName)
    {
    	return packages.get(packageName);
    }

    // method for stapler
    public C getDynamic(String name, StaplerRequest req, StaplerResponse rsp) {
        return get(name);
    }
}