ImplementMethods refactor: Minimum test & prepare refactor stage

Author: pfcoperez

src/main/scala/scala/tools/refactoring/implementations/ImplementMethods.scala

@@ -0,0 +1,32 @@
+package scala.tools.refactoring.implementations
+
+import scala.tools.refactoring.common.InteractiveScalaCompiler
+import scala.tools.refactoring.{MultiStageRefactoring, analysis}
+
+abstract class ImplementMethods extends MultiStageRefactoring with analysis.Indexes with InteractiveScalaCompiler {
+
+  import global._
+
+  override type PreparationResult = Seq[DefDef]
+
+  override def prepare(s: Selection): Either[PreparationError, PreparationResult] = {
+    val methodsToImplement = for {
+      selectedClassDeclaration <- index.declaration(s.enclosingTree.symbol).toSeq collect {
+        case traitDeclaration: ClassDef => traitDeclaration
+      }
+      unimplementedMethod <- selectedClassDeclaration.impl.body collect {
+        case methodDeclaration: DefDef if methodDeclaration.rhs.isEmpty => methodDeclaration
+      }
+    } yield unimplementedMethod
+    if(methodsToImplement.isEmpty) Left {
+      PreparationError("There are not methods to implement")
+    }
+    else Right(methodsToImplement)
+  }
+
+  override type RefactoringParameters = Unit
+
+  override def perform(selection: Selection, prepared: PreparationResult, params: RefactoringParameters) = {
+    ???
+  }
+}

src/test/scala/scala/tools/refactoring/tests/implementations/ImplementMethodsTest.scala

@@ -0,0 +1,56 @@
+package scala.tools.refactoring
+package tests.implementations
+
+import scala.tools.refactoring.implementations.ImplementMethods
+import scala.tools.refactoring.tests.util.{TestHelper, TestRefactoring}
+
+class ImplementMethodsTest extends TestHelper with TestRefactoring {
+  outer =>
+
+  val fromSource =
+    """
+      |trait T {
+      | def f(x: Int): String
+      |}
+      |
+      |object Obj extends /*(*/T/*)*/ {
+      |}
+    """.stripMargin
+
+  val toSource =
+    """
+      |trait T {
+      | def f(x: Int): String
+      |}
+      |
+      |object Obj extends T {
+      |  def g(): Int = {
+      |    ???    Left(PreparationError("crash!"))
+
+      |  }
+      |}
+    """.stripMargin
+
+  /*@Test
+  def addMethod() = {
+    global.ask { () =>
+      val refactoring = new AddMethod {
+        val global: Global = outer.global
+        val file = addToCompiler(UniqueNames.basename(), fromSource)
+        val change = addMethod(file, "Obj", "g", List(Nil), Nil, Some("Int"), AddToObject)
+      }
+      assertEquals(toSource, Change.applyChanges(refactoring.change, fromSource))
+    }
+  }*/
+
+  def implementMethods(pro: FileSet) = new TestRefactoringImpl(pro) {
+    override val refactoring = new ImplementMethods with TestProjectIndex
+    val changes = performRefactoring(())
+  }.changes
+
+  @Test
+  def doRefactor() = new FileSet() {
+    fromSource becomes toSource
+  } applyRefactoring implementMethods
+
+}