Extract snapshots integration

indoorvivants · Feb 10, 2024 · c068c7c · c068c7c
1 parent e1cf32c
commit c068c7c
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 28 deletions.
diff --git a/modules/example/src/test/scala/ExampleTests.scala b/modules/example/src/test/scala/ExampleTests.scala
@@ -7,34 +7,6 @@ import munit.Assertions
 import munit.FunSuite
 import example.Snapshots
 
-// This is a sample integration for Munit
-trait SnapshotsIntegration {
-  self: FunSuite =>
-  def assertSnapshot(name: String, contents: String) = {
-    Snapshots.read(name) match {
-      case None =>
-        Snapshots.recordChanges(
-          name,
-          contents,
-          Diffs.create(contents, "").createDiffOnlyReport()
-        )
-
-        Assertions.fail(
-          s"No snapshot was found for $name, please run checkSnapshots command and accept a snapshot for this test"
-        )
-
-      case Some(value) =>
-        val diff = Diffs.create(contents, value)
-        if (!diff.isEmpty) {
-          val diffReport = diff.createDiffOnlyReport()
-          Snapshots.recordChanges(name, contents, diffReport)
-          Assertions.assertNoDiff(contents, value)
-        } else
-          Snapshots.clearChanges(name)
-    }
-  }
-}
-
 class ExampleTests extends FunSuite with SnapshotsIntegration {
   test("hello") {
     assertSnapshot("my.snapshot", "hello?")

diff --git a/modules/example/src/test/scala/SnapshotsIntegration.scala b/modules/example/src/test/scala/SnapshotsIntegration.scala
@@ -0,0 +1,34 @@
+import munit._
+import com.indoorvivants.snapshots._
+import munit.internal.difflib.Diffs
+import munit.Assertions
+import munit.FunSuite
+import example.Snapshots
+
+// This is a sample integration for Munit
+trait SnapshotsIntegration {
+  self: FunSuite =>
+  def assertSnapshot(name: String, contents: String) = {
+    Snapshots.read(name) match {
+      case None =>
+        Snapshots.recordChanges(
+          name,
+          contents,
+          Diffs.create(contents, "").createDiffOnlyReport()
+        )
+
+        Assertions.fail(
+          s"No snapshot was found for $name, please run checkSnapshots command and accept a snapshot for this test"
+        )
+
+      case Some(value) =>
+        val diff = Diffs.create(contents, value)
+        if (!diff.isEmpty) {
+          val diffReport = diff.createDiffOnlyReport()
+          Snapshots.recordChanges(name, contents, diffReport)
+          Assertions.assertNoDiff(contents, value)
+        } else
+          Snapshots.clearChanges(name)
+    }
+  }
+}