test: TypedAnnotationPlugin

src/test/scala/com/github/tkqubo/typed_annotation/TypedAnnotationPluginSpec.scala

@@ -0,0 +1,100 @@
+package com.github.tkqubo.typed_annotation
+
+import java.net.URLClassLoader
+
+import org.specs2.matcher.Matcher
+import org.specs2.mutable.Specification
+
+import scala.reflect.internal.util.BatchSourceFile
+import scala.tools.nsc.reporters.StoreReporter
+import scala.tools.nsc.util.ClassPath
+import scala.tools.nsc.{Global, Settings}
+
+class TypedAnnotationPluginSpec
+  extends Specification {
+
+  val code =
+    s"""package com.github.tkqubo.typed_annotation
+       |
+       |import scala.annotation.StaticAnnotation
+       |
+       |trait GrandParent
+       |
+       |trait Parent extends GrandParent
+       |
+       |trait Child extends Parent
+       |
+       |@TypedAnnotation(classOf[Parent])
+       |case class parentNamed(value: String) extends StaticAnnotation
+       |
+       |@TypedAnnotation(classOf[Child])
+       |case class childNamed(value: String) extends StaticAnnotation
+       |
+       |object TypedAnnotationPluginTest {
+       |  @parentNamed("grand parent") // compilation error
+       |  @childNamed("grand parent") // compilation error
+       |  var grandParent: GrandParent = null
+       |
+       |  @parentNamed("parent")
+       |  @childNamed("parent") // compilation error
+       |  var parent: Parent = null
+       |
+       |  @parentNamed("child")
+       |  @childNamed("child")
+       |  var child: Child = null
+       |}
+     """.stripMargin
+  val sources = List(new BatchSourceFile("<test>", code))
+  val settings = new Settings
+  val loader = getClass.getClassLoader.asInstanceOf[URLClassLoader]
+  val entries = loader.getURLs.map(_.getPath)
+  val sclpath = entries
+    .find(_.endsWith("scala-compiler.jar"))
+    .map(_.replaceAll("scala-compiler.jar", "scala-library.jar"))
+  settings.classpath.value = ClassPath.join(entries ++ sclpath : _*)
+
+
+  "TypedAnnotationPlugin" should {
+    "pass with complex class hierarchy" in {
+      val reporter: StoreReporter = new StoreReporter
+      def equalTo(start: Int, end: Int, expectedType: String, actualType: String): Matcher[reporter.Info] = { (info: reporter.Info) =>
+        info.pos.start === start
+        info.pos.end === end
+        info.msg === s"Not annotated correctly.  This definition is annotated as $expectedType but actually $actualType"
+      }
+
+      val compiler = new Global(settings, reporter) {
+        override protected def computeInternalPhases () {
+          super.computeInternalPhases
+          for (phase <- new TypedAnnotationPlugin(this).components)
+            phasesSet += phase
+        }
+      }
+      new compiler.Run() compileSources sources
+
+      reporter.infos.size === 3
+      val infos: Seq[reporter.Info] = reporter.infos.toSeq.sortBy(_.pos.start)
+      val grandParentDefinition: String = "var grandParent: GrandParent"
+      infos(0) must equalTo(
+        code.indexOf(grandParentDefinition),
+        code.indexOf(grandParentDefinition) + 4,
+        "com.github.tkqubo.typed_annotation.Parent",
+        "com.github.tkqubo.typed_annotation.GrandParent"
+      )
+      infos(1) must equalTo(
+        code.indexOf(grandParentDefinition),
+        code.indexOf(grandParentDefinition) + 4,
+        "com.github.tkqubo.typed_annotation.Child",
+        "com.github.tkqubo.typed_annotation.GrandParent"
+      )
+      val parentDefinition: String = "var parent: Parent"
+      infos(2) must equalTo(
+        code.indexOf(parentDefinition),
+        code.indexOf(parentDefinition) + 4,
+        "com.github.tkqubo.typed_annotation.Child",
+        "com.github.tkqubo.typed_annotation.Parent"
+      )
+    }
+  }
+
+}