Refactor error message checks

scalalandio · Sep 11, 2023 · 996b0db · 996b0db
1 parent 9d14600
commit 996b0db
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 13 deletions.
diff --git a/chimney/src/test/scala-3/io/scalaland/chimney/TotalTransformerEnumSpec.scala b/chimney/src/test/scala-3/io/scalaland/chimney/TotalTransformerEnumSpec.scala
@@ -137,8 +137,8 @@ class TotalTransformerEnumSpec extends ChimneySpec {
       "Consult https://chimney.readthedocs.io for usage examples."
     )
 
-    assert(
-      !error.contains("coproduct instance Circle of io.scalaland.chimney.fixtures.shapes5enums.Shape is ambiguous")
+    error.checkNot(
+      "coproduct instance Circle of io.scalaland.chimney.fixtures.shapes5enums.Shape is ambiguous"
     )
   }
 

diff --git a/chimney/src/test/scala/io/scalaland/chimney/ChimneySpec.scala b/chimney/src/test/scala/io/scalaland/chimney/ChimneySpec.scala
@@ -38,15 +38,31 @@ trait ChimneySpec extends munit.BaseFunSuite with VersionCompat { self =>
 
   implicit class CompileErrorsCheck(msg: String) {
 
-    def check(msgs: String*): Unit = for (msg <- msgs)
-      Predef.assert(
-        ChimneySpec.AnsiControlCode.replaceAllIn(this.msg, "").contains(msg),
-        "Error message did not contain expected snippet\n" +
-          "Error message\n" +
-          this.msg + "\n" +
-          "Expected Snippet\n" +
-          msg
-      )
+    def check(msgs: String*): Unit = {
+      val msgNoColors = ChimneySpec.AnsiControlCode.replaceAllIn(this.msg, "")
+      for (msg <- msgs)
+        Predef.assert(
+          msgNoColors.contains(msg),
+          s"""Error message did not contain expected snippet
+            |Error message
+            |${this.msg}
+            |Expected Snippet
+            |$msg""".stripMargin
+        )
+    }
+
+    def checkNot(msgs: String*): Unit = {
+      val msgNoColors = ChimneySpec.AnsiControlCode.replaceAllIn(this.msg, "")
+      for (msg <- msgs)
+        Predef.assert(
+          !msgNoColors.contains(msg),
+          s"""Error message contain snippet that was expected to be not there
+            |Error message
+            |${this.msg}
+            |Not Expected Snippet
+            |$msg""".stripMargin
+        )
+    }
 
     def arePresent(): Unit = Predef.assert(msg.nonEmpty, "Expected compilation errors")
   }

diff --git a/chimney/src/test/scala/io/scalaland/chimney/TotalTransformerSealedHierarchySpec.scala b/chimney/src/test/scala/io/scalaland/chimney/TotalTransformerSealedHierarchySpec.scala
@@ -104,8 +104,8 @@ class TotalTransformerSealedHierarchySpec extends ChimneySpec {
       "Consult https://chimney.readthedocs.io for usage examples."
     )
 
-    assert(
-      !error.contains("coproduct instance Circle of io.scalaland.chimney.fixtures.shapes5.Shape is ambiguous")
+    error.checkNot(
+      "coproduct instance Circle of io.scalaland.chimney.fixtures.shapes5.Shape is ambiguous"
     )
   }