Spaces:
Build error
Build error
File size: 10,009 Bytes
d46f4a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 |
package net.minecraft.gametest.framework;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;
import com.mojang.logging.LogUtils;
import it.unimi.dsi.fastutil.longs.LongArraySet;
import it.unimi.dsi.fastutil.longs.LongSet;
import java.util.Collection;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import javax.annotation.Nullable;
import net.minecraft.Util;
import net.minecraft.network.protocol.game.DebugPackets;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.level.ChunkPos;
import org.slf4j.Logger;
public class GameTestRunner {
public static final int DEFAULT_TESTS_PER_ROW = 8;
private static final Logger LOGGER = LogUtils.getLogger();
final ServerLevel level;
private final GameTestTicker testTicker;
private final List<GameTestInfo> allTestInfos;
private ImmutableList<GameTestBatch> batches;
final List<GameTestBatchListener> batchListeners = Lists.newArrayList();
private final List<GameTestInfo> scheduledForRerun = Lists.newArrayList();
private final GameTestRunner.GameTestBatcher testBatcher;
private boolean stopped = true;
@Nullable
GameTestBatch currentBatch;
private final GameTestRunner.StructureSpawner existingStructureSpawner;
private final GameTestRunner.StructureSpawner newStructureSpawner;
final boolean haltOnError;
protected GameTestRunner(
GameTestRunner.GameTestBatcher p_332546_,
Collection<GameTestBatch> p_332555_,
ServerLevel p_331060_,
GameTestTicker p_335326_,
GameTestRunner.StructureSpawner p_336219_,
GameTestRunner.StructureSpawner p_330306_,
boolean p_342567_
) {
this.level = p_331060_;
this.testTicker = p_335326_;
this.testBatcher = p_332546_;
this.existingStructureSpawner = p_336219_;
this.newStructureSpawner = p_330306_;
this.batches = ImmutableList.copyOf(p_332555_);
this.haltOnError = p_342567_;
this.allTestInfos = this.batches.stream().flatMap(p_325950_ -> p_325950_.gameTestInfos().stream()).collect(Util.toMutableList());
p_335326_.setRunner(this);
this.allTestInfos.forEach(p_325945_ -> p_325945_.addListener(new ReportGameListener()));
}
public List<GameTestInfo> getTestInfos() {
return this.allTestInfos;
}
public void start() {
this.stopped = false;
this.runBatch(0);
}
public void stop() {
this.stopped = true;
if (this.currentBatch != null) {
this.currentBatch.afterBatchFunction().accept(this.level);
}
}
public void rerunTest(GameTestInfo p_335947_) {
GameTestInfo gametestinfo = p_335947_.copyReset();
p_335947_.getListeners().forEach(p_325948_ -> p_325948_.testAddedForRerun(p_335947_, gametestinfo, this));
this.allTestInfos.add(gametestinfo);
this.scheduledForRerun.add(gametestinfo);
if (this.stopped) {
this.runScheduledRerunTests();
}
}
void runBatch(final int p_336310_) {
if (p_336310_ >= this.batches.size()) {
this.runScheduledRerunTests();
} else {
this.currentBatch = this.batches.get(p_336310_);
this.existingStructureSpawner.onBatchStart(this.level);
this.newStructureSpawner.onBatchStart(this.level);
Collection<GameTestInfo> collection = this.createStructuresForBatch(this.currentBatch.gameTestInfos());
String s = this.currentBatch.name();
LOGGER.info("Running test batch '{}' ({} tests)...", s, collection.size());
this.currentBatch.beforeBatchFunction().accept(this.level);
this.batchListeners.forEach(p_325951_ -> p_325951_.testBatchStarting(this.currentBatch));
final MultipleTestTracker multipletesttracker = new MultipleTestTracker();
collection.forEach(multipletesttracker::addTestToTrack);
multipletesttracker.addListener(new GameTestListener() {
private void testCompleted() {
if (multipletesttracker.isDone()) {
GameTestRunner.this.currentBatch.afterBatchFunction().accept(GameTestRunner.this.level);
GameTestRunner.this.batchListeners.forEach(p_329497_ -> p_329497_.testBatchFinished(GameTestRunner.this.currentBatch));
LongSet longset = new LongArraySet(GameTestRunner.this.level.getForcedChunks());
longset.forEach(p_328493_ -> GameTestRunner.this.level.setChunkForced(ChunkPos.getX(p_328493_), ChunkPos.getZ(p_328493_), false));
GameTestRunner.this.runBatch(p_336310_ + 1);
}
}
@Override
public void testStructureLoaded(GameTestInfo p_336002_) {
}
@Override
public void testPassed(GameTestInfo p_334410_, GameTestRunner p_329201_) {
this.testCompleted();
}
@Override
public void testFailed(GameTestInfo p_335430_, GameTestRunner p_330830_) {
if (GameTestRunner.this.haltOnError) {
GameTestRunner.this.currentBatch.afterBatchFunction().accept(GameTestRunner.this.level);
LongSet longset = new LongArraySet(GameTestRunner.this.level.getForcedChunks());
longset.forEach(p_341095_ -> GameTestRunner.this.level.setChunkForced(ChunkPos.getX(p_341095_), ChunkPos.getZ(p_341095_), false));
GameTestTicker.SINGLETON.clear();
} else {
this.testCompleted();
}
}
@Override
public void testAddedForRerun(GameTestInfo p_329460_, GameTestInfo p_328079_, GameTestRunner p_334962_) {
}
});
collection.forEach(this.testTicker::add);
}
}
private void runScheduledRerunTests() {
if (!this.scheduledForRerun.isEmpty()) {
LOGGER.info(
"Starting re-run of tests: {}", this.scheduledForRerun.stream().map(p_325949_ -> p_325949_.getTestFunction().testName()).collect(Collectors.joining(", "))
);
this.batches = ImmutableList.copyOf(this.testBatcher.batch(this.scheduledForRerun));
this.scheduledForRerun.clear();
this.stopped = false;
this.runBatch(0);
} else {
this.batches = ImmutableList.of();
this.stopped = true;
}
}
public void addListener(GameTestBatchListener p_329355_) {
this.batchListeners.add(p_329355_);
}
private Collection<GameTestInfo> createStructuresForBatch(Collection<GameTestInfo> p_335557_) {
return p_335557_.stream().map(this::spawn).flatMap(Optional::stream).toList();
}
private Optional<GameTestInfo> spawn(GameTestInfo p_330408_) {
return p_330408_.getStructureBlockPos() == null ? this.newStructureSpawner.spawnStructure(p_330408_) : this.existingStructureSpawner.spawnStructure(p_330408_);
}
public static void clearMarkers(ServerLevel p_127686_) {
DebugPackets.sendGameTestClearPacket(p_127686_);
}
public static class Builder {
private final ServerLevel level;
private final GameTestTicker testTicker = GameTestTicker.SINGLETON;
private GameTestRunner.GameTestBatcher batcher = GameTestBatchFactory.fromGameTestInfo();
private GameTestRunner.StructureSpawner existingStructureSpawner = GameTestRunner.StructureSpawner.IN_PLACE;
private GameTestRunner.StructureSpawner newStructureSpawner = GameTestRunner.StructureSpawner.NOT_SET;
private final Collection<GameTestBatch> batches;
private boolean haltOnError = false;
private Builder(Collection<GameTestBatch> p_329875_, ServerLevel p_329864_) {
this.batches = p_329875_;
this.level = p_329864_;
}
public static GameTestRunner.Builder fromBatches(Collection<GameTestBatch> p_329486_, ServerLevel p_336096_) {
return new GameTestRunner.Builder(p_329486_, p_336096_);
}
public static GameTestRunner.Builder fromInfo(Collection<GameTestInfo> p_335004_, ServerLevel p_328084_) {
return fromBatches(GameTestBatchFactory.fromGameTestInfo().batch(p_335004_), p_328084_);
}
public GameTestRunner.Builder haltOnError(boolean p_344352_) {
this.haltOnError = p_344352_;
return this;
}
public GameTestRunner.Builder newStructureSpawner(GameTestRunner.StructureSpawner p_329789_) {
this.newStructureSpawner = p_329789_;
return this;
}
public GameTestRunner.Builder existingStructureSpawner(StructureGridSpawner p_342571_) {
this.existingStructureSpawner = p_342571_;
return this;
}
public GameTestRunner.Builder batcher(GameTestRunner.GameTestBatcher p_344402_) {
this.batcher = p_344402_;
return this;
}
public GameTestRunner build() {
return new GameTestRunner(this.batcher, this.batches, this.level, this.testTicker, this.existingStructureSpawner, this.newStructureSpawner, this.haltOnError);
}
}
public interface GameTestBatcher {
Collection<GameTestBatch> batch(Collection<GameTestInfo> p_335979_);
}
public interface StructureSpawner {
GameTestRunner.StructureSpawner IN_PLACE = p_329780_ -> Optional.of(p_329780_.prepareTestStructure().placeStructure().startExecution(1));
GameTestRunner.StructureSpawner NOT_SET = p_329287_ -> Optional.empty();
Optional<GameTestInfo> spawnStructure(GameTestInfo p_334318_);
default void onBatchStart(ServerLevel p_343975_) {
}
}
} |