1 files changed, 58 insertions, 0 deletions
diff --git a/src/test/scala/chiselTests/aop/InjectionSpec.scala b/src/test/scala/chiselTests/aop/InjectionSpec.scala
new file mode 100644
index 00000000..6c022d60
--- /dev/null
+++ b/src/test/scala/chiselTests/aop/InjectionSpec.scala
@@ -0,0 +1,58 @@
+// See LICENSE for license details.
+
+package chiselTests.aop
+
+import chisel3.testers.BasicTester
+import chiselTests.ChiselFlatSpec
+import chisel3._
+import chisel3.aop.injecting.InjectingAspect
+
+class AspectTester(results: Seq[Int]) extends BasicTester {
+  val values = VecInit(results.map(_.U))
+  val counter = RegInit(0.U(results.length.W))
+  counter := counter + 1.U
+  when(counter >= values.length.U) {
+    stop()
+  }.otherwise {
+    when(reset.asBool() === false.B) {
+      printf("values(%d) = %d\n", counter, values(counter))
+      assert(counter === values(counter))
+    }
+  }
+}
+
+class InjectionSpec extends ChiselFlatSpec {
+  val correctValueAspect = InjectingAspect(
+    {dut: AspectTester => Seq(dut)},
+    {dut: AspectTester =>
+      for(i <- 0 until dut.values.length) {
+        dut.values(i) := i.U
+      }
+    }
+  )
+
+  val wrongValueAspect = InjectingAspect(
+    {dut: AspectTester => Seq(dut)},
+    {dut: AspectTester =>
+      for(i <- 0 until dut.values.length) {
+        dut.values(i) := (i + 1).U
+      }
+    }
+  )
+
+  "Test" should "pass if inserted the correct values" in {
+    assertTesterPasses{ new AspectTester(Seq(0, 1, 2)) }
+  }
+  "Test" should "fail if inserted the wrong values" in {
+    assertTesterFails{ new AspectTester(Seq(9, 9, 9)) }
+  }
+  "Test" should "pass if pass wrong values, but correct with aspect" in {
+    assertTesterPasses({ new AspectTester(Seq(9, 9, 9))} , Nil, Seq(correctValueAspect))
+  }
+  "Test" should "pass if pass wrong values, then wrong aspect, then correct aspect" in {
+    assertTesterPasses({ new AspectTester(Seq(9, 9, 9))} , Nil, Seq(wrongValueAspect, correctValueAspect))
+  }
+  "Test" should "fail if pass wrong values, then correct aspect, then wrong aspect" in {
+    assertTesterFails({ new AspectTester(Seq(9, 9, 9))} , Nil, Seq(correctValueAspect, wrongValueAspect))
+  }
+}