Advertisement
Guest User

Untitled

a guest
Nov 22nd, 2017
160
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.92 KB | None | 0 0
  1. const faker = require('faker');
  2. const should = require('chai').should();
  3. const mongoose = require('mongoose');
  4.  
  5. const {User} = require('../src/models/user');
  6. const {Munch} = require('../src/models/munch');
  7. const {Group} = require('../src/models/group');
  8.  
  9. function seedMunchMinderDatabase() {
  10. console.info('Seeding Users, Munches & Groups collections...');
  11. // const seedUserData = [];
  12. // const seedMunchData = [];
  13. // const seedGroupData = [];
  14. let i = 0;
  15. while (i < 10) {
  16. createTestUserAndPostMunches();
  17. // seedUserData.push(generateUserData());
  18. // seedMunchData.push(generateMunchData());
  19. i++;
  20. };
  21. // const insertUserData = User.insertMany(seedUserData);
  22. // const insertMunchData = Munch.insertMany(seedMunchData);
  23. // return Promise.all([insertUserData, insertMunchData])
  24. }
  25.  
  26. function generateUserData() {
  27. return {
  28. userName: faker.internet.userName(),
  29. userEmail: faker.internet.email(),
  30. password: faker.internet.password(),
  31. };
  32. }
  33.  
  34. function generateMunchData() {
  35. return {
  36. date: faker.date.past(),
  37. title: faker.lorem.words(),
  38. description: faker.lorem.sentence(),
  39. emoji: faker.image.avatar(),
  40. image: faker.image.imageUrl(),
  41. thumbsUp: faker.random.boolean(),
  42. }
  43. }
  44.  
  45. function createTestUser() {
  46. return User.create(generateUserData());
  47. }
  48.  
  49. function createTestUserAndPostMunches() {
  50. User.create(generateUserData())
  51. .then(user => {
  52. let userId = user._id;
  53. let username = user.userName
  54. let j = 0;
  55. while (j < 10) {
  56. let newMunch = generateMunchData();
  57. newMunch.postedBy = userId;
  58. newMunch.userName = username;
  59. Munch.create(newMunch);
  60. j++;
  61. }
  62. })
  63. }
  64.  
  65. function teardownDatabase() {
  66. console.warn('Deleting database...');
  67. return mongoose.connection.dropDatabase();
  68. }
  69.  
  70. module.exports = {seedMunchMinderDatabase, generateUserData, generateMunchData, createTestUser, teardownDatabase}
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement