Add 'all' method to entity store (#40)

Author: davidtkramer

package.json

@@ -1,6 +1,6 @@
 {
   "name": "typical-data",
-  "version": "0.4.2",
+  "version": "0.5.0",
   "description": "Test data factory",
   "homepage": "https://github.com/davidtkramer/typical-data",
   "main": "./dist/index.js",

src/__tests__/database.test.ts

@@ -219,6 +219,40 @@ describe('entity store', () => {
     });
   });
 
+  describe('all', () => {
+    it('returns all entities for regular entity store', () => {
+      db.users.createList(2);
+
+      const users = db.users.all();
+
+      expect(users).toHaveLength(2);
+      // returns shallow copy of entities array
+      db.users.create();
+      expect(users).toHaveLength(2);
+    });
+
+    it('returns all entities for polymorphic entity store', () => {
+      const db = createDatabase({
+        factories: {
+          contacts: {
+            individual: createFactory({ id: 1, fullName: '' }),
+            business: createFactory({ id: 1, businessName: '' }),
+          },
+        },
+      });
+
+      db.contacts.individual.create();
+      db.contacts.business.create();
+
+      const contacts = db.contacts.all();
+
+      expect(contacts).toHaveLength(2);
+      // returns shallow copy of entities array
+      db.contacts.individual.create();
+      expect(contacts).toHaveLength(2);
+    });
+  });
+
   describe('reset', () => {
     it('resets entity store', () => {
       db.users.createList(2);

src/database.ts

@@ -29,13 +29,15 @@ interface EntityStore<
   create: Factory['build'];
   createList: Factory['buildList'];
   reset(): Array<Entity>;
+  all(): Array<Entity>;
 }
 
 interface ParentEntityStore<
   Factory extends EntityFactory<unknown, unknown, unknown>,
   Entity = EntityFromFactory<Factory>
 > extends Array<Entity> {
   reset(): Array<Entity>;
+  all(): Array<Entity>;
 }
 
 type EntityStores<FC extends FactoryConfig> = {
@@ -103,6 +105,9 @@ export function createDatabase<
           factory.rewindSequence();
           return entities.splice(0, entities.length);
         },
+        all() {
+          return [...entities];
+        },
       });
       database[key] = store;
     } else if (typeof item === 'object') {
@@ -113,6 +118,9 @@ export function createDatabase<
           sequence.count = -1;
           return entities.splice(0, entities.length);
         },
+        all() {
+          return [...entities];
+        },
       });
 
       for (const key in item) {