add more tests

xuwei-k · Mar 10, 2024 · 0b06358 · 0b06358
1 parent 726af15
commit 0b06358
Showing 1 changed file with 41 additions and 4 deletions.
diff --git a/localServer/src/test/scala/scalameta_ast/IntegrationTest.scala b/localServer/src/test/scala/scalameta_ast/IntegrationTest.scala
@@ -95,10 +95,22 @@ abstract class IntegrationTest(browserType: Playwright => BrowserType) extends A
     input.press("\n")
   }
 
+  private def formatOutput(page: Page): Locator = {
+    getById(page, AriaRole.CHECKBOX, "format")
+  }
+
   private def wildcardImport(page: Page): Locator = {
     getById(page, AriaRole.CHECKBOX, "wildcard_import")
   }
 
+  private def removeNewFields(page: Page): Locator = {
+    getById(page, AriaRole.CHECKBOX, "remove_new_fields")
+  }
+
+  private def initialExtractor(page: Page): Locator = {
+    getById(page, AriaRole.CHECKBOX, "initial_extractor")
+  }
+
   private def changeOutputType(page: Page, outputType: String): Unit = {
     page
       .getByRole(AriaRole.RADIO)
@@ -121,6 +133,17 @@ abstract class IntegrationTest(browserType: Playwright => BrowserType) extends A
     getTextboxById(page, "rule_name")
   }
 
+  private def selectedDialect(page: Page): String = {
+    page.getByLabel("dialect").all().asScala.map(_.inputValue()).toList match {
+      case List(value) =>
+        value
+      case Nil =>
+        sys.error("not found dialect")
+      case values =>
+        sys.error(s"found multi values ${values}")
+    }
+  }
+
   "change input" in withBrowser { page =>
     setInput(page, "class A")
     val expect = Seq(
@@ -211,6 +234,7 @@ abstract class IntegrationTest(browserType: Playwright => BrowserType) extends A
     changeOutputType(page, "raw")
     assert(output(page).textContent() contains """Term.Select(Term.Name("enum"), Term.Name("A"))""")
     page.selectOption("select#dialect", "Scala3")
+    assert(selectedDialect(page) == "Scala3")
     assert(output(page).textContent() contains "Defn.Enum")
   }
 
@@ -302,32 +326,36 @@ abstract class IntegrationTest(browserType: Playwright => BrowserType) extends A
 
     assert(outSingleLine() == """Term.If(Term.Name("a"), Lit.Int(2), Lit.Int(3), Nil)""")
 
-    getById(page, AriaRole.CHECKBOX, "remove_new_fields").check()
+    removeNewFields(page).check()
     assert(outSingleLine() == """Term.If(Term.Name("a"), Lit.Int(2), Lit.Int(3))""")
 
-    getById(page, AriaRole.CHECKBOX, "initial_extractor").check()
+    initialExtractor(page).check()
     assert(outSingleLine() == """Term.If.Initial(Term.Name("a"), Lit.Int(2), Lit.Int(3))""")
   }
 
   "localStorage" in withBrowser { page =>
     def check(
       scalafmt: String,
+      formatOut: Boolean,
       wildcard: Boolean,
+      _initialExtractor: Boolean,
       outputType: String,
+      dialect: String,
       pkg: String,
       rule: String,
       input: String
     ) = {
       assert(scalafmtConfig(page).inputValue() == scalafmt)
+      assert(formatOutput(page).isChecked == formatOut)
       assert(wildcardImport(page).isChecked == wildcard)
+      assert(initialExtractor(page).isChecked == _initialExtractor)
       assert(
         page.getByRole(AriaRole.RADIO).all().asScala.filter(_.isChecked).map(_.getAttribute("id")) == Seq(outputType)
       )
+      assert(selectedDialect(page) == dialect)
       assert(packageName(page).inputValue() == pkg)
       assert(ruleName(page).inputValue() == rule)
       assert(inputElem(page).inputValue() == input)
-
-      // TODO check more inputs
     }
 
     changeOutputType(page, "syntactic")
@@ -340,16 +368,22 @@ abstract class IntegrationTest(browserType: Playwright => BrowserType) extends A
         """continuationIndent.defnSite = 2""",
         """continuationIndent.extendSite = 2""",
       ).mkString("\n"),
+      formatOut = true,
       wildcard = false,
+      _initialExtractor = false,
       outputType = "syntactic",
+      dialect = "Auto",
       pkg = "fix",
       rule = "",
       input = "def a = b",
     )
 
     setScalafmtConfig(page, Seq("runner.dialect = Scala213"))
+    formatOutput(page).uncheck()
     wildcardImport(page).check()
+    initialExtractor(page).check()
     changeOutputType(page, "semantic")
+    page.selectOption("select#dialect", "Scala211")
     packageName(page).fill("ppppppppp")
     ruleName(page).fill("FFFFFFFFFF")
     setInput(page, "aaa")
@@ -358,8 +392,11 @@ abstract class IntegrationTest(browserType: Playwright => BrowserType) extends A
 
     check(
       scalafmt = "runner.dialect = Scala213",
+      formatOut = false,
       wildcard = true,
+      _initialExtractor = true,
       outputType = "semantic",
+      dialect = "Scala211",
       pkg = "ppppppppp",
       rule = "FFFFFFFFFF",
       input = "aaa\n",