Fix #9751: Widen references to catch inline parameters with unit type

nicolasstucki · nicolasstucki · commit 5f8f19e4ff7f · 2020-09-08T18:17:24.000+02:00
diff --git a/compiler/src/dotty/tools/dotc/typer/Typer.scala b/compiler/src/dotty/tools/dotc/typer/Typer.scala
@@ -3629,8 +3629,12 @@ class Typer extends Namer
     }
 
   private def checkStatementPurity(tree: tpd.Tree)(original: untpd.Tree, exprOwner: Symbol)(using Context): Unit =
-    if (!tree.tpe.isErroneous && !ctx.isAfterTyper && isPureExpr(tree) &&
-        !tree.tpe.isRef(defn.UnitClass) && !isSelfOrSuperConstrCall(tree))
+    if !tree.tpe.isErroneous
+      && !ctx.isAfterTyper
+      && isPureExpr(tree)
+      && !tree.tpe.widenTermRefExpr.isRef(defn.UnitClass)
+      && !isSelfOrSuperConstrCall(tree)
+    then
       report.warning(PureExpressionInStatementPosition(original, exprOwner), original.srcPos)
 
   /** Types the body Scala 2 macro declaration `def f = macro <body>` */
diff --git a/tests/pos-special/fatal-warnings/i9751.scala b/tests/pos-special/fatal-warnings/i9751.scala
@@ -0,0 +1,11 @@
+object Test {
+  extension (x: Int)
+    inline def times(inline op: Unit): Unit = {
+      var count = 0
+      while count < x do
+        op
+        count += 1
+    }
+
+  10.times { println("hello") }
+}
