feat: can use scenarioWidgetBuilder when snapshot

playbook-ui · Sep 7, 2024 · bdff38f · bdff38f
1 parent f27d55a
commit bdff38f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/packages/playbook_snapshot/lib/src/snapshot.dart b/packages/playbook_snapshot/lib/src/snapshot.dart
@@ -46,6 +46,7 @@ class Snapshot implements TestTool {
     Playbook playbook,
     WidgetTester tester,
     PlaybookBuilder builder, {
+    ScenarioWidgetBuilder? scenarioWidgetBuilder,
     Future<void> Function(WidgetTester tester)? setUpEachTest,
   }) async {
     await tester.runAsync(() async {
@@ -80,6 +81,7 @@ class Snapshot implements TestTool {
               checkeredColor: checkeredColor,
               checkeredRectSize: checkeredRectSize,
               useMaterial: useMaterial,
+              builder: scenarioWidgetBuilder,
               scenario: scenario,
             ),
             device,

diff --git a/packages/playbook_snapshot/lib/src/test_tool.dart b/packages/playbook_snapshot/lib/src/test_tool.dart
@@ -10,6 +10,7 @@ abstract class TestTool {
     Playbook playbook,
     WidgetTester tester,
     PlaybookBuilder builder, {
+    ScenarioWidgetBuilder? scenarioWidgetBuilder,
     Future<void> Function(WidgetTester tester)? setUpEachTest,
   });
 }
@@ -19,12 +20,14 @@ extension PlaybookExt on Playbook {
     TestTool test,
     WidgetTester tester,
     PlaybookBuilder builder, {
+    ScenarioWidgetBuilder? scenarioWidgetBuilder,
     Future<void> Function(WidgetTester tester)? setUpEachTest,
   }) async {
     await test.run(
       this,
       tester,
       builder,
+      scenarioWidgetBuilder: scenarioWidgetBuilder,
       setUpEachTest: setUpEachTest,
     );
   }