create store methods

tonyb983 · Oct 21, 2019 · 225b07c · 225b07c
1 parent c56defb
commit 225b07c
Show file tree

Hide file tree

Showing 2 changed files with 88 additions and 13 deletions.
diff --git a/src/Models/Blog/StoreModel.js b/src/Models/Blog/StoreModel.js
@@ -1,5 +1,5 @@
 import { types, getEnv } from 'mobx-state-tree';
-import { random } from 'lodash';
+import { random, times, isNumber, isArray, isMatch, isEmpty, isNil, has } from 'lodash';
 import faker from 'faker';
 import { Post } from './PostModel';
 import { Blog } from './BlogModel';
@@ -67,9 +67,21 @@ export const createPreloadedStore = () => {
   return store;
 };
 
-export const createRandomStore = () => {};
+export const createStoreFromSnapshot = (snapShot) => {
+  if (!snapShot) return createFreshStore();
+
+  return Store.create(snapShot);
+};
+
+export const createRandomStore = () => {
+
+  const users = CreateUserStore(times(random(5, 10), (_) => CreateRandomUser()));
+  const blog = CreateBlog(times(random(10,40), (_) => CreateRandomPost()))
+
+  return Store.create({blog, users});
+};
 
-const CreateUser = () => {
+export const CreateRandomUser = () => {
   const u = faker.helpers.userCard();
   const p = faker.random.alphaNumeric(random(8, 14));
   const settings = UserSettings.create();
@@ -82,17 +94,65 @@ const CreateUser = () => {
   return User.create({ login: u.email, password: p, settings });
 };
 
-const CreatePost = () => {
+export const CreateUserStore = (options = undefined) => {
+  const store = UserStore.create();
+
+  if(!options || isNil(options)) return store;
+
+  if(isNumber(options) && options > 0){
+    times(count, (_) => CreateRandomUser()).forEach(u => store.registerUser(u));
+    return store;
+  }
+
+  if(has(options,['min', 'max']) &&
+      isNumber(options.min) &&
+      isNumber(options.max) &&
+      options.min >= 0 &&
+      options.max >= options.min){
+    times(random(options.min, options.max), (_) => CreateRandomUser()).forEach(u => store.registerUser(u));
+    return store;
+  }
+
+  if(isArray(options) && options.length > 0 && options.every(i => !isNil(i))){
+    users.forEach(u => store.registerUser(u));
+    return store;
+  }
+
+  return store;
+}
+
+export const CreateRandomPost = () => {
   const title = faker.hacker.phrase;
-  let content = '';
-  const _ = new Array(random(1, 6)).forEach((_) => (content += faker.company.catchPhrase() + ' '));
-  const tags = new Array(random(6));
-  tags.forEach((t, i) => (tags[i] = faker.commerce.department));
+  const content = times(random(1, 6), (_) => faker.company.catchPhrase).join('. ');
+  const tags = times(random(6), (_) => faker.commerce.department);
   return Post.create({ title, content, tags });
 };
 
-export const createStoreFromSnapshot = (snapShot) => {
-  if (!snapShot) return createFreshStore();
+export const CreateBlog = (options = undefined) => {
+  const store = Blog.create();
+
+  if(!options || isNil(options)) return store;
+
+  if(isNumber(options) && options > 0){
+    times(count, (_) => CreateRandomPost()).forEach(p => store.addPost(p));
+    return store;
+  }
+
+  if(has(options,['min', 'max']) &&
+      isNumber(options.min) &&
+      isNumber(options.max) &&
+      options.min >= 0 &&
+      options.max >= options.min){
+    times(random(options.min, options.max), (_) => CreateRandomPost()).forEach(p => store.addPost(p));
+    return store;
+  }
+
+  if(isArray(options) && options.length > 0 && options.every(i => !isNil(i))){
+    users.forEach(p => store.addPost(p));
+    return store;
+  }
+
+  return store;
+}
+
 
-  return Store.create(snapShot);
-};
diff --git a/src/Models/Blog/StoreModel.spec.js b/src/Models/Blog/StoreModel.spec.js
@@ -1,9 +1,15 @@
 import {
   Store,
-  createFreshStore,
+  createEmptyStore,
   createPreloadedStore,
   createStoreFromSnapshot,
+  createRandomStore,
+  CreateRandomPost,
+  CreateRandomUser,
+  CreateBlog,
+  CreateUserStore
 } from './StoreModel';
+import { isString } from 'lodash';
 
 describe('Store Model Tests', () => {
   it('Can create a new Store', () => {
@@ -12,3 +18,12 @@ describe('Store Model Tests', () => {
     expect(store).toBeDefined();
   });
 });
+
+describe('Create Random Functions', () => {
+  it('Can create a random User', () => {
+    const user = CreateRandomUser();
+    expect(user).toBeDefined();
+    expect(isString(user.login)).toBe(true);
+    console.log(`Random User: ${JSON.stringify(user, null, 2)}`);
+  });
+});