Allow regenerating a subset of snapshots

sourcegraph · Aug 6, 2024 · f8f12f6 · f8f12f6
1 parent 115d6ce
commit f8f12f6
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/snapshots/src/main/scala/tests/SaveSnapshots.scala b/tests/snapshots/src/main/scala/tests/SaveSnapshots.scala
@@ -3,7 +3,19 @@ package tests
 object SaveSnapshots {
   def main(args: Array[String]): Unit = {
     val expectDirectory = tests.snapshots.BuildInfo.snapshotDirectory.toPath
-    SemanticdbJavacSnapshotGenerator
-      .run(SnapshotContext(expectDirectory), new SaveSnapshotHandler)
+    val mapping = Map(
+      "minimized" -> new MinimizedSnapshotScipGenerator(),
+      "library" -> new LibrarySnapshotGenerator()
+    )
+
+    val enabledGenerators =
+      if (args.isEmpty)
+        mapping.values.toList
+      else
+        args.flatMap(mapping.get).toList
+
+    val generator = new AggregateSnapshotGenerator(enabledGenerators)
+
+    generator.run(SnapshotContext(expectDirectory), new SaveSnapshotHandler)
   }
 }