Merge pull request #4 from eug-vs/populateDb

Populate db script

3 files changed, 66 insertions, 2 deletions

diff --git a/models/polls/poll.schema.ts b/models/polls/poll.schema.ts
index ea91051..83de688 100644
--- a/models/polls/poll.schema.ts
+++ b/models/polls/poll.schema.ts
@@ -1,7 +1,7 @@
 import { Document, Schema, Types } from 'mongoose';
 import { User } from '../users/user.schema'
 
-interface ImageData {
+export interface ImageData {
   url: string;
   votes: number;
 }
diff --git a/package-lock.json b/package-lock.json
index 35cc880..034f4ee 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,5 +1,5 @@
 {
-  "name": "@oneflow/which-api",
+  "name": "which-api",
   "version": "1.0.0",
   "lockfileVersion": 1,
   "requires": true,
diff --git a/populateDb.ts b/populateDb.ts
new file mode 100644
index 0000000..8000111
--- /dev/null
+++ b/populateDb.ts
@@ -0,0 +1,64 @@
+import mongoose from 'mongoose';
+import bluebird from 'bluebird';
+import _ from 'lodash';
+import app from './app';
+import { UserSchema } from './models/users/user.schema';
+import { PollSchema, ImageData } from './models/polls/poll.schema';
+
+mongoose.connect('mongodb://localhost:27017/which', { useNewUrlParser: true });
+
+const POLLS_AMOUNT = 20;
+
+const imageUrls: string[] = [
+  // eslint-disable max-len
+  'https://cdn.psychologytoday.com/sites/default/files/field_blog_entry_images/2019-06/pexels-photo-556667.jpeg',
+  'https://images.pexels.com/photos/556666/pexels-photo-556666.jpeg?auto=compress&cs=tinysrgb&dpr=1&w=500',
+  'https://i.pinimg.com/originals/50/91/3e/50913eeb04768a5b1fa9985c16704d96.jpg',
+  'https://grazia.wwmindia.com/photogallery/2017/apr/1_1491461089.jpg'
+];
+
+const names: string[] = [
+  'Emma',
+  'Elise',
+  'Jack',
+  'Oliver',
+  'Jamie',
+  'Adam',
+  'Jordan',
+  'William'
+];
+
+const generateImageData = (): ImageData => ({
+  url: _.sample(imageUrls) || '',
+  votes: Math.floor(Math.random() * 101)
+});
+
+const createPoll = (authorId: string): Promise<PollSchema> => {
+  return app.service('polls').create({
+    contents: {
+      left: generateImageData(),
+      right: generateImageData(),
+    },
+    authorId
+  });
+};
+
+const createUser = (name: string): Promise<UserSchema> => {
+  return app.service('users').create({
+    avatarUrl: _.sample(imageUrls) || '',
+    name
+  });
+};
+
+
+const populate = async () => {
+  const users = await bluebird.map(names, name => createUser(name));
+
+  for (let i = 0; i < POLLS_AMOUNT; i++) {
+    const sampleUser = _.sample(users);
+    await createPoll(sampleUser?._id);
+  };
+};
+
+populate().finally(mongoose.disconnect);
+