org.pitest.mutationtest.statistics.MutationStatisticsPrecursor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pitest-entry Show documentation

Pitest entry point

There is a newer version: 1.17.1

package org.pitest.mutationtest.statistics;

import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.function.BiFunction;
import java.util.function.Consumer;

import org.pitest.classinfo.ClassName;
import org.pitest.functional.FCollection;
import org.pitest.mutationtest.MutationResult;

class MutationStatisticsPrecursor {
  private final Map mutatorTotalMap  = new HashMap<>();
  private final Set mutatedClasses = new HashSet<>();
  private long                              numberOfTestsRun = 0;

  public void registerResults(final Collection results) {
    results.forEach(register());
  }

  public void registerClass(ClassName mutatedClass) {
    mutatedClasses.add(mutatedClass);
  }

  public Set mutatedClasses() {
    return mutatedClasses;
  }

  private Consumer register() {
    return mr -> {
      MutationStatisticsPrecursor.this.numberOfTestsRun = MutationStatisticsPrecursor.this.numberOfTestsRun
          + mr.getNumberOfTestsRun();
      final String key = mr.getDetails().getId().getMutator();
      ScorePrecursor total = MutationStatisticsPrecursor.this.mutatorTotalMap
          .get(key);
      if (total == null) {
        total = new ScorePrecursor(key);
        MutationStatisticsPrecursor.this.mutatorTotalMap.put(key, total);
      }
      total.registerResult(mr.getStatus());
    };
  }

  public MutationStatistics toStatistics() {
    final Iterable scores = getScores();
    final long totalMutations = FCollection.fold(addTotals(), 0L, scores);
    final long totalDetected = FCollection
        .fold(addDetectedTotals(), 0L, scores);
    final long totalWithCoverage = FCollection.fold(addCoveredTotals(), 0L, scores);
    return new MutationStatistics(scores, totalMutations, totalDetected, totalWithCoverage,
        this.numberOfTestsRun, mutatedClasses());
  }

  Iterable getScores() {
    return FCollection.map(this.mutatorTotalMap.values(), ScorePrecursor::toScore);
  }


  private static BiFunction addTotals() {
    return (a, b) -> a + b.getTotalMutations();
  }

  private static BiFunction addDetectedTotals() {
    return (a, b) -> a + b.getTotalDetectedMutations();
  }

  private static BiFunction addCoveredTotals() {
    return (a, b) -> a + b.getTotalWithCoverage();
  }


}