From 16c89b963e2058611298c19ec4f8c5c5a5ea2e0e Mon Sep 17 00:00:00 2001
From: OLEGSHA <kvadropups@gmail.com>
Date: Sat, 2 Jan 2021 11:26:06 +0300
Subject: [PATCH] Fixed test generator not populating chunks that were loaded
 from disk

---
 .../windcorp/progressia/test/gen/TestWorldGenerator.java | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/main/java/ru/windcorp/progressia/test/gen/TestWorldGenerator.java b/src/main/java/ru/windcorp/progressia/test/gen/TestWorldGenerator.java
index 2304813..3981d61 100644
--- a/src/main/java/ru/windcorp/progressia/test/gen/TestWorldGenerator.java
+++ b/src/main/java/ru/windcorp/progressia/test/gen/TestWorldGenerator.java
@@ -12,6 +12,7 @@ import ru.windcorp.progressia.common.world.ChunkData;
 import ru.windcorp.progressia.common.world.Coordinates;
 import ru.windcorp.progressia.common.world.DecodingException;
 import ru.windcorp.progressia.common.world.WorldData;
+import ru.windcorp.progressia.common.world.WorldDataListener;
 import ru.windcorp.progressia.common.world.block.BlockData;
 import ru.windcorp.progressia.common.world.block.BlockDataRegistry;
 import ru.windcorp.progressia.common.world.block.BlockFace;
@@ -24,6 +25,13 @@ public class TestWorldGenerator extends AbstractWorldGenerator<Boolean> {
 	
 	public TestWorldGenerator(WorldLogic world) {
 		super("Test:WorldGenerator", Boolean.class);
+		
+		world.getData().addListener(new WorldDataListener() {
+			@Override
+			public void onChunkLoaded(WorldData world, ChunkData chunk) {
+				findAndPopulate(chunk.getPosition(), world);
+			}
+		});
 	}
 	
 	@Override
@@ -45,7 +53,6 @@ public class TestWorldGenerator extends AbstractWorldGenerator<Boolean> {
 	public ChunkData generate(Vec3i chunkPos, WorldData world) {
 		ChunkData chunk = generateUnpopulated(chunkPos, world);
 		world.addChunk(chunk);
-		findAndPopulate(chunkPos, world);
 		return chunk;
 	}