Renamed project to MoGraSim

[Mograsim.git] / era.mi / src / mograsim / logic / core / tests / TestBitDisplay.java

diff --git a/era.mi/src/mograsim/logic/core/tests/TestBitDisplay.java b/era.mi/src/mograsim/logic/core/tests/TestBitDisplay.java
new file mode 100644 (file)
index 0000000..80eef5f
--- /dev/null
+++ b/era.mi/src/mograsim/logic/core/tests/TestBitDisplay.java
@@ -0,0 +1,47 @@
+package mograsim.logic.core.tests;
+
+import static org.junit.jupiter.api.Assertions.assertArrayEquals;
+
+import java.util.function.LongConsumer;
+
+import mograsim.logic.core.components.BitDisplay;
+import mograsim.logic.core.timeline.Timeline;
+import mograsim.logic.core.types.Bit;
+import mograsim.logic.core.wires.Wire.ReadEnd;
+
+public final class TestBitDisplay extends BitDisplay
+{
+
+       public TestBitDisplay(Timeline timeline, ReadEnd in)
+       {
+               super(timeline, in);
+       }
+
+       public void assertDisplays(Bit... expected)
+       {
+               assertArrayEquals(expected, getDisplayedValue().getBits());
+       }
+
+       public void assertAfterSimulationIs(Bit... expected)
+       {
+               timeline.executeAll();
+               assertDisplays(expected);
+       }
+
+       public void assertAfterSimulationIs(LongConsumer r, Bit... expected)
+       {
+               while (timeline.hasNext())
+               {
+                       timeline.executeNext();
+                       r.accept(timeline.getSimulationTime());
+               }
+               assertDisplays(expected);
+       }
+
+       @Override
+       protected void compute()
+       {
+               super.compute();
+               System.out.println("update: value is " + getDisplayedValue());
+       }
+}