From babc04f7960e04a2b1868e40ab204ca3a6d66332 Mon Sep 17 00:00:00 2001
From: Leway Colin
Date: Fri, 28 Jun 2019 09:32:57 +0800
Subject: Add Test for AddDefaults phase (#1106)

* Add Test for AddDefaults phase

* Refactor AddDefaultsSpec
---
 .../options/phases/AddDefaultsSpec.scala           | 27 ++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 src/test/scala/firrtlTests/options/phases/AddDefaultsSpec.scala

(limited to 'src')

diff --git a/src/test/scala/firrtlTests/options/phases/AddDefaultsSpec.scala b/src/test/scala/firrtlTests/options/phases/AddDefaultsSpec.scala
new file mode 100644
index 00000000..362d139f
--- /dev/null
+++ b/src/test/scala/firrtlTests/options/phases/AddDefaultsSpec.scala
@@ -0,0 +1,27 @@
+// See LICENSE for license details.
+
+package firrtlTests.options.phases
+
+import org.scalatest.{FlatSpec, Matchers}
+
+import firrtl.options.{Phase, TargetDirAnnotation}
+import firrtl.options.phases.AddDefaults
+
+class AddDefaultsSpec extends FlatSpec with Matchers {
+
+  class Fixture {
+    val phase: Phase = new AddDefaults
+    val targetDir = TargetDirAnnotation("foo")
+    val defaultDir = TargetDirAnnotation(".")
+  }
+
+  behavior of classOf[AddDefaults].toString
+
+  it should "add a TargetDirAnnotation if it does not exist" in new Fixture {
+    phase.transform(Seq.empty).toSeq should be (Seq(defaultDir))
+  }
+
+  it should "don't add a TargetDirAnnotation if it exists" in new Fixture {
+    phase.transform(Seq(targetDir)).toSeq should be (Seq(targetDir))
+  }
+}
-- 
cgit v1.2.3