package org.jetbrains.plugins.scala
package lang
package formatter
package tests
package scala3

import com.intellij.lang.Language

class Scala3FormatterTest extends AbstractScalaFormatterTestBase {
  override protected def language: Language = Scala3Language.INSTANCE

  def testClassColon(): Unit = doTextTest(
    """
      |class Test :
      |  def test = ()
      |""".stripMargin,
    """
      |class Test:
      |  def test = ()
      |""".stripMargin
  )

  def testClassEnd(): Unit = doTextTest(
    """
      |class Test:
      |  def test = ()
      |end Test
      |""".stripMargin
  )

}

package org.jetbrains.plugins.scala
package lang
package formatter
package tests
package scala3

import com.intellij.lang.Language

class Scala3FormatterTest extends AbstractScalaFormatterTestBase {
  override protected def language: Language = Scala3Language.INSTANCE

  def testClassColon(): Unit = doTextTest(
    """
      |class Test :
      |  def test = ()
      |""".stripMargin,
    """
      |class Test:
      |  def test = ()
      |""".stripMargin
  )

  // TODO: fix end parsing
  //def testClassEnd(): Unit = doTextTest(
  //  """
  //    |class Test:
  //    |  def test = ()
  //    |end Test
  //    |""".stripMargin
  //)

}

--- 
+++ 
@@ -20,13 +20,12 @@
       |""".stripMargin
   )
 
-  // TODO: fix end parsing
-  //def testClassEnd(): Unit = doTextTest(
-  //  """
-  //    |class Test:
-  //    |  def test = ()
-  //    |end Test
-  //    |""".stripMargin
-  //)
+  def testClassEnd(): Unit = doTextTest(
+    """
+      |class Test:
+      |  def test = ()
+      |end Test
+      |""".stripMargin
+  )
 
 }
Revert "Disable end formatting test"

This reverts commit 4544d455

 #SCL-16158
