From 3249118320e72bd7164e4cf446b02309cc199540 Mon Sep 17 00:00:00 2001
From: MEFThunders7035 <157488669+MEFThunders7035@users.noreply.github.com>
Date: Mon, 26 Feb 2024 23:29:50 +0300
Subject: [PATCH] add tests for the color sensor

---
 src/test/java/ColorSensorV3WrapperTests.java | 38 ++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 src/test/java/ColorSensorV3WrapperTests.java

diff --git a/src/test/java/ColorSensorV3WrapperTests.java b/src/test/java/ColorSensorV3WrapperTests.java
new file mode 100644
index 0000000..a6a31cb
--- /dev/null
+++ b/src/test/java/ColorSensorV3WrapperTests.java
@@ -0,0 +1,38 @@
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import edu.wpi.first.hal.HAL;
+import edu.wpi.first.wpilibj.I2C.Port;
+import frc.utils.sim_utils.ColorSensorV3Wrapped;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+public class ColorSensorV3WrapperTests {
+  private ColorSensorV3Wrapped colorSensor;
+
+  @BeforeEach
+  public void setUp() {
+    HAL.initialize(500, 0); // init HAL just in case
+    colorSensor = new ColorSensorV3Wrapped(Port.kMXP);
+  }
+
+  @AfterEach
+  public void tearDown() {
+    colorSensor.close();
+  }
+
+  @Test
+  void testColorSensorV3Wrapping() {
+    ColorSensorV3Wrapped.setRGBD(2500, 300, 100, 900);
+    assertEquals(2500, colorSensor.getRed());
+    assertEquals(300, colorSensor.getGreen());
+    assertEquals(100, colorSensor.getBlue());
+    assertEquals(900, colorSensor.getProximity());
+
+    ColorSensorV3Wrapped.setRGBD(0, 0, 0, 0);
+    assertEquals(0, colorSensor.getRed());
+    assertEquals(0, colorSensor.getGreen());
+    assertEquals(0, colorSensor.getBlue());
+    assertEquals(0, colorSensor.getProximity());
+  }
+}