WIP

Author: nicolasstucki

tests/neg/tasty-macro-assert/quoted_1.scala renamed to tests/neg/tasty-macro-assert-1/quoted_1.scala

@@ -33,7 +33,7 @@ object Asserts {
     }
 
     tree match {
-      case Term.Inlined(_, Nil, Term.Apply(Term.Select(OpsTree(left), op), right :: Nil)) =>
+      case Term.Apply(Term.Select(OpsTree(left), op), right :: Nil) =>
         '(assertTrue(~left.seal[Boolean])) // Buggy code. To generate the errors
       case _ =>
         '(assertTrue(~cond))

tests/neg/tasty-macro-assert-1/quoted_2.scala

@@ -0,0 +1,10 @@
+
+import Asserts._
+
+object Test {
+  def main(args: Array[String]): Unit = {
+    macroAssert(true === "cde")
+    macroAssert("acb" === "cde") // error
+  }
+
+}

tests/neg/tasty-macro-assert-2/quoted_1.scala

@@ -0,0 +1,69 @@
+import scala.quoted._
+
+import scala.tasty._
+
+object Asserts {
+
+  implicit class Ops[T](left: T) {
+    def ===(right: T): Boolean = left == right
+    def !==(right: T): Boolean = left != right
+  }
+
+  object Ops
+
+  inline def macroAssert(cond: => Boolean): Unit =
+    ~impl('(cond))
+
+  def impl(cond: Expr[Boolean])(implicit reflect: Reflection): Expr[Unit] = {
+    import reflect._
+
+    val tree = cond.unseal
+
+    def isOps(tpe: TypeOrBounds): Boolean = tpe match {
+      case Type.SymRef(IsDefSymbol(sym), _) => sym.name == "Ops" // TODO check that the parent is Asserts
+      case _ => false
+    }
+
+    object OpsTree {
+      def unapply(arg: Term): Option[Term] = arg match {
+        case Term.Apply(Term.TypeApply(term, _), left :: Nil) if isOps(term.tpe) =>
+          Some(left)
+        case _ => None
+      }
+    }
+
+    tree match {
+      case Term.Apply(Term.Select(OpsTree(left), op), right :: Nil) =>
+        '(assertTrue(~left.seal[Boolean])) // Buggy code. To generate the errors
+      case _ =>
+        '(assertTrue(~cond))
+    }
+
+  }
+
+  def assertEquals[T](left: T, right: T): Unit = {
+    if (left != right) {
+      println(
+        s"""Error left did not equal right:
+           |  left  = $left
+           |  right = $right""".stripMargin)
+    }
+
+  }
+
+  def assertNotEquals[T](left: T, right: T): Unit = {
+    if (left == right) {
+      println(
+        s"""Error left was equal to right:
+           |  left  = $left
+           |  right = $right""".stripMargin)
+    }
+
+  }
+
+  def assertTrue(cond: Boolean): Unit = {
+    if (!cond)
+      println("Condition was false")
+  }
+
+}

tests/neg/tasty-macro-assert/quoted_2.scala renamed to tests/neg/tasty-macro-assert-2/quoted_2.scala

@@ -3,8 +3,6 @@ import Asserts._
 
 object Test {
   def main(args: Array[String]): Unit = {
-    macroAssert(true === "cde")
-    macroAssert("acb" === "cde") // error
     macroAssert(false !== "acb")
     macroAssert("acb" !== "acb") // error
   }

tests/run/tasty-macro-assert/quoted_1.scala

@@ -33,7 +33,7 @@ object Asserts {
     }
 
     tree match {
-      case Term.Inlined(_, Nil, Term.Apply(Term.Select(OpsTree(left), op), right :: Nil)) =>
+      case Term.Apply(Term.Select(OpsTree(left), op), right :: Nil) =>
         op match {
           case "===" => '(assertEquals(~left.seal[Any], ~right.seal[Any]))
           case "!==" => '(assertNotEquals(~left.seal[Any], ~right.seal[Any]))