SignificantIndentationSuite.scala

package scala.meta.tests.parsers.dotty

import scala.meta._

class SignificantIndentationSuite extends BaseDottySuite {

  val defx = Decl.Def(Nil, tname("f"), Nil, Nil, pname("Int"))
  val defy = Defn.Def(
    Nil,
    tname("y"),
    Nil,
    Nil,
    Some(pname("String")),
    Term.Block(List(Term.Apply(tname("fa"), Nil), Term.Apply(tname("fb"), Nil)))
  )

  test("basic-example") {
    val code = """|trait A:
                  |  def f: Int
                  |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some("trait A { def f: Int }"))(
      Defn.Trait(
        Nil,
        pname("A"),
        Nil,
        EmptyCtor(),
        tpl(defx)
      )
    )
  }

  test("package-comment") {
    val code = """|package mysadpackage:
                  |  def f: Int
                  |package anotherpackage:
                  |  def f: Int
                  |""".stripMargin
    runTestAssert[Source](
      code,
      assertLayout = Some(
        """|package mysadpackage {
           |  def f: Int
           |}
           |package anotherpackage {
           |  def f: Int
           |}
           |""".stripMargin
      )
    )(
      Source(
        List(
          Pkg(
            tname("mysadpackage"),
            List(Decl.Def(Nil, tname("f"), Nil, Nil, pname("Int")))
          ),
          Pkg(
            tname("anotherpackage"),
            List(Decl.Def(Nil, tname("f"), Nil, Nil, pname("Int")))
          )
        )
      )
    )
  }

  test("multistat-example") {
    val code = """|trait A:
                  |  def f: Int
                  |  def y: String = { fa(); fb() }""".stripMargin
    runTestAssert[Stat](code, assertLayout = None)(
      Defn.Trait(
        Nil,
        pname("A"),
        Nil,
        EmptyCtor(),
        tpl(defx, defy)
      )
    )
  }

  test("anonymous-class") {
    val code = """|new A:
                  |  def f: Int
                  |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some("new A { def f: Int }"))(
      Term.NewAnonymous(
        tpl(
          List(Init(pname("A"), anon, emptyArgClause)),
          List(Decl.Def(Nil, tname("f"), Nil, Nil, pname("Int")))
        )
      )
    )
  }

  // https://github.com/scalameta/scalameta/issues/3093
  test("anonymous-class-in-block") {
    val code = """|if(x) {
                  |  new A:
                  |    def f: Int
                  |}
                  |""".stripMargin
    runTestAssert[Stat](code, assertLayout = None)(
      Term.If(
        tname("x"),
        Term.Block(
          Term.NewAnonymous(
            tpl(
              List(Init(pname("A"), anon, emptyArgClause)),
              List(Decl.Def(Nil, tname("f"), Nil, pname("Int")))
            )
          ) :: Nil
        ),
        Lit.Unit(),
        Nil
      )
    )
  }

  test("empty-anonymous-class") {
    val code = """|new:
                  |  def f: Int
                  |  
                  |  def g: Int
                  |""".stripMargin
    runTestAssert[Stat](
      code,
      assertLayout = Some(
        """|new {
           |  def f: Int
           |  def g: Int
           |}
           |""".stripMargin
      )
    )(
      Term.NewAnonymous(
        tpl(
          Decl.Def(Nil, tname("f"), Nil, Nil, pname("Int")),
          Decl.Def(Nil, tname("g"), Nil, Nil, pname("Int"))
        )
      )
    )
  }

  test("indent-and-back") {
    val code = """|object O:
                  |  class C:
                  |    def f: Int = 3
                  |  trait T:
                  |    def f: Int = 4
                  |""".stripMargin
    runTestAssert[Stat](code, assertLayout = None)(
      Defn.Object(
        Nil,
        tname("O"),
        tpl(
          Defn.Class(
            Nil,
            pname("C"),
            Nil,
            ctor,
            Template(
              Nil,
              Nil,
              Self(anon, None),
              List(Defn.Def(Nil, tname("f"), Nil, Nil, Some(pname("Int")), int(3)))
            )
          ),
          Defn.Trait(
            Nil,
            pname("T"),
            Nil,
            EmptyCtor(),
            Template(
              Nil,
              Nil,
              Self(anon, None),
              List(Defn.Def(Nil, tname("f"), Nil, Nil, Some(pname("Int")), int(4)))
            )
          )
        )
      )
    )
  }

  test("then-no-indent") {
    // this test is related to dotty issue: https://github.com/lampepfl/dotty/issues/9790
    // It should either assert error during parsing: "illegal start of simple expression"
    // Or accept mismatch with parsing rules and parse as 'if (cond) { truep } else {falsep }'
    // Why error is thrown is described in mentioned issue.
    val code = """|def fn: Unit =
                  |    if cond then
                  |  truep
                  |    else
                  |  falsep
                  |    otherStatement()
                  |""".stripMargin
    runTestAssert[Stat](
      code,
      assertLayout = Some(
        """|def fn: Unit = {
           |  if (cond) truep else falsep
           |  otherStatement()
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("fn"),
        Nil,
        Nil,
        Some(pname("Unit")),
        Term.Block(
          List(
            Term.If(tname("cond"), tname("truep"), tname("falsep")),
            Term.Apply(tname("otherStatement"), Nil)
          )
        )
      )
    )
  }

  test("then-no-indent-wrong") {
    // this test is related to dotty issue: https://github.com/lampepfl/dotty/issues/9790
    // It should either assert error during parsing: "illegal start of simple expression"
    // Or accept mismatch with parsing rules and parse as 'if (cond) { truep } else {falsep }'
    // Why error is thrown is described in mentioned issue.
    val code = """|def fn: Unit =
                  |    if cond then
                  |  truep1
                  |  truep2
                  |    else
                  |  falsep
                  |""".stripMargin
    runTestError[Stat](code, "expected but else found")
  }

  test("indent-inside-brace-ok") {
    val code = """|class X {
                  |  def fx(): Unit =
                  |    f1
                  |    f2
                  |}
                  |""".stripMargin
    val output = """|class X {
                    |  def fx(): Unit = {
                    |    f1
                    |    f2
                    |  }
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("X"),
        Nil,
        EmptyCtor(),
        tpl(
          Defn.Def(
            Nil,
            tname("fx"),
            Nil,
            List(List()),
            Some(pname("Unit")),
            Term.Block(List(tname("f1"), tname("f2")))
          )
        )
      )
    )
  }

  test("indent-inside-brace-not") {
    val code = """|class X {
                  |      def fx(): Unit =
                  |    f1
                  |    f2
                  |}
                  |""".stripMargin
    val output = """|class X {
                    |  def fx(): Unit = f1
                    |  f2
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("X"),
        Nil,
        EmptyCtor(),
        tpl(
          Defn.Def(
            Nil,
            tname("fx"),
            Nil,
            List(List()),
            Some(pname("Unit")),
            tname("f1")
          ),
          tname("f2")
        )
      )
    )
  }

  test("should-indent-yet-brace") {
    val code = """|class X {
                  |  def fx(): Unit =
                  |  {
                  |
                  |  }
                  |  private def f2: Int = 1
                  |}
                  |""".stripMargin
    val output = """|class X {
                    |  def fx(): Unit = {}
                    |  private def f2: Int = 1
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("X"),
        Nil,
        EmptyCtor(),
        tpl(
          Defn.Def(
            Nil,
            tname("fx"),
            Nil,
            List(List()),
            Some(pname("Unit")),
            Term.Block(Nil)
          ),
          Defn.Def(
            List(Mod.Private(anon)),
            tname("f2"),
            Nil,
            Nil,
            Some(pname("Int")),
            int(1)
          )
        )
      )
    )
  }

  test("selftype-class") {
    val code = """|class A extends B:
                  |  thisPhase =>
                  |  expr1
                  |  expr2
                  |""".stripMargin
    val output = """|class A extends B { thisPhase =>
                    |  expr1
                    |  expr2
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("A"),
        Nil,
        EmptyCtor(),
        Template(
          Nil,
          List(Init(pname("B"), anon, emptyArgClause)),
          self("thisPhase"),
          List(tname("expr1"), tname("expr2"))
        )
      )
    )
  }

  test("lambda-method") {
    val code = """|object X:
                  |  val fn = (pa, pb) =>
                  |    def helper = 3
                  |    3
                  |  end fn
                  |""".stripMargin
    val output = """|object X {
                    |  val fn = (pa, pb) => {
                    |    def helper = 3
                    |    3
                    |  }
                    |  end fn
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Object(
        Nil,
        tname("X"),
        tpl(
          Defn.Val(
            Nil,
            List(Pat.Var(tname("fn"))),
            None,
            Term.Function(
              List(
                tparam("pa"),
                tparam("pb")
              ),
              Term.Block(
                List(Defn.Def(Nil, tname("helper"), Nil, Nil, None, int(3)), int(3))
              )
            )
          ),
          Term.EndMarker(tname("fn"))
        )
      )
    )
  }

  test("case-for-in-match") {
    val code = """|class A {
                  |    def forward: Unit = parents match
                  |      case a =>
                  |        for case a: TP <- body do
                  |          fordo
                  |      case b => ok
                  |
                  |    private def transformAnnot: Tree
                  |}
                  |""".stripMargin
    val output = """|class A {
                    |  def forward: Unit = parents match {
                    |    case a =>
                    |      for ( case a: TP <- body) fordo
                    |    case b =>
                    |      ok
                    |  }
                    |  private def transformAnnot: Tree
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("A"),
        Nil,
        ctor,
        tpl(
          Defn.Def(
            Nil,
            tname("forward"),
            Nil,
            Nil,
            Some(pname("Unit")),
            Term.Match(
              tname("parents"),
              List(
                Case(
                  Pat.Var(tname("a")),
                  None,
                  Term.For(
                    List(
                      Enumerator.CaseGenerator(
                        Pat.Typed(Pat.Var(tname("a")), pname("TP")),
                        tname("body")
                      )
                    ),
                    tname("fordo")
                  )
                ),
                Case(Pat.Var(tname("b")), None, tname("ok"))
              )
            )
          ),
          Decl.Def(
            List(Mod.Private(anon)),
            tname("transformAnnot"),
            Nil,
            Nil,
            pname("Tree")
          )
        )
      )
    )
  }

  test("this-constructor") {
    val code = """|class A:
                  |  def this(msg: String) =
                  |    this(message, false)
                  |""".stripMargin
    val output = "class A { def this(msg: String) = this(message, false) }"
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("A"),
        Nil,
        EmptyCtor(),
        tpl(
          Ctor.Secondary(
            Nil,
            Name.This(),
            List(List(tparam("msg", "String"))),
            Init(
              Type.Singleton(Term.This(anon)),
              anon,
              List(List(tname("message"), bool(false)))
            ),
            Nil
          )
        )
      )
    )
  }

  test("this-constructor-indented-block") {
    val code = """|class A:
                  |  def this(msg: String) =
                  |    this(message, false)
                  |    otherStat
                  |""".stripMargin
    val output = """|class A {
                    |  def this(msg: String) {
                    |    this(message, false)
                    |    otherStat
                    |  }
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Class(
        Nil,
        pname("A"),
        Nil,
        EmptyCtor(),
        tpl(
          Ctor.Secondary(
            Nil,
            Name.This(),
            List(List(tparam("msg", "String"))),
            Init(
              Type.Singleton(Term.This(anon)),
              anon,
              List(List(tname("message"), bool(false)))
            ),
            List(tname("otherStat"))
          )
        )
      )
    )
  }

  test("indent-equals") {
    runTestAssert[Stat](
      """|def genApply() = {
         |      app match {
         |        case Apply2() =>
         |          generatedType =
         |            genTypeApply(t)
         |
         |        case Apply() =>
         |      }
         |} 
         |""".stripMargin,
      assertLayout = Some(
        """|def genApply() = {
           |  app match {
           |    case Apply2() =>
           |      generatedType = genTypeApply(t)
           |    case Apply() =>
           |  }
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("genApply"),
        Nil,
        List(List()),
        None,
        Term.Block(
          List(
            Term.Match(
              tname("app"),
              List(
                Case(
                  Pat.Extract(tname("Apply2"), Nil),
                  None,
                  Term.Assign(
                    tname("generatedType"),
                    Term.Apply(tname("genTypeApply"), List(tname("t")))
                  )
                ),
                Case(Pat.Extract(tname("Apply"), Nil), None, Term.Block(Nil))
              ),
              Nil
            )
          )
        )
      )
    )
  }

  test("outdent-with-prev-check") {
    runTestAssert[Stat](
      """|def wrapPlaceholders(t: Tree) = try
         |    if (placeholderParams.isEmpty) t
         |    else new WildcardFunction(placeholderParams.reverse, t)
         |  finally placeholderParams = saved
         |""".stripMargin,
      assertLayout = Some(
        "def wrapPlaceholders(t: Tree) = try if (placeholderParams.isEmpty) t else new WildcardFunction(placeholderParams.reverse, t) finally placeholderParams = saved"
      )
    )(
      Defn.Def(
        Nil,
        tname("wrapPlaceholders"),
        Nil,
        List(List(tparam("t", "Tree"))),
        None,
        Term.Try(
          Term.If(
            Term.Select(tname("placeholderParams"), tname("isEmpty")),
            tname("t"),
            Term.New(
              Init(
                pname("WildcardFunction"),
                anon,
                List(
                  List(
                    Term.Select(tname("placeholderParams"), tname("reverse")),
                    tname("t")
                  )
                )
              )
            )
          ),
          Nil,
          Some(Term.Assign(tname("placeholderParams"), tname("saved")))
        )
      )
    )
  }

  test("type-in-next-line") {
    runTestAssert[Stat](
      """|abstract class Documentation{
         |
         |  class Graph {
         |    type Node = Int
         |    val a:
         |      Int = 3
         |  }
         |
         |  val refinementTest:
         |    Graph {
         |      def x: Int
         |    }
         |
         |}
         |""".stripMargin,
      assertLayout = Some(
        """|abstract class Documentation {
           |  class Graph {
           |    type Node = Int
           |    val a: Int = 3
           |  }
           |  val refinementTest: Graph { def x: Int }
           |}
           |""".stripMargin
      )
    )(
      Defn.Class(
        List(Mod.Abstract()),
        pname("Documentation"),
        Nil,
        EmptyCtor(),
        tpl(
          Defn.Class(
            Nil,
            pname("Graph"),
            Nil,
            EmptyCtor(),
            tpl(
              Defn.Type(Nil, pname("Node"), Nil, pname("Int")),
              Defn.Val(Nil, List(Pat.Var(tname("a"))), Some(pname("Int")), int(3))
            )
          ),
          Decl.Val(
            Nil,
            List(Pat.Var(tname("refinementTest"))),
            Type.Refine(
              Some(pname("Graph")),
              List(Decl.Def(Nil, tname("x"), Nil, Nil, pname("Int")))
            )
          )
        )
      )
    )
  }

  test("type-in-next-line-equals") {
    runTestAssert[Stat](
      """|val refinementTest:
         |      Int = 3
         |""".stripMargin,
      assertLayout = Some("val refinementTest: Int = 3")
    )(
      Defn.Val(Nil, List(Pat.Var(tname("refinementTest"))), Some(pname("Int")), int(3))
    )
  }

  test("type-in-next-line-equals-newline") {
    runTestAssert[Stat](
      """|val refinementTest:
         |      Int = 
         |3
         |""".stripMargin,
      assertLayout = Some("val refinementTest: Int = 3")
    )(
      Defn.Val(Nil, List(Pat.Var(tname("refinementTest"))), Some(pname("Int")), int(3))
    )
  }

  test("type-equals-separate") {
    runTestAssert[Stat](
      """|def refinementTest(a :
         |      Int = 
         |3) = a
         |""".stripMargin,
      assertLayout = Some("def refinementTest(a: Int = 3) = a")
    )(
      Defn.Def(
        Nil,
        tname("refinementTest"),
        Nil,
        List(List(Term.Param(Nil, tname("a"), Some(pname("Int")), Some(int(3))))),
        None,
        tname("a")
      )
    )
  }

  test("type-multi-seq") {
    runTestAssert[Stat](
      """|val refinementTest:
         |    Int = 
         |  fx
         |  fy
         |""".stripMargin,
      assertLayout = Some("val refinementTest: Int = {\n  fx\n  fy\n}")
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("refinementTest"))),
        Some(pname("Int")),
        Term.Block(List(tname("fx"), tname("fy")))
      )
    )
  }

  test("equals-block") {
    runTestAssert[Stat](
      """|val refinementTest:
         |  Int = 
         |    fx
         |    fy
         |""".stripMargin,
      assertLayout = Some("val refinementTest: Int = {\n  fx\n  fy\n}")
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("refinementTest"))),
        Some(pname("Int")),
        Term.Block(List(tname("fx"), tname("fy")))
      )
    )
  }

  test("given-block-indent") {
    runTestAssert[Stat](
      """|given intOrd: Ord[Int] with
         |   def fa: Int = 1
         |   def fb: Int = 2
         |""".stripMargin,
      assertLayout = Some(
        """|given intOrd: Ord[Int] with {
           |  def fa: Int = 1
           |  def fb: Int = 2
           |}""".stripMargin
      )
    )(
      Defn.Given(
        Nil,
        tname("intOrd"),
        Nil,
        Nil,
        tpl(
          List(Init(Type.Apply(pname("Ord"), List(pname("Int"))), anon, emptyArgClause)),
          List(
            Defn.Def(Nil, tname("fa"), Nil, Nil, Some(pname("Int")), int(1)),
            Defn.Def(Nil, tname("fb"), Nil, Nil, Some(pname("Int")), int(2))
          )
        )
      )
    )
  }

  test("given-block-indent-edge-cases") {
    runTestError[Stat](
      """|given intOrd: Ord[Int] with
         |def fa: Int = 1
         |def fb: Int = 2
         |""".stripMargin,
      "expected '{' or indentation"
    )

    runTestError[Stat](
      """|given intOrd: Ord[Int]:
         |  def fa: Int = 1
         |  def fb: Int = 2
         |""".stripMargin,
      "; expected but : found"
    )

    runTestAssert[Stat](
      """|class A extends A with 
         |  B
         |""".stripMargin,
      assertLayout = Some("class A extends A with B")
    )(
      Defn.Class(
        Nil,
        pname("A"),
        Nil,
        EmptyCtor(),
        tpl(
          List(
            Init(pname("A"), anon, emptyArgClause),
            Init(pname("B"), anon, emptyArgClause)
          ),
          Nil
        )
      )
    )
  }

  test("nested-coloneol") {
    runTestAssert[Stat](
      """|case class Test(
         |  a: A = new A,
         |):
         |  def hello = 1
         |""".stripMargin,
      assertLayout = Some("case class Test(a: A = new A) { def hello = 1 }")
    )(
      Defn.Class(
        List(Mod.Case()),
        pname("Test"),
        Nil,
        Ctor.Primary(
          Nil,
          anon,
          List(
            List(
              Term.Param(
                Nil,
                tname("a"),
                Some(pname("A")),
                Some(Term.New(Init(pname("A"), anon, emptyArgClause)))
              )
            )
          )
        ),
        tpl(Defn.Def(Nil, tname("hello"), Nil, Nil, None, int(1)))
      )
    )
  }

  test("semicolon-closes-indent-region") {
    runTestAssert[Stat](
      """|val z =
         |  val a = 
         |    0;
         |  f(a)
         |""".stripMargin,
      assertLayout = Some(
        """|val z = {
           |  val a = 0
           |  f(a)
           |}""".stripMargin
      )
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("z"))),
        None,
        Term.Block(
          List(
            Defn.Val(Nil, List(Pat.Var(tname("a"))), None, int(0)),
            Term.Apply(tname("f"), List(tname("a")))
          )
        )
      )
    )
  }

  test("observe-indented-in-braces") {
    val code = """|object X:
                  |  if (cond)
                  |    (f)
                  |  foo
                  |""".stripMargin
    val output = """|object X {
                    |  if (cond) f
                    |  foo
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Object(
        Nil,
        tname("X"),
        tpl(
          Term.If(tname("cond"), tname("f"), Lit.Unit(), Nil),
          tname("foo")
        )
      )
    )
  }

  test("match-case-same-line") {
    runTestAssert[Stat](
      """|widen match
         |  case tp @ OrNull(tp1): OrType =>
         |  case tp => tp
         |""".stripMargin,
      assertLayout = Some(
        """|widen match {
           |  case (tp @ OrNull(tp1)): OrType =>
           |  case tp => tp
           |}
           |""".stripMargin
      )
    )(
      Term.Match(
        tname("widen"),
        List(
          Case(
            Pat.Typed(
              Pat.Bind(
                Pat.Var(tname("tp")),
                Pat.Extract(tname("OrNull"), List(Pat.Var(tname("tp1"))))
              ),
              pname("OrType")
            ),
            None,
            Term.Block(Nil)
          ),
          Case(Pat.Var(tname("tp")), None, tname("tp"))
        ),
        Nil
      )
    )
  }

  test("object-type") {
    runTestAssert[Stat](
      """|object typeAndObjects:
         |  type Ala
         |""".stripMargin,
      assertLayout = Some(
        "object typeAndObjects { type Ala }"
      )
    )(
      Defn.Object(
        Nil,
        tname("typeAndObjects"),
        tpl(
          Decl.Type(Nil, pname("Ala"), Nil, Type.Bounds(None, None))
        )
      )
    )
  }

  test("old-try-catch-same-indent") {
    val code = """|try 
                  |  fx
                  |  gx 
                  |  catch 
                  |  case aa =>
                  |  case bb =>
                  |  finally
                  |  cc
                  |  dd
                  |""".stripMargin
    val output = """|try {
                    |  fx
                    |  gx
                    |} catch {
                    |  case aa =>
                    |  case bb =>
                    |} finally {
                    |  cc
                    |  dd
                    |}
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Term.Try(
        Term.Block(List(tname("fx"), tname("gx"))),
        List(
          Case(Pat.Var(tname("aa")), None, Term.Block(Nil)),
          Case(Pat.Var(tname("bb")), None, Term.Block(Nil))
        ),
        Some(Term.Block(List(tname("cc"), tname("dd"))))
      )
    )
  }

  test("if-else-same-indent") {
    val code = """|if 
                  |  cond
                  |  cond2
                  |  then 
                  |  fx
                  |  gx
                  |  else gx
                  |""".stripMargin
    val output = """|if ({
                    |  cond
                    |  cond2
                    |}) {
                    |  fx
                    |  gx
                    |} else gx
                    |""".stripMargin
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Term.If(
        Term.Block(List(tname("cond"), tname("cond2"))),
        Term.Block(List(tname("fx"), tname("gx"))),
        tname("gx"),
        Nil
      )
    )
  }

  test("new-fordo-same-indent") {
    val code = """|for
                  |  a <- gen
                  |  do fx
                  |""".stripMargin
    val output = "for (a <- gen) fx"
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Term.For(
        List(Enumerator.Generator(Pat.Var(tname("a")), tname("gen"))),
        tname("fx")
      )
    )
  }

  test("new-for-yield-same-indent") {
    val code = """|for
                  |  a <- x
                  |  b <- y
                  |  yield fx
                  |""".stripMargin
    val output = "for (a <- x; b <- y) yield fx"
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Term.ForYield(
        List(
          Enumerator.Generator(Pat.Var(tname("a")), tname("x")),
          Enumerator.Generator(Pat.Var(tname("b")), tname("y"))
        ),
        tname("fx")
      )
    )
  }

  test("new-for-yield-if-with-indent") {
    val code =
      """|for
         |  a <- x
         |  b <- y
         |  if
         |    a < b
         |  yield fx
         |""".stripMargin
    val output = "for (a <- x; b <- y; if a < b) yield fx"
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Term.ForYield(
        List(
          Enumerator.Generator(Pat.Var(tname("a")), tname("x")),
          Enumerator.Generator(Pat.Var(tname("b")), tname("y")),
          Enumerator.Guard(Term.ApplyInfix(tname("a"), tname("<"), Nil, List(tname("b"))))
        ),
        tname("fx")
      )
    )
  }

  test("old-for-yield-if-with-indent") {
    val code =
      """|for {
         |  a <- x
         |  b <- y
         |  if
         |    a < b
         |} yield fx
         |""".stripMargin
    val output = "for (a <- x; b <- y; if a < b) yield fx"
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Term.ForYield(
        List(
          Enumerator.Generator(Pat.Var(tname("a")), tname("x")),
          Enumerator.Generator(Pat.Var(tname("b")), tname("y")),
          Enumerator.Guard(Term.ApplyInfix(tname("a"), tname("<"), Nil, List(tname("b"))))
        ),
        tname("fx")
      )
    )
  }

  test("for-in-parens") {
    val code = """|def foo =
                  |  (for a <- List(1)
                  |       b <- List(2)
                  |   yield a + b).toSet.size
                  |""".stripMargin
    val output = "def foo = (for (a <- List(1); b <- List(2)) yield a + b).toSet.size"
    runTestAssert[Stat](code, assertLayout = Some(output))(
      Defn.Def(
        Nil,
        tname("foo"),
        Nil,
        Nil,
        None,
        Term.Select(
          Term.Select(
            Term.ForYield(
              List(
                Enumerator.Generator(
                  Pat.Var(tname("a")),
                  Term.Apply(tname("List"), List(int(1)))
                ),
                Enumerator.Generator(
                  Pat.Var(tname("b")),
                  Term.Apply(tname("List"), List(int(2)))
                )
              ),
              Term.ApplyInfix(tname("a"), tname("+"), Nil, List(tname("b")))
            ),
            tname("toSet")
          ),
          tname("size")
        )
      )
    )
  }

  test("match-chained-same-indent") {
    runTestAssert[Stat](
      """|val hello = xs match
         |  case Nil => "empty"
         |  case x :: xs1 => "nonempty"
         |  match
         |  case true => 0
         |  case false => 1
         |
         |""".stripMargin,
      Some(
        """|val hello = (xs match {
           |  case Nil => "empty"
           |  case x :: xs1 => "nonempty"
           |}) match {
           |  case true => 0
           |  case false => 1
           |}
           |""".stripMargin
      )
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("hello"))),
        None,
        Term.Match(
          Term.Match(
            tname("xs"),
            List(
              Case(tname("Nil"), None, str("empty")),
              Case(
                Pat.ExtractInfix(
                  Pat.Var(tname("x")),
                  tname("::"),
                  List(Pat.Var(tname("xs1")))
                ),
                None,
                str("nonempty")
              )
            ),
            Nil
          ),
          List(
            Case(bool(true), None, int(0)),
            Case(bool(false), None, int(1))
          ),
          Nil
        )
      )
    )

  }

  test("match-if-indent") {
    runTestAssert[Stat](
      """|def hackGetmembers = a match
         |   case sym if
         |     cond =>
         |     sym
         |
         |""".stripMargin,
      assertLayout = Some(
        """|def hackGetmembers = a match {
           |  case sym if cond => sym
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("hackGetmembers"),
        Nil,
        Nil,
        None,
        Term.Match(
          tname("a"),
          List(Case(Pat.Var(tname("sym")), Some(tname("cond")), tname("sym"))),
          Nil
        )
      )
    )

  }
  test("match-empty") {
    runTestAssert[Stat](
      """|def mapSymbols =
         |  originals.foreach { a =>
         |    copy.denot match
         |      case cd: ClassDenotation =>
         |      case _ =>
         |  }
         |""".stripMargin,
      assertLayout = Some(
        """|def mapSymbols = originals.foreach {
           |  a => copy.denot match {
           |    case cd: ClassDenotation =>
           |    case _ =>
           |  }
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("mapSymbols"),
        Nil,
        Nil,
        None,
        Term.Apply(
          Term.Select(tname("originals"), tname("foreach")),
          Term.Block(
            Term.Function(
              List(tparam("a")),
              Term.Match(
                Term.Select(tname("copy"), tname("denot")),
                List(
                  Case(
                    Pat.Typed(Pat.Var(tname("cd")), pname("ClassDenotation")),
                    None,
                    Term.Block(Nil)
                  ),
                  Case(Pat.Wildcard(), None, Term.Block(Nil))
                ),
                Nil
              )
            ) :: Nil
          ) :: Nil
        )
      )
    )

  }

  test("return-indent") {
    runTestAssert[Stat](
      """|def method =
         |   return
         |     val a = 2 + 3
         |     a
         |
         |""".stripMargin,
      assertLayout = Some(
        """|def method = return {
           |  val a = 2 + 3
           |  a
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("method"),
        Nil,
        Nil,
        None,
        Term.Return(
          Term.Block(
            List(
              Defn.Val(
                Nil,
                List(Pat.Var(tname("a"))),
                None,
                Term.ApplyInfix(int(2), tname("+"), Nil, List(int(3)))
              ),
              tname("a")
            )
          )
        )
      )
    )
  }

  test("return-single-indent") {
    runTestAssert[Stat](
      """|def method =
         |   return
         |     2 
         |     + 3
         |""".stripMargin,
      assertLayout = Some(
        "def method = return 2 + 3"
      )
    )(
      Defn.Def(
        Nil,
        tname("method"),
        Nil,
        Nil,
        None,
        Term.Return(
          Term.ApplyInfix(int(2), tname("+"), Nil, List(int(3)))
        )
      )
    )
  }

  test("empty-return") {
    runTestAssert[Stat](
      """|    def skip = {
         |        token match {
         |          case RBRACE =>
         |            if (true)
         |              return
         |            change(-1)
         |          case LBRACE =>
         |             if (true)
         |               return
         |             change(-1)
         |        }
         |    }
         |""".stripMargin,
      assertLayout = Some(
        """|def skip = {
           |  token match {
           |    case RBRACE =>
           |      if (true) return
           |      change(-1)
           |    case LBRACE =>
           |      if (true) return
           |      change(-1)
           |  }
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("skip"),
        Nil,
        Nil,
        None,
        Term.Block(
          List(
            Term.Match(
              tname("token"),
              List(
                Case(
                  tname("RBRACE"),
                  None,
                  Term.Block(
                    List(
                      Term.If(bool(true), Term.Return(Lit.Unit()), Lit.Unit(), Nil),
                      Term.Apply(tname("change"), List(int(-1)))
                    )
                  )
                ),
                Case(
                  tname("LBRACE"),
                  None,
                  Term.Block(
                    List(
                      Term.If(bool(true), Term.Return(Lit.Unit()), Lit.Unit(), Nil),
                      Term.Apply(tname("change"), List(int(-1)))
                    )
                  )
                )
              ),
              Nil
            )
          )
        )
      )
    )
  }

  test("case-block") {
    runTestAssert[Stat](
      """|val success = suffixes.find { suffix =>
         |  try {
         |    true
         |  } catch {
         |    case e: StorageException =>
         |      false
         |  }
         |}
         |""".stripMargin,
      assertLayout = Some(
        """|val success = suffixes.find {
           |  suffix => try {
           |    true
           |  } catch {
           |    case e: StorageException => false
           |  }
           |}
           |""".stripMargin
      )
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("success"))),
        None,
        Term.Apply(
          Term.Select(tname("suffixes"), tname("find")),
          Term.Block(
            Term.Function(
              List(tparam("suffix")),
              Term.Try(
                Term.Block(List(bool(true))),
                Case(
                  Pat.Typed(Pat.Var(tname("e")), pname("StorageException")),
                  None,
                  bool(false)
                ) :: Nil,
                None
              )
            ) :: Nil
          ) :: Nil
        )
      )
    )
  }

  test("complext-match-else") {
    runTestAssert[Stat](
      """|val calleeType = a match {
         |  case _ =>
         |    if cond then
         |      expr match
         |        case _ =>
         |          f
         |    else NoType
         |  case _ =>
         |     NoType
         |}
         |""".stripMargin,
      assertLayout = Some(
        """|val calleeType = a match {
           |  case _ =>
           |    if (cond) expr match {
           |      case _ => f
           |    } else NoType
           |  case _ =>
           |    NoType
           |}
           |""".stripMargin
      )
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("calleeType"))),
        None,
        Term.Match(
          tname("a"),
          List(
            Case(
              Pat.Wildcard(),
              None,
              Term.If(
                tname("cond"),
                Term
                  .Match(tname("expr"), List(Case(Pat.Wildcard(), None, tname("f"))), Nil),
                tname("NoType"),
                Nil
              )
            ),
            Case(Pat.Wildcard(), None, tname("NoType"))
          ),
          Nil
        )
      )
    )
  }

  test("partial-function") {
    runTestAssert[Stat](
      """|val withDefault: Option[Int] => Int =
         |  case Some(x) => x
         |  case None => 0
         |""".stripMargin,
      assertLayout = Some(
        """|val withDefault: Option[Int] => Int = {
           |  case Some(x) => x
           |  case None => 0
           |}
           |""".stripMargin
      )
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("withDefault"))),
        Some(
          Type.Function(
            List(Type.Apply(pname("Option"), List(pname("Int")))),
            pname("Int")
          )
        ),
        Term.PartialFunction(
          List(
            Case(
              Pat.Extract(tname("Some"), List(Pat.Var(tname("x")))),
              None,
              tname("x")
            ),
            Case(tname("None"), None, int(0))
          )
        )
      )
    )
  }

  test("for-left-arrow") {
    runTestAssert[Stat](
      """|for
         |  a <-
         |    val b = 123
         |    Some(b)
         |yield
         |  a
         |""".stripMargin,
      assertLayout = Some(
        """|for (a <- {
           |  val b = 123
           |  Some(b)
           |}) yield a
           |""".stripMargin
      )
    )(
      Term.ForYield(
        List(
          Enumerator.Generator(
            Pat.Var(tname("a")),
            Term.Block(
              List(
                Defn.Val(Nil, List(Pat.Var(tname("b"))), None, int(123)),
                Term.Apply(tname("Some"), List(tname("b")))
              )
            )
          )
        ),
        tname("a")
      )
    )
  }

  test("context-arrow") {
    runTestAssert[Stat](
      """|val a = (s: Int) ?=> 
         |  val a = 123
         |  s + a
         |
         |""".stripMargin,
      assertLayout = Some(
        """|val a = (s: Int) ?=> {
           |  val a = 123
           |  s + a
           |}
           |""".stripMargin
      )
    )(
      Defn.Val(
        Nil,
        List(Pat.Var(tname("a"))),
        None,
        Term.ContextFunction(
          List(tparam("s", "Int")),
          Term.Block(
            List(
              Defn.Val(Nil, List(Pat.Var(tname("a"))), None, int(123)),
              Term.ApplyInfix(tname("s"), tname("+"), Nil, List(tname("a")))
            )
          )
        )
      )
    )
  }

  test("indented-apply") {
    runTestAssert[Stat](
      """|def method = 
         |  fun(a,b,c)
         |    (d, e)
         |""".stripMargin,
      assertLayout = Some(
        "def method = fun(a, b, c)(d, e)"
      )
    )(
      Defn.Def(
        Nil,
        tname("method"),
        Nil,
        Nil,
        None,
        Term.Apply(
          Term.Apply(tname("fun"), List(tname("a"), tname("b"), tname("c"))),
          List(tname("d"), tname("e"))
        )
      )
    )
  }

  test("#3531 apply with optional braces and trailing comma") {
    val code =
      """|Request(
         |  value = true match
         |    case true => Some(1)
         |    case _ => None,
         |  b = "xxx"
         |)
         |""".stripMargin
    val layout =
      """|Request(value = true match {
         |  case true =>
         |    Some(1)
         |  case _ =>
         |    None
         |}, b = "xxx")
         |""".stripMargin
    val tree = Term.Apply(
      tname("Request"),
      List(
        Term.Assign(
          tname("value"),
          Term.Match(
            bool(true),
            List(
              Case(bool(true), None, Term.Apply(tname("Some"), List(int(1)))),
              Case(Pat.Wildcard(), None, tname("None"))
            ),
            Nil
          )
        ),
        Term.Assign(tname("b"), str("xxx"))
      )
    )
    runTestAssert[Stat](code, layout)(tree)
  }

  test("indented-double-apply") {
    runTestAssert[Stat](
      """|def method = 
         |  fun(a,b,c)
         |    (d, e)
         |    (f, g)
         |""".stripMargin,
      assertLayout = Some(
        "def method = fun(a, b, c)(d, e)(f, g)"
      )
    )(
      Defn.Def(
        Nil,
        tname("method"),
        Nil,
        Nil,
        None,
        Term.Apply(
          Term.Apply(
            Term.Apply(tname("fun"), List(tname("a"), tname("b"), tname("c"))),
            List(tname("d"), tname("e"))
          ),
          List(tname("f"), tname("g"))
        )
      )
    )
  }

  test("indented-for") {
    runTestAssert[Stat](
      """|for { project <- projects
         |      (source, id) <- project.sources.zipWithIndex } yield source 
         |""".stripMargin,
      assertLayout = Some(
        "for (project <- projects; (source, id) <- project.sources.zipWithIndex) yield source"
      )
    )(
      Term.ForYield(
        List(
          Enumerator.Generator(Pat.Var(tname("project")), tname("projects")),
          Enumerator.Generator(
            Pat.Tuple(List(Pat.Var(tname("source")), Pat.Var(tname("id")))),
            Term.Select(
              Term.Select(tname("project"), tname("sources")),
              tname("zipWithIndex")
            )
          )
        ),
        tname("source")
      )
    )
  }

  test("non-indented-apply") {
    runTestAssert[Stat](
      """|def method = 
         |  fun(a,b,c)
         |  (d, e)
         |""".stripMargin,
      assertLayout = Some(
        """|def method = {
           |  fun(a, b, c)
           |  (d, e)
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("method"),
        Nil,
        Nil,
        None,
        Term.Block(
          List(
            Term.Apply(tname("fun"), List(tname("a"), tname("b"), tname("c"))),
            Term.Tuple(List(tname("d"), tname("e")))
          )
        )
      )
    )
  }

  test("indented-apply-braces") {
    runTestAssert[Stat](
      """|def method: String = 
         |  fun(1, 2, 3) 
         |    {4} 
         |""".stripMargin,
      assertLayout = Some(
        """|def method: String = fun(1, 2, 3) {
           |  4
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("method"),
        Nil,
        Nil,
        Some(pname("String")),
        Term.Apply(
          Term.Apply(tname("fun"), List(int(1), int(2), int(3))),
          List(Term.Block(List(int(4))))
        )
      )
    )
  }

  test("non-indented-apply-braces") {
    runTestAssert[Stat](
      """|def method2: String = 
         |  fun(1, 2, 3)
         |  {4}
         |""".stripMargin,
      assertLayout = Some(
        """|def method2: String = {
           |  fun(1, 2, 3)
           |  {
           |    4
           |  }
           |}
           |""".stripMargin
      )
    )(
      Defn.Def(
        Nil,
        tname("method2"),
        Nil,
        Nil,
        Some(pname("String")),
        Term.Block(
          List(
            Term.Apply(tname("fun"), List(int(1), int(2), int(3))),
            Term.Block(List(int(4)))
          )
        )
      )
    )
  }

  test("indented-contructor-params") {
    runTestAssert[Stat](
      """|object ExampleThing
         |    extends CompositeThing
         |          (
         |                "One",
         |                "Two",
         |                "Three",
         |                "Four")
         |""".stripMargin,
      assertLayout = Some(
        """|object ExampleThing extends CompositeThing("One", "Two", "Three", "Four")
           |""".stripMargin
      )
    )(
      Defn.Object(
        Nil,
        tname("ExampleThing"),
        tpl(
          Init(
            pname("CompositeThing"),
            anon,
            List(
              List(str("One"), str("Two"), str("Three"), str("Four"))
            )
          ) :: Nil,
          Nil
        )
      )
    )
  }

  test("non-indented-contructor-params") {
    runTestAssert[Stat](
      """|object O:
         |  object ExampleThing
         |    extends CompositeThing
         |  (
         |                "One",
         |                "Two",
         |                "Three",
         |                "Four")
         |""".stripMargin,
      assertLayout = Some(
        """|object O {
           |  object ExampleThing extends CompositeThing
           |  ("One", "Two", "Three", "Four")
           |}
           |""".stripMargin
      )
    )(
      Defn.Object(
        Nil,
        tname("O"),
        tpl(
          Defn.Object(
            Nil,
            tname("ExampleThing"),
            Template(
              Nil,
              List(Init(pname("CompositeThing"), anon, emptyArgClause)),
              Self(anon, None),
              Nil,
              Nil
            )
          ),
          Term.Tuple(
            List(str("One"), str("Two"), str("Three"), str("Four"))
          )
        )
      )
    )
  }

  test("indented-enum-contructor-params") {
    runTestAssert[Stat](
      """|enum Namespace(val uri: String | Null):
         |  case xhtml
         |      extends Namespace // Defn.EnumCase ends here
         |        ("http://www.w3.org/1999/xhtml") // str
         |""".stripMargin,
      assertLayout = Some(
        """|enum Namespace(val uri: String | Null) { case xhtml extends Namespace("http://www.w3.org/1999/xhtml") }
           |""".stripMargin
      )
    )(
      Defn.Enum(
        Nil,
        pname("Namespace"),
        Nil,
        Ctor.Primary(
          Nil,
          anon,
          List(
            List(
              tparam(
                List(Mod.ValParam()),
                "uri",
                Type.ApplyInfix(pname("String"), pname("|"), pname("Null"))
              )
            )
          )
        ),
        tpl(
          Defn.EnumCase(
            Nil,
            tname("xhtml"),
            Nil,
            EmptyCtor(),
            List(
              Init(
                pname("Namespace"),
                anon,
                List(List(str("http://www.w3.org/1999/xhtml")))
              )
            )
          )
        )
      )
    )
  }

  test("indented-double-new") {
    runTestAssert[Stat](
      """|new fun
         |    (a,b,c)
         |    (d, e)
         |    (f, g)
         |""".stripMargin,
      assertLayout = Some(
        "new fun(a, b, c)(d, e)(f, g)"
      )
    )(
      Term.New(
        Init(
          pname("fun"),
          anon,
          List(
            List(tname("a"), tname("b"), tname("c")),
            List(tname("d"), tname("e")),
            List(tname("f"), tname("g"))
          )
        )
      )
    )
  }

  test("then-same-line") {
    runTestAssert[Stat](
      """|def f =
         |   if
         |      x.exists
         |         (x => x == 10) then
         |      println("Yes")
         |   else
         |      println("No")
         |""".stripMargin,
      assertLayout = Some(
        "def f = if (x.exists(x => x == 10)) println(\"Yes\") else println(\"No\")"
      )
    )(
      Defn.Def(
        Nil,
        tname("f"),
        Nil,
        Nil,
        None,
        Term.If(
          Term.Apply(
            Term.Select(tname("x"), tname("exists")),
            List(
              Term.Function(
                List(tparam("x")),
                Term.ApplyInfix(tname("x"), tname("=="), Nil, List(int(10)))
              )
            )
          ),
          Term.Apply(tname("println"), List(str("Yes"))),
          Term.Apply(tname("println"), List(str("No"))),
          Nil
        )
      )
    )
  }

  test("then-same-line-nested") {
    runTestAssert[Stat](
      """|def f =
         |   if
         |      if a > 0 then true else false
         |   then
         |      println("Yes")
         |   else
         |      println("No")
         |""".stripMargin,
      assertLayout = Some(
        "def f = if (if (a > 0) true else false) println(\"Yes\") else println(\"No\")"
      )
    )(
      Defn.Def(
        Nil,
        tname("f"),
        None,
        None,
        Term.If(
          Term.If(
            Term.ApplyInfix(tname("a"), tname(">"), Nil, List(int(0))),
            bool(true),
            bool(false),
            Nil
          ),
          Term.Apply(tname("println"), List(str("Yes"))),
          Term.Apply(tname("println"), List(str("No"))),
          Nil
        )
      )
    )
  }

  test("package-mixed") {
    runTestAssert[Source](
      """|package tests.site
         |
         |package some.other:
         |  class SomeOtherPackage
         |
         |class BrokenLink
         |""".stripMargin,
      assertLayout = Some(
        """|package tests.site
           |package some.other {
           |  class SomeOtherPackage
           |}
           |class BrokenLink
           |""".stripMargin
      )
    )(
      Source(
        List(
          Pkg(
            Term.Select(tname("tests"), tname("site")),
            List(
              Pkg(
                Term.Select(tname("some"), tname("other")),
                List(
                  Defn.Class(
                    Nil,
                    pname("SomeOtherPackage"),
                    Nil,
                    EmptyCtor(),
                    EmptyTemplate()
                  )
                )
              ),
              Defn.Class(
                Nil,
                pname("BrokenLink"),
                Nil,
                EmptyCtor(),
                EmptyTemplate()
              )
            )
          )
        )
      )
    )
  }

  test("empty-case-end-marker") {
    runTestAssert[Source](
      """|
         |  def abc: Unit =
         |    x match
         |      case _ =>
         |  end abc
         |""".stripMargin,
      assertLayout = None
    )(
      Source(
        List(
          Defn.Def(
            Nil,
            tname("abc"),
            Nil,
            Nil,
            Some(pname("Unit")),
            Term.Match(tname("x"), List(Case(Pat.Wildcard(), None, Term.Block(Nil))), Nil)
          ),
          Term.EndMarker(tname("abc"))
        )
      )
    )
  }

  test("infix-operator-with-alpha") {
    runTestAssert[Stat](
      """|def send() =
         |  c ! "hello"
         |    ! "world"
         |    send_! "!"
         |""".stripMargin,
      assertLayout = Some("""def send() = c ! "hello" ! "world" send_! "!"""")
    )(
      Defn.Def(
        Nil,
        tname("send"),
        Nil,
        List(List()),
        None,
        Term.ApplyInfix(
          Term.ApplyInfix(
            Term.ApplyInfix(tname("c"), tname("!"), Nil, List(str("hello"))),
            tname("!"),
            Nil,
            List(str("world"))
          ),
          tname("send_!"),
          Nil,
          List(str("!"))
        )
      )
    )
  }

  test("colon-eol-comment1") {
    runTestAssert[Stat](
      """|object Foo:
         |  /*inline*/ def foo: Int = ???
         |  def bar: Int = ???
         |""".stripMargin,
      assertLayout = None
    )(
      Defn.Object(
        Nil,
        tname("Foo"),
        tpl(
          Defn.Def(Nil, tname("foo"), Nil, Nil, Some(pname("Int")), tname("???")),
          Defn.Def(Nil, tname("bar"), Nil, Nil, Some(pname("Int")), tname("???"))
        )
      )
    )
  }

  test("colon-eol-comment2") {
    runTestAssert[Stat](
      """|object Foo: /* comment*/
         |  def foo: Int = ???
         |""".stripMargin,
      assertLayout = None
    )(
      Defn.Object(
        Nil,
        tname("Foo"),
        tpl(
          Defn.Def(Nil, tname("foo"), Nil, Nil, Some(pname("Int")), tname("???"))
        )
      )
    )
  }

  test("colon-eol-multiline-comment") {
    runTestAssert[Stat](
      """|object Foo:/* multi
         |  line
         |   comment */ def foo: Int = ???
         |   def bar: Int = ???
         |""".stripMargin,
      assertLayout = None
    )(
      Defn.Object(
        Nil,
        tname("Foo"),
        tpl(
          Defn.Def(Nil, tname("foo"), Nil, Nil, Some(pname("Int")), tname("???")),
          Defn.Def(Nil, tname("bar"), Nil, Nil, Some(pname("Int")), tname("???"))
        )
      )
    )
  }

  test("given-with-comment") {
    runTestAssert[Stat](
      """|given Foo with
         |   /* comment */  def foo: Int = ???
         |   def bar: Int = ???
         |""".stripMargin,
      assertLayout = None
    )(
      Defn.Given(
        Nil,
        anon,
        Nil,
        Nil,
        tpl(
          List(Init(pname("Foo"), anon, emptyArgClause)),
          List(
            Defn.Def(Nil, tname("foo"), Nil, Nil, Some(pname("Int")), tname("???")),
            Defn.Def(Nil, tname("bar"), Nil, Nil, Some(pname("Int")), tname("???"))
          )
        )
      )
    )
  }

  test("given-with-miltiline-comment") {
    runTestAssert[Stat](
      """|given Foo with /* multi
         |   line
         |   comment */  def foo: Int = ???
         |   def bar: Int = ???
         |""".stripMargin,
      assertLayout = None
    )(
      Defn.Given(
        Nil,
        anon,
        Nil,
        Nil,
        tpl(
          List(Init(pname("Foo"), anon, emptyArgClause)),
          List(
            Defn.Def(Nil, tname("foo"), Nil, Nil, Some(pname("Int")), tname("???")),
            Defn.Def(Nil, tname("bar"), Nil, Nil, Some(pname("Int")), tname("???"))
          )
        )
      )
    )
  }

  test("comment-bettween-annotation-set") {
    runTestAssert[Source](
      s"""|
          |class A1 extends scala.annotation.StaticAnnotation
          |class A2 extends scala.annotation.StaticAnnotation
          | @A1
          | /*
          | * hello
          | */
          | @A2
          |class B
          |""".stripMargin,
      assertLayout = None
    )(
      Source(
        List(
          Defn.Class(
            Nil,
            pname("A1"),
            Nil,
            EmptyCtor(),
            tpl(
              Init(
                Type.Select(
                  Term.Select(tname("scala"), tname("annotation")),
                  pname("StaticAnnotation")
                ),
                anon,
                emptyArgClause
              ) :: Nil,
              Nil
            )
          ),
          Defn.Class(
            Nil,
            pname("A2"),
            Nil,
            EmptyCtor(),
            tpl(
              Init(
                Type.Select(
                  Term.Select(tname("scala"), tname("annotation")),
                  pname("StaticAnnotation")
                ),
                anon,
                emptyArgClause
              ) :: Nil,
              Nil
            )
          ),
          Defn.Class(
            List(
              Mod.Annot(Init(pname("A1"), anon, emptyArgClause)),
              Mod.Annot(Init(pname("A2"), anon, emptyArgClause))
            ),
            pname("B"),
            Nil,
            EmptyCtor(),
            EmptyTemplate()
          )
        )
      )
    )
  }

  test("i2505") {
    runTestAssert[Source](
      """|trait T2 { self: T =>
         |  enum T2Enum:
         |    case EnumCase
         |  
         |  extension (n: Int) def negate: Int = -n
         |}""".stripMargin,
      assertLayout = Some(
        """|trait T2 { self: T =>
           |  enum T2Enum { case EnumCase }
           |  extension (n: Int) def negate: Int = -n
           |}
           |""".stripMargin
      )
    )(
      Source(
        List(
          Defn.Trait(
            Nil,
            pname("T2"),
            Nil,
            EmptyCtor(),
            Template(
              Nil,
              Nil,
              self("self", "T"),
              List(
                Defn.Enum(
                  Nil,
                  pname("T2Enum"),
                  Nil,
                  EmptyCtor(),
                  tpl(
                    Defn.EnumCase(
                      Nil,
                      tname("EnumCase"),
                      Nil,
                      EmptyCtor(),
                      Nil
                    )
                  )
                ),
                Defn.ExtensionGroup(
                  Nil,
                  List(List(tparam("n", "Int"))),
                  Defn.Def(
                    Nil,
                    tname("negate"),
                    Nil,
                    Nil,
                    Some(pname("Int")),
                    Term.ApplyUnary(tname("-"), tname("n"))
                  )
                )
              ),
              Nil
            )
          )
        )
      )
    )

  }

  test("#3113") {
    runTestAssert[Source](
      """|object Hello {
         |  val fun = () =>
         |    if (true) {
         |      new Object { obj =>
         |        println(toString)
         |      }
         |    }
         |
         |  def main(args: Array[String]): Unit = fun()
         |}""".stripMargin,
      assertLayout = Some(
        """|object Hello {
           |  val fun = () => if (true) {
           |    new Object { obj => println(toString) }
           |  }
           |  def main(args: Array[String]): Unit = fun()
           |}
           |""".stripMargin
      )
    )(
      Source(
        Defn.Object(
          Nil,
          tname("Hello"),
          tpl(
            Defn.Val(
              Nil,
              List(Pat.Var(tname("fun"))),
              None,
              Term.Function(
                Nil,
                Term.If(
                  bool(true),
                  Term.Block(
                    Term.NewAnonymous(
                      Template(
                        Nil,
                        List(Init(pname("Object"), anon, emptyArgClause)),
                        Self(tname("obj"), None),
                        List(Term.Apply(tname("println"), List(tname("toString")))),
                        Nil
                      )
                    ) :: Nil
                  ),
                  Lit.Unit(),
                  Nil
                )
              )
            ),
            Defn.Def(
              Nil,
              tname("main"),
              Nil,
              List(List(tparam("args", Type.Apply(pname("Array"), List(pname("String")))))),
              Some(pname("Unit")),
              Term.Apply(tname("fun"), Nil)
            )
          )
        ) :: Nil
      )
    )
  }

  test("newline within self-type") {
    val layout =
      """|trait T2 {
         |  self { T =>
         |  }
         |  enum T2Enum { case EnumCase }
         |}
         |""".stripMargin
    val tree = Defn.Trait(
      Nil,
      pname("T2"),
      Nil,
      ctor,
      tpl(
        Term.Apply(
          tname("self"),
          Term.Block(List(Term.Function(List(tparam("T")), Term.Block(Nil)))) :: Nil
        ),
        Defn.Enum(
          Nil,
          pname("T2Enum"),
          Nil,
          ctor,
          tpl(
            Defn.EnumCase(Nil, tname("EnumCase"), Nil, ctor, Nil)
          )
        )
      )
    )
    runTestAssert[Stat](
      """|trait T2 {
         |  self:
         |    T =>
         |  enum T2Enum:
         |    case EnumCase
         |}""".stripMargin,
      assertLayout = Some(layout)
    )(tree)
    runTestAssert[Stat](
      """|trait T2:
         |  self:
         |    T
         |    =>
         |  enum T2Enum:
         |    case EnumCase
         |""".stripMargin,
      assertLayout = Some(layout)
    )(tree)
  }

  test("#3210") {
    val code =
      """|a3 match {
         |  case Some(_) =>
         |    case class A6(a7: A8)
         |
         |    object A9
         |}
         |""".stripMargin
    val layout =
      """|a3 match {
         |  case Some(_) =>
         |    case class A6(a7: A8)
         |    object A9
         |}
         |""".stripMargin
    val tree = Term.Match(
      tname("a3"),
      Case(
        Pat.Extract(tname("Some"), List(Pat.Wildcard())),
        None,
        Term.Block(
          List(
            Defn.Class(
              List(Mod.Case()),
              pname("A6"),
              Nil,
              ctorp(tparam("a7", "A8")),
              EmptyTemplate()
            ),
            Defn.Object(Nil, tname("A9"), EmptyTemplate())
          )
        )
      ) :: Nil,
      Nil
    )
    runTestAssert[Stat](code, assertLayout = Some(layout))(tree)
  }

  test("def body is partial function") {
    val code =
      """|object a:
         |  def foo: Bar =
         |    case sym
         |        if baz =>
         |  end foo
         |""".stripMargin
    val layout =
      """|object a {
         |  def foo: Bar = {
         |    case sym if baz =>
         |  }
         |  end foo
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Defn.Def(
            Nil,
            tname("foo"),
            Nil,
            Some(pname("Bar")),
            Term.PartialFunction(
              List(Case(Pat.Var(tname("sym")), Some(tname("baz")), Term.Block(Nil)))
            )
          ),
          Term.EndMarker(tname("foo"))
        )
      )
    }
  }

  test("def body is non-partial function") {
    val code =
      """|object a:
         |  def foo: Bar =
         |    case class Baz(baz: Int)
         |    new Baz(0)
         |""".stripMargin
    val layout =
      """|object a {
         |  def foo: Bar = {
         |    case class Baz(baz: Int)
         |    new Baz(0)
         |  }
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Defn.Def(
            Nil,
            tname("foo"),
            Nil,
            Some(pname("Bar")),
            Term.Block(
              List(
                Defn.Class(
                  List(Mod.Case()),
                  pname("Baz"),
                  Nil,
                  Ctor.Primary(
                    Nil,
                    anon,
                    List(List(tparam("baz", "Int")))
                  ),
                  EmptyTemplate()
                ),
                Term.New(Init(pname("Baz"), anon, List(List(int(0)))))
              )
            )
          )
        )
      )
    }
  }

  test("#3252 `new` in arg, then comma 1") {
    val code =
      """|object a:
         |  A(
         |    b = new B,
         |    c =
         |      d,
         |    e = f
         |  )
         |""".stripMargin
    val layout = "object a { A(b = new B, c = d, e = f) }"
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Term.Apply(
            tname("A"),
            List(
              Term.Assign(tname("b"), Term.New(init("B"))),
              Term.Assign(tname("c"), tname("d")),
              Term.Assign(tname("e"), tname("f"))
            )
          )
        )
      )
    }
  }

  test("#3252 `new` in arg, then comma 2") {
    val code =
      """|object a:
         |  A(
         |    b = new B(0),
         |    c =
         |      d,
         |    e = f
         |  )
         |""".stripMargin
    val layout = "object a { A(b = new B(0), c = d, e = f) }"
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Term.Apply(
            tname("A"),
            List(
              Term.Assign(
                tname("b"),
                Term.New(init("B", List(List(int(0)))))
              ),
              Term.Assign(tname("c"), tname("d")),
              Term.Assign(tname("e"), tname("f"))
            )
          )
        )
      )
    }
  }

  test("#3252 `new` in arg, then comma 3") {
    val code =
      """|object a:
         |  A(
         |    b = new B(0) { def foo = ??? },
         |    c =
         |      d,
         |    e = f
         |  )
         |""".stripMargin
    val layout = "object a { A(b = new B(0) { def foo = ??? }, c = d, e = f) }"
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Term.Apply(
            tname("A"),
            List(
              Term.Assign(
                tname("b"),
                Term.NewAnonymous(
                  tpl(
                    List(init("B", List(List(int(0))))),
                    List(Defn.Def(Nil, tname("foo"), Nil, None, tname("???")))
                  )
                )
              ),
              Term.Assign(tname("c"), tname("d")),
              Term.Assign(tname("e"), tname("f"))
            )
          )
        )
      )
    }
  }

  test("#3257 `new` in arg value after indent, then comma") {
    val code =
      """|object a:
         |  A(
         |    b =
         |      new B,
         |    c = d
         |  )
         |""".stripMargin
    val layout = "object a { A(b = new B, c = d) }"
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Term.Apply(
            tname("A"),
            List(
              Term.Assign(tname("b"), Term.New(init("B"))),
              Term.Assign(tname("c"), tname("d"))
            )
          )
        )
      )
    }
  }

  test("#3257 `if-no-else` in arg value after indent, then comma") {
    val code =
      """|object a:
         |  A(
         |    b =
         |      if (a + b) c,
         |    c = d
         |  )
         |""".stripMargin
    val layout = "object a { A(b = if (a + b) c, c = d) }"
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("a"),
        tpl(
          Term.Apply(
            tname("A"),
            List(
              Term.Assign(
                tname("b"),
                Term.If(
                  Term.ApplyInfix(tname("a"), tname("+"), Nil, List(tname("b"))),
                  tname("c"),
                  Lit.Unit(),
                  Nil
                )
              ),
              Term.Assign(tname("c"), tname("d"))
            )
          )
        )
      )
    }
  }

  test("#3261 chained `match` with outdent, no block, in assign") {
    val code =
      """|object small:
         |  val value =
         |    Nil match
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  match
         |    case "empty"    => 0
         |    case "nonempty" => 1
         |""".stripMargin
    val layout =
      """|object small {
         |  val value = (Nil match {
         |    case Nil => "empty"
         |    case _ => "nonempty"
         |  }) match {
         |    case "empty" => 0
         |    case "nonempty" => 1
         |  }
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Defn.Val(
            Nil,
            List(Pat.Var(tname("value"))),
            None,
            Term.Match(
              Term.Match(
                tname("Nil"),
                List(
                  Case(tname("Nil"), None, str("empty")),
                  Case(Pat.Wildcard(), None, str("nonempty"))
                ),
                Nil
              ),
              List(
                Case(str("empty"), None, int(0)),
                Case(str("nonempty"), None, int(1))
              ),
              Nil
            )
          )
        )
      )
    }
  }

  test("#3261 chained `match` with outdent, block, in assign") {
    val code =
      """|object small:
         |  val value =
         |    foo()
         |    Nil match
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  match
         |    case "empty"    => 0
         |    case "nonempty" => 1
         |""".stripMargin
    val layout =
      """|object small {
         |  val value = {
         |    foo()
         |    Nil match {
         |      case Nil => "empty"
         |      case _ => "nonempty"
         |    }
         |  } match {
         |    case "empty" => 0
         |    case "nonempty" => 1
         |  }
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Defn.Val(
            Nil,
            List(Pat.Var(tname("value"))),
            None,
            Term.Match(
              Term.Block(
                List(
                  Term.Apply(tname("foo"), Term.ArgClause(Nil, None)),
                  Term.Match(
                    tname("Nil"),
                    List(
                      Case(tname("Nil"), None, str("empty")),
                      Case(Pat.Wildcard(), None, str("nonempty"))
                    ),
                    Nil
                  )
                )
              ),
              List(
                Case(str("empty"), None, int(0)),
                Case(str("nonempty"), None, int(1))
              ),
              Nil
            )
          )
        )
      )
    }
  }

  test("#3261 chained `match` with outdent, no block, in if cond") {
    val code =
      """|object small:
         |  if
         |    Nil match
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  match
         |    case "empty"    => true
         |    case "nonempty" => false
         |  then
         |    bar
         |""".stripMargin
    val layout =
      """|object small {
         |  if ((Nil match {
         |    case Nil => "empty"
         |    case _ => "nonempty"
         |  }) match {
         |    case "empty" => true
         |    case "nonempty" => false
         |  }) bar
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Term.If(
            Term.Match(
              Term.Match(
                tname("Nil"),
                List(
                  Case(tname("Nil"), None, str("empty")),
                  Case(Pat.Wildcard(), None, str("nonempty"))
                ),
                Nil
              ),
              List(
                Case(str("empty"), None, bool(true)),
                Case(str("nonempty"), None, bool(false))
              ),
              Nil
            ),
            tname("bar"),
            Lit.Unit(),
            Nil
          )
        )
      )
    }
  }

  test("#3261 chained `match` with outdent, block, in if cond") {
    val code =
      """|object small:
         |  if
         |    foo()
         |    Nil match
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  match
         |    case "empty"    => true
         |    case "nonempty" => false
         |  then
         |    bar
         |""".stripMargin
    val layout =
      """|object small {
         |  if ({
         |    foo()
         |    Nil match {
         |      case Nil => "empty"
         |      case _ => "nonempty"
         |    }
         |  } match {
         |    case "empty" => true
         |    case "nonempty" => false
         |  }) bar
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Term.If(
            Term.Match(
              Term.Block(
                List(
                  Term.Apply(tname("foo"), Nil),
                  Term.Match(
                    tname("Nil"),
                    List(
                      Case(tname("Nil"), None, str("empty")),
                      Case(Pat.Wildcard(), None, str("nonempty"))
                    ),
                    Nil
                  )
                )
              ),
              List(
                Case(str("empty"), None, bool(true)),
                Case(str("nonempty"), None, bool(false))
              ),
              Nil
            ),
            tname("bar"),
            Lit.Unit(),
            Nil
          )
        )
      )
    }
  }

  test("#3261 chained `match` with outdent, no block, in try") {
    val code =
      """|object small:
         |  try
         |    Nil match
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  match
         |    case "empty"    => 0
         |    case "nonempty" => 1
         |  catch
         |    case e => e.getMessage()
         |""".stripMargin
    val layout =
      """|object small {
         |  try (Nil match {
         |    case Nil => "empty"
         |    case _ => "nonempty"
         |  }) match {
         |    case "empty" => 0
         |    case "nonempty" => 1
         |  } catch {
         |    case e =>
         |      e.getMessage()
         |  }
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Term.Try(
            Term.Match(
              Term.Match(
                tname("Nil"),
                List(
                  Case(tname("Nil"), None, str("empty")),
                  Case(Pat.Wildcard(), None, str("nonempty"))
                ),
                Nil
              ),
              List(
                Case(str("empty"), None, int(0)),
                Case(str("nonempty"), None, int(1))
              ),
              Nil
            ),
            Case(
              Pat.Var(tname("e")),
              None,
              Term.Apply(Term.Select(tname("e"), tname("getMessage")), Nil)
            ) :: Nil,
            None
          )
        )
      )
    }
  }

  test("#3261 chained `match` with outdent, block, in try") {
    val code =
      """|object small:
         |  try
         |    foo()
         |    Nil match
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  match
         |    case "empty"    => 0
         |    case "nonempty" => 1
         |  catch
         |    case e => e.getMessage()
         |""".stripMargin
    val layout =
      """|object small {
         |  try {
         |    foo()
         |    Nil match {
         |      case Nil => "empty"
         |      case _ => "nonempty"
         |    }
         |  } match {
         |    case "empty" => 0
         |    case "nonempty" => 1
         |  } catch {
         |    case e =>
         |      e.getMessage()
         |  }
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Term.Try(
            Term.Match(
              Term.Block(
                List(
                  Term.Apply(tname("foo"), Term.ArgClause(Nil, None)),
                  Term.Match(
                    tname("Nil"),
                    List(
                      Case(tname("Nil"), None, str("empty")),
                      Case(Pat.Wildcard(), None, str("nonempty"))
                    ),
                    Nil
                  )
                )
              ),
              List(
                Case(str("empty"), None, int(0)),
                Case(str("nonempty"), None, int(1))
              ),
              Nil
            ),
            Case(
              Pat.Var(tname("e")),
              None,
              Term.Apply(Term.Select(tname("e"), tname("getMessage")), Nil)
            ) :: Nil,
            None
          )
        )
      )
    }
  }

  test("#3261 partial func, in try") {
    val code =
      """|object small:
         |  try
         |    case Nil => "empty"
         |    case _   => "nonempty"
         |  catch
         |    case e => e.getMessage()
         |""".stripMargin
    val layout =
      """|object small {
         |  try {
         |    case Nil => "empty"
         |    case _ => "nonempty"
         |  } catch {
         |    case e =>
         |      e.getMessage()
         |  }
         |}
         |""".stripMargin
    runTestAssert[Stat](code, Some(layout)) {
      Defn.Object(
        Nil,
        tname("small"),
        tpl(
          Term.Try(
            Term.PartialFunction(
              List(
                Case(tname("Nil"), None, str("empty")),
                Case(Pat.Wildcard(), None, str("nonempty"))
              )
            ),
            Case(
              Pat.Var(tname("e")),
              None,
              Term.Apply(Term.Select(tname("e"), tname("getMessage")), Nil)
            ) :: Nil,
            None
          )
        )
      )
    }
  }

  test("blank after template 1") {
    val code =
      """|class DerivationSpec {
         |  case class Foo()
         |
         |  {
         |    deriveEncoder[Foo]
         |  }
         |
         |  {
         |    deriveEncoder[Foo]
         |  }
         |}
         |""".stripMargin
    val layout =
      """|class DerivationSpec {
         |  case class Foo()
         |
         |  {
         |    deriveEncoder[Foo]
         |  }
         |  {
         |    deriveEncoder[Foo]
         |  }
         |}
         |""".stripMargin
    val tree = Defn.Class(
      Nil,
      pname("DerivationSpec"),
      Nil,
      ctor,
      tpl(
        Defn.Class(
          List(Mod.Case()),
          pname("Foo"),
          Nil,
          ctorp(Nil),
          EmptyTemplate()
        ),
        Term.Block(List(Term.ApplyType(tname("deriveEncoder"), List(pname("Foo"))))),
        Term.Block(List(Term.ApplyType(tname("deriveEncoder"), List(pname("Foo")))))
      )
    )
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("blank after template 2") {
    val code =
      """|class DerivationSpec {
         |  case class Foo() extends Bar
         |
         |  {
         |    deriveEncoder[Foo]
         |  }
         |
         |  {
         |    deriveEncoder[Foo]
         |  }
         |}
         |""".stripMargin
    val layout =
      """|class DerivationSpec {
         |  case class Foo() extends Bar
         |
         |  {
         |    deriveEncoder[Foo]
         |  }
         |  {
         |    deriveEncoder[Foo]
         |  }
         |}
         |""".stripMargin
    val tree = Defn.Class(
      Nil,
      pname("DerivationSpec"),
      Nil,
      ctor,
      tpl(
        Defn.Class(
          List(Mod.Case()),
          pname("Foo"),
          Nil,
          ctorp(Nil),
          tpl(List(init("Bar")), Nil)
        ),
        Term.Block(List(Term.ApplyType(tname("deriveEncoder"), List(pname("Foo"))))),
        Term.Block(List(Term.ApplyType(tname("deriveEncoder"), List(pname("Foo")))))
      )
    )
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("def name after newline") {
    val code =
      """|def
         |  foo: Bar =
         |    baz
         |""".stripMargin
    val layout = "def foo: Bar = baz"
    val tree = Defn.Def(Nil, tname("foo"), Nil, Some(pname("Bar")), tname("baz"))
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("def colon after newline") {
    val code =
      """|def foo
         |  : Bar =
         |    baz
         |""".stripMargin
    val layout = "def foo: Bar = baz"
    val tree = Defn.Def(Nil, tname("foo"), Nil, Some(pname("Bar")), tname("baz"))
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("def newline after colon") {
    val code =
      """|def foo
         |  :
         |  Bar =
         |    baz
         |""".stripMargin
    val layout = "def foo: Bar = baz"
    val tree = Defn.Def(Nil, tname("foo"), Nil, Some(pname("Bar")), tname("baz"))
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("def newline around hash") {
    val code =
      """|def foo: Bar
         |  #
         |  SubBar =
         |    baz
         |""".stripMargin
    val layout = "def foo: Bar#SubBar = baz"
    val tree = Defn.Def(
      Nil,
      tname("foo"),
      Nil,
      Some(Type.Project(pname("Bar"), pname("SubBar"))),
      tname("baz")
    )
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("def equals after newline, without type") {
    val code =
      """|def foo
         |  =
         |    baz
         |""".stripMargin
    val layout = "def foo = baz"
    val tree = Defn.Def(Nil, tname("foo"), Nil, None, tname("baz"))
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("def equals after newline, with type") {
    val code =
      """|def foo: Bar
         |  =
         |    baz
         |""".stripMargin
    val layout = "def foo: Bar = baz"
    val tree = Defn.Def(Nil, tname("foo"), Nil, Some(pname("Bar")), tname("baz"))
    runTestAssert[Stat](code, Some(layout))(tree)
  }

  test("#3571 scala3") {
    val code =
      """|new A {
         |  def b: C =
         |    ???
         |}
         |""".stripMargin
    val layout =
      """|new A { def b: C = ??? }
         |""".stripMargin
    val tree = Term.NewAnonymous(
      tpl(
        List(Init(pname("A"), anon, Nil)),
        List(Defn.Def(Nil, tname("b"), Nil, Some(pname("C")), tname("???")))
      )
    )
    runTestAssert[Stat](code, layout)(tree)
  }

  test("#3605 scala3") {
    val code =
      """|new A {
         |  def b: C =
         |    ???
         |
         |}
         |""".stripMargin
    val layout =
      """|new A { def b: C = ??? }
         |""".stripMargin
    val tree = Term.NewAnonymous(
      tpl(
        List(Init(pname("A"), anon, Nil)),
        List(Defn.Def(Nil, tname("b"), Nil, Some(pname("C")), tname("???")))
      )
    )
    runTestAssert[Stat](code, layout)(tree)
  }

}