fix: move volatileData to MockFirebaseDatabase

sitatec · Sep 19, 2022 · 84c3e53 · 84c3e53
1 parent c8ddf1c
commit 84c3e53
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/lib/src/mock_database_reference.dart b/lib/src/mock_database_reference.dart
@@ -11,9 +11,9 @@ class MockDatabaseReference extends Mock implements DatabaseReference {
 
   // ignore: prefer_final_fields
   static Map<String, dynamic>? _persistedData = <String, dynamic>{};
-  Map<String, dynamic>? _volatileData = <String, dynamic>{};
+  Map<String, dynamic>? _volatileData;
 
-  MockDatabaseReference();
+  MockDatabaseReference([this._volatileData]);
 
   MockDatabaseReference._(nodePath, [this._volatileData]) {
     _nodePath += nodePath;

diff --git a/lib/src/mock_firebase_database.dart b/lib/src/mock_firebase_database.dart
@@ -6,15 +6,18 @@ import 'mock_database_reference.dart';
 class MockFirebaseDatabase extends Mock implements FirebaseDatabase {
   static FirebaseDatabase get instance => MockFirebaseDatabase();
   static get persistData => _persistData;
+
+  Map<String, dynamic> _volatileData = <String, dynamic>{};
+
   @override
   DatabaseReference reference() => ref();
 
   @override
   DatabaseReference ref([String? path]) {
     if (path != null) {
-      return MockDatabaseReference().child(path);
+      return MockDatabaseReference(_volatileData).child(path);
     }
-    return MockDatabaseReference();
+    return MockDatabaseReference(_volatileData);
   }
 
   // ignore: unused_field