Add class method to reset all stores

hugollm · May 7, 2019 · 5d17783 · 5d17783
1 parent 54040a0
commit 5d17783
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/src/true-store.js b/src/true-store.js
@@ -10,6 +10,7 @@ class TrueStore {
         this.stateMap = this.initialMap;
         this.observers = [];
         this.transactionDepth = 0;
+        TrueStore.stores.push(this);
     }
 
     get(key = null) {
@@ -84,6 +85,13 @@ class TrueStore {
     }
 }
 
+TrueStore.stores = [];
+
+TrueStore.resetAll = function() {
+    for (let i in TrueStore.stores)
+        TrueStore.stores[i].reset();
+}
+
 
 class TrueStoreObserver {
 

diff --git a/tests/reset-all.test.js b/tests/reset-all.test.js
@@ -0,0 +1,14 @@
+const TrueStore = require('../src/true-store');
+
+describe('resetAll', () => {
+
+    it('returns all stores to their initial state', () => {
+        let store1 = new TrueStore({foo: 1});
+        let store2 = new TrueStore({bar: 2});
+        store1.set('foo',3);
+        store2.set('bar', 4);
+        TrueStore.resetAll();
+        expect(store1.get()).toEqual({foo: 1});
+        expect(store2.get()).toEqual({bar: 2});
+    });
+});