apps/CtsVerifier/src/com/android/cts/verifier/TestResultHistoryCollection.java

package com.android.cts.verifier;

import com.android.compatibility.common.util.TestResultHistory;

import java.io.Serializable;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

public class TestResultHistoryCollection implements Serializable {

    private static final long serialVersionUID = 0L;
    private final Set<TestResultHistory> mHistoryCollection = new HashSet<>();

    /**
     * Covert object to set.
     *
     * @return A set of test result history.
     */
    public Set<TestResultHistory> asSet() {
        return mHistoryCollection;
    }

    /**
     * Add a test result history with test name, start time, end time and isAutomated.
     *
     * @param test a string of test name.
     * @param start start time of a test.
     * @param end end time of a test.
     * @param isAutomated whether test case was executed through automation.
     */
    public void add(String test, long start, long end, boolean isAutomated) {
        Set<TestResultHistory.ExecutionRecord> executionRecords
                = new HashSet<TestResultHistory.ExecutionRecord> ();
        executionRecords.add(new TestResultHistory.ExecutionRecord(start, end, isAutomated));
        mHistoryCollection.add(new TestResultHistory(test, executionRecords));
    }

    /**
     * Add test result histories for tests containing test name and a set of ExecutionRecords
     *
     * @param test test name.
     * @param executionRecords set of ExecutionRecords.
     */
    public void addAll(String test, Set<TestResultHistory.ExecutionRecord> executionRecords) {
        TestResultHistory history = new TestResultHistory(test, executionRecords);
        boolean match = false;
        for (TestResultHistory resultHistory: mHistoryCollection) {
            if (resultHistory.getTestName().equals(test)) {
                resultHistory.getExecutionRecords().addAll(executionRecords);
                match = true;
                break;
            }
        }
        if (match == false) {
            mHistoryCollection.add(history);
        }
    }

    /**
     * Merge test with its sub-tests result histories.
     *
     * @param prefix optional test name prefix to apply.
     * @param resultHistoryCollection a set of test result histories.
     */
    public void merge(String prefix, TestResultHistoryCollection resultHistoryCollection) {
        if (resultHistoryCollection != null) {
            resultHistoryCollection.asSet().forEach(t-> addAll(
                prefix != null
                        ? prefix + ":" + t.getTestName()
                        : t.getTestName(), t.getExecutionRecords()));
        }
    }

    /**
     * Merge test with its sub-tests result histories.
     *
     * @param prefix optional test name prefix to apply.
     * @param resultHistories a list of test result history collection.
     */
    public void merge(String prefix, List<TestResultHistoryCollection> resultHistories) {
        resultHistories.forEach(resultHistoryCollection -> merge(prefix, resultHistoryCollection));
    }
}