Allow optional rewrite rule in testScala2Mode

odersky · odersky · commit 244257bd0e9b · 2017-06-23T11:07:05.000+02:00
diff --git a/compiler/src/dotty/tools/dotc/core/TypeOps.scala b/compiler/src/dotty/tools/dotc/core/TypeOps.scala
@@ -518,8 +518,11 @@ trait TypeOps { this: Context => // TODO: Make standalone object.
   def dynamicsEnabled =
     featureEnabled(defn.LanguageModuleClass, nme.dynamics)
 
-  def testScala2Mode(msg: => String, pos: Position) = {
-    if (scala2Mode) migrationWarning(msg, pos)
+  def testScala2Mode(msg: => String, pos: Position, rewrite: => Unit = ()) = {
+    if (scala2Mode) {
+      migrationWarning(msg, pos)
+      rewrite
+    }
     scala2Mode
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -518,8 +518,11 @@ trait TypeOps { this: Context => // TODO: Make standalone object.`
`518`	`518`	`def dynamicsEnabled =`
`519`	`519`	`featureEnabled(defn.LanguageModuleClass, nme.dynamics)`
`520`	`520`
`521`		`- def testScala2Mode(msg: => String, pos: Position) = {`
`522`		`- if (scala2Mode) migrationWarning(msg, pos)`
	`521`	`+ def testScala2Mode(msg: => String, pos: Position, rewrite: => Unit = ()) = {`
	`522`	`+ if (scala2Mode) {`
	`523`	`+ migrationWarning(msg, pos)`
	`524`	`+ rewrite`
	`525`	`+ }`
`523`	`526`	`scala2Mode`
`524`	`527`	`}`
`525`	`528`	`}`