Add Profile Aware Waypoint

2 files changed, 47 insertions, 8 deletions

diff --git a/src/test/java/de/hysky/skyblocker/utils/waypoint/ProfileAwareWaypointTest.java b/src/test/java/de/hysky/skyblocker/utils/waypoint/ProfileAwareWaypointTest.java
new file mode 100644
index 00000000..9dc5b2b9
--- /dev/null
+++ b/src/test/java/de/hysky/skyblocker/utils/waypoint/ProfileAwareWaypointTest.java
@@ -0,0 +1,38 @@
+package de.hysky.skyblocker.utils.waypoint;
+
+import net.minecraft.util.math.BlockPos;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+
+public class ProfileAwareWaypointTest {
+    @Test
+    void testShouldRender() {
+        ProfileAwareWaypoint waypoint = new ProfileAwareWaypoint(BlockPos.ORIGIN, null, null, null);
+        waypoint.setFound("profile");
+        Assertions.assertTrue(waypoint.shouldRender());
+        waypoint.setFound("");
+        Assertions.assertFalse(waypoint.shouldRender());
+        waypoint.setMissing();
+        Assertions.assertTrue(waypoint.shouldRender());
+    }
+
+    @Test
+    void testGetColorComponents() {
+        ProfileAwareWaypoint waypoint = new ProfileAwareWaypoint(BlockPos.ORIGIN, null, new float[]{0f, 0.5f, 1f}, new float[]{1f, 0.5f, 0f});
+        waypoint.setFound("profile");
+        float[] colorComponents = waypoint.getColorComponents();
+        Assertions.assertEquals(0f, colorComponents[0]);
+        Assertions.assertEquals(0.5f, colorComponents[1]);
+        Assertions.assertEquals(1f, colorComponents[2]);
+        waypoint.setFound("");
+        colorComponents = waypoint.getColorComponents();
+        Assertions.assertEquals(1f, colorComponents[0]);
+        Assertions.assertEquals(0.5f, colorComponents[1]);
+        Assertions.assertEquals(0f, colorComponents[2]);
+        waypoint.setMissing();
+        colorComponents = waypoint.getColorComponents();
+        Assertions.assertEquals(0f, colorComponents[0]);
+        Assertions.assertEquals(0.5f, colorComponents[1]);
+        Assertions.assertEquals(1f, colorComponents[2]);
+    }
+}
diff --git a/src/test/java/de/hysky/skyblocker/utils/waypoint/WaypointTest.java b/src/test/java/de/hysky/skyblocker/utils/waypoint/WaypointTest.java
index 41f5a530..b3be64b7 100644
--- a/src/test/java/de/hysky/skyblocker/utils/waypoint/WaypointTest.java
+++ b/src/test/java/de/hysky/skyblocker/utils/waypoint/WaypointTest.java
@@ -7,10 +7,11 @@ import org.junit.jupiter.api.Test;
 
 public class WaypointTest {
     private Waypoint.Type type;
+    private final float[] colorComponents = new float[]{0f, 0.5f, 1f};
 
     @Test
     void testDefaultConstructor() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0f, 0.5f, 1f});
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents);
         Assertions.assertEquals(BlockPos.ORIGIN, waypoint.pos);
         Assertions.assertEquals(new Box(BlockPos.ORIGIN), waypoint.box);
         Assertions.assertEquals(type, waypoint.typeSupplier.get());
@@ -25,37 +26,37 @@ public class WaypointTest {
 
     @Test
     void testTypeConstructor() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, Waypoint.Type.WAYPOINT, new float[]{0, 0, 0}, Waypoint.DEFAULT_HIGHLIGHT_ALPHA);
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, Waypoint.Type.WAYPOINT, colorComponents, Waypoint.DEFAULT_HIGHLIGHT_ALPHA);
         Assertions.assertEquals(Waypoint.Type.WAYPOINT, waypoint.typeSupplier.get());
     }
 
     @Test
     void testAlphaConstructor() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0, 0, 0}, 1f);
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents, 1f);
         Assertions.assertEquals(1f, waypoint.alpha);
     }
 
     @Test
     void testLineWidthConstructor() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0, 0, 0}, Waypoint.DEFAULT_HIGHLIGHT_ALPHA, 10f);
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents, Waypoint.DEFAULT_HIGHLIGHT_ALPHA, 10f);
         Assertions.assertEquals(10f, waypoint.lineWidth);
     }
 
     @Test
     void testThroughWallsConstructor() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0, 0, 0}, Waypoint.DEFAULT_HIGHLIGHT_ALPHA, Waypoint.DEFAULT_LINE_WIDTH, false);
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents, Waypoint.DEFAULT_HIGHLIGHT_ALPHA, Waypoint.DEFAULT_LINE_WIDTH, false);
         Assertions.assertFalse(waypoint.throughWalls);
     }
 
     @Test
     void testShouldRenderConstructor() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0, 0, 0}, Waypoint.DEFAULT_HIGHLIGHT_ALPHA, Waypoint.DEFAULT_LINE_WIDTH, true, false);
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents, Waypoint.DEFAULT_HIGHLIGHT_ALPHA, Waypoint.DEFAULT_LINE_WIDTH, true, false);
         Assertions.assertFalse(waypoint.shouldRender());
     }
 
     @Test
     void testFound() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0, 0, 0});
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents);
         Assertions.assertTrue(waypoint.shouldRender());
         waypoint.setFound();
         Assertions.assertFalse(waypoint.shouldRender());
@@ -65,7 +66,7 @@ public class WaypointTest {
 
     @Test
     void testType() {
-        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, new float[]{0, 0, 0});
+        Waypoint waypoint = new Waypoint(BlockPos.ORIGIN, () -> type, colorComponents);
         Assertions.assertEquals(type, waypoint.typeSupplier.get());
         type = Waypoint.Type.WAYPOINT;
         Assertions.assertEquals(type, waypoint.typeSupplier.get());