MockitoPureEff.scala

package fix

import scalafix.Patch
import scalafix.v1.SyntacticDocument
import scalafix.v1.SyntacticRule
import scala.meta.Term
import scala.meta.Type

class MockitoPureEff extends SyntacticRule("MockitoPureEff") {
  override def fix(implicit doc: SyntacticDocument): Patch = {
    doc.tree.collect {
      case Term.Apply(
            Term.Select(
              Term.Apply(
                Term.Name("when"),
                List(
                  Term.Apply(
                    Term.ApplyType(
                      Term.Select(
                        _,
                        _
                      ),
                      List(
                        Type.Name(tpe)
                      )
                    ),
                    _
                  )
                )
              ),
              Term.Name("thenReturn")
            ),
            List(
              ret @ Term.Select(
                _,
                Term.Name("pureEff")
              )
            )
          ) =>
        Patch.addRight(ret, s"[${tpe}]")
    }.asPatch
  }
}