test: Add optional Account and InitialSnapshot to store() factory

gnprice · Oct 12, 2023 · 2cf1b36 · 2cf1b36
1 parent 664f4a5
commit 2cf1b36
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/test/example_data.dart b/test/example_data.dart
@@ -273,11 +273,12 @@ InitialSnapshot initialSnapshot({
     crossRealmBots: crossRealmBots ?? [],
   );
 }
+const _initialSnapshot = initialSnapshot;
 
-PerAccountStore store() {
+PerAccountStore store({Account? account, InitialSnapshot? initialSnapshot}) {
   return PerAccountStore.fromInitialSnapshot(
-    account: selfAccount,
-    connection: FakeApiConnection.fromAccount(selfAccount),
-    initialSnapshot: initialSnapshot(),
+    account: account ?? selfAccount,
+    connection: FakeApiConnection.fromAccount(account ?? selfAccount),
+    initialSnapshot: initialSnapshot ?? _initialSnapshot(),
   );
 }