minecraft-src/net/minecraft/gametest/framework/MultipleTestTracker.java
2025-07-04 01:41:11 +03:00

116 lines
3.4 KiB
Java

package net.minecraft.gametest.framework;
import com.google.common.collect.Lists;
import java.util.Collection;
import java.util.function.Consumer;
import java.util.stream.Collectors;
public class MultipleTestTracker {
private static final char NOT_STARTED_TEST_CHAR = ' ';
private static final char ONGOING_TEST_CHAR = '_';
private static final char SUCCESSFUL_TEST_CHAR = '+';
private static final char FAILED_OPTIONAL_TEST_CHAR = 'x';
private static final char FAILED_REQUIRED_TEST_CHAR = 'X';
private final Collection<GameTestInfo> tests = Lists.<GameTestInfo>newArrayList();
private final Collection<GameTestListener> listeners = Lists.<GameTestListener>newArrayList();
public MultipleTestTracker() {
}
public MultipleTestTracker(Collection<GameTestInfo> testInfos) {
this.tests.addAll(testInfos);
}
public void addTestToTrack(GameTestInfo testInfo) {
this.tests.add(testInfo);
this.listeners.forEach(testInfo::addListener);
}
public void addListener(GameTestListener testListener) {
this.listeners.add(testListener);
this.tests.forEach(gameTestInfo -> gameTestInfo.addListener(testListener));
}
public void addFailureListener(Consumer<GameTestInfo> onFail) {
this.addListener(new GameTestListener() {
@Override
public void testStructureLoaded(GameTestInfo testInfo) {
}
@Override
public void testPassed(GameTestInfo test, GameTestRunner runner) {
}
@Override
public void testFailed(GameTestInfo test, GameTestRunner runner) {
onFail.accept(test);
}
@Override
public void testAddedForRerun(GameTestInfo oldTest, GameTestInfo newTest, GameTestRunner runner) {
}
});
}
public int getFailedRequiredCount() {
return (int)this.tests.stream().filter(GameTestInfo::hasFailed).filter(GameTestInfo::isRequired).count();
}
public int getFailedOptionalCount() {
return (int)this.tests.stream().filter(GameTestInfo::hasFailed).filter(GameTestInfo::isOptional).count();
}
public int getDoneCount() {
return (int)this.tests.stream().filter(GameTestInfo::isDone).count();
}
public boolean hasFailedRequired() {
return this.getFailedRequiredCount() > 0;
}
public boolean hasFailedOptional() {
return this.getFailedOptionalCount() > 0;
}
public Collection<GameTestInfo> getFailedRequired() {
return (Collection<GameTestInfo>)this.tests.stream().filter(GameTestInfo::hasFailed).filter(GameTestInfo::isRequired).collect(Collectors.toList());
}
public Collection<GameTestInfo> getFailedOptional() {
return (Collection<GameTestInfo>)this.tests.stream().filter(GameTestInfo::hasFailed).filter(GameTestInfo::isOptional).collect(Collectors.toList());
}
public int getTotalCount() {
return this.tests.size();
}
public boolean isDone() {
return this.getDoneCount() == this.getTotalCount();
}
public String getProgressBar() {
StringBuffer stringBuffer = new StringBuffer();
stringBuffer.append('[');
this.tests.forEach(gameTestInfo -> {
if (!gameTestInfo.hasStarted()) {
stringBuffer.append(' ');
} else if (gameTestInfo.hasSucceeded()) {
stringBuffer.append('+');
} else if (gameTestInfo.hasFailed()) {
stringBuffer.append((char)(gameTestInfo.isRequired() ? 'X' : 'x'));
} else {
stringBuffer.append('_');
}
});
stringBuffer.append(']');
return stringBuffer.toString();
}
public String toString() {
return this.getProgressBar();
}
public void remove(GameTestInfo gameTestInfo) {
this.tests.remove(gameTestInfo);
}
}