generate-test-data.js

/*
 * Copyright (C) 2016 TopCoder Inc., All Rights Reserved.
 */
'use strict';
/**
 * Module to generate test data
 *
 * @author      TSCCODER
 * @version     1.0
 */

require('./bootstrap');
const config = require('config');
const _ = require('lodash');
const logger = require('./common/logger');
const helper = require('./common/helper');
const models = require('./models');
const moment = require('moment');

const Drone = models.Drone;
const User = models.User;
const Category = models.Category;
const Provider = models.Provider;
const Package = models.Package;
const Mission = models.Mission;
const Review = models.Review;
const PackageRequest = models.PackageRequest;
const Notification = models.Notification;
const Service = models.Service;
const DronePosition = models.DronePosition;
const NoFlyZone = models.NoFlyZone;
const Question = models.Question;

let drones;
if (process.argv[2] === 'drones-within-area') {
  drones = require('./data/drones-within-area.json');
} else {
  drones = require('./data/drones.json');
}
const users = require('./data/users.json');
const categories = require('./data/categories.json');
const providers = require('./data/providers.json');
const packages = require('./data/packages.json');
const missions = require('./data/missions.json');
const requests = require('./data/packageRequests.json');
const notifications = require('./data/notifications.json');
const providerUsers = require('./data/provider-users.json');
const positions = require('./data/dronePositions.json');
const noFlyZones = require('./data/no-fly-zones.json');
const questions = require('./data/questions.json');
const waypoints = require('./data/waypoints.json');

const MissionStatus = require('./enum').MissionStatus;
const RequestStatus = require('./enum').RequestStatus;


for (let i = 0; i < requests.length; i++) {
  requests[i].whatToBeDelivered = 'things';
  requests[i].weight = Math.random() * 10;
  requests[i].payout = Math.random() * 10;
}

// players json data
const co = require('co');

co(function*() {
  logger.info('deleting previous data');
  yield Drone.remove({});
  yield User.remove({});
  yield Category.remove({});
  yield Provider.remove({});
  yield Package.remove({});
  yield Mission.remove({});
  yield Review.remove({});
  yield PackageRequest.remove({});
  yield Notification.remove({});
  yield DronePosition.remove({});
  yield NoFlyZone.remove({});
  yield Question.remove({});

    // encrypt password
  yield _.map(users, (u) => function* () {
    if (u.password) {
      u.password = yield helper.hashString(u.password, config.SALT_WORK_FACTOR);
    }
    return;
  });

  logger.info(`creating ${users.length} users`);
  const userDocs = yield User.create(users);

  logger.info(`creating ${categories.length} categories`);
  const categoryDocs = yield Category.create(categories);

  logger.info(`creating ${providers.length} providers`);
  const providerDocs = yield Provider.create(providers);

  yield _.map(providerUsers, (u) => function*() {
    if (u.password) {
      u.password = yield helper.hashString(u.password, config.SALT_WORK_FACTOR);
    }
  });
  _.each(providerUsers, (u, i) => {
    u.provider = providerDocs[i % providerDocs.length].id;
  });
  logger.info(`creating provider ${providerUsers.length} users`);
  const providerUserDocs = yield User.create(providerUsers);


  logger.info('creating 100 services');
  const serviceDcos = [];
  for (let i = 0; i < 100; i++) {
    serviceDcos.push(yield Service.create({
      name: 'service ' + i,
      pricing: i,
      description: 'hello ,i am service ' + i,
      provider: providerDocs[i % providerDocs.length].id,
      category: categoryDocs[i % categoryDocs.length].id,
    }));
  }

  _.each(packages, (p, i) => {
    p.provider = providerDocs[i % providerDocs.length].id;
    p.service = serviceDcos[i % serviceDcos.length].id;
    p.discount = 95;
    p.discountPrice = 95;
    p.insuranceClaim = Math.random() * 100;
    p.costPerMile = Math.random() * 10;
    _.each(p.location, (a) => {
      a.state = 'state';
    });
  });

  logger.info(`creating ${drones.length} drones`);
  const droneDocs = [];
  for (let i = 0; i < drones.length; i++) {
    drones[i].provider = providerDocs[i % providerDocs.length].id;
    drones[i].pilots = [providerUserDocs[i % providerUserDocs.length]];
    drones[i].serialNumber = 'serialNumber' + i;
    droneDocs.push(yield Drone.create(drones[i]));
  }

  // update drones to add accessURL's
  for (let i = 0; i < droneDocs.length; i += 1) {
    const drone = droneDocs[i];
    drone.accessURL = `http://localhost:4040/${drone.id}`;
    yield drone.save();
  }

  logger.info(`creating ${packages.length} packages`);
  const packageDocs = yield Package.create(packages);

  _.each(missions, (m, i) => {
    m.package = packageDocs[i % packageDocs.length].id;
    m.provider = packageDocs[i % packageDocs.length].provider;
    if (!m.pilot) {
      m.pilot = userDocs[0].id; // setting all to first user for testing convinience
    }
    m.drone = droneDocs[i % droneDocs.length].id;
    m.status = _.values(MissionStatus)[Math.floor(Math.random() * _.values(MissionStatus).length)];
    m.telemetry = {
      lat: 6,
      long: 7,
      speed: 15,
      distance: 12,
    };
    m.eta = 2;
    m.frontCameraUrl = '/assets/front-camera.jpg';
    m.backCameraUrl = '/assets/back-camera.jpg';
    m.gallery = [{
      thumbnailUrl: 'http://google.com',
      videoUrl: 'http://google.com',
      imageUrl: `/assets/mission-gallery-image-0${i % 4 + 1}.jpg`,
    }];
    m.plannedHomePosition = waypoints.plannedHomePosition;
    m.missionItems = waypoints.missionItems;
  });
  logger.info(`creating ${missions.length} missions`);
  const missionDocs = yield Mission.create(
    _.map(missions, (mission, index) => _.assign(mission, {missionName: mission.missionName + ' ' + index}))
  );

  _.each(requests, (r, i) => {
    r.user = userDocs[0].id; // setting all to first user for testing convenience
    r.package = packageDocs[i % packageDocs.length].id;
    r.provider = packageDocs[i % packageDocs.length].provider;
  });

  logger.info(`creating ${requests.length} requests`);
  const requestDocs = yield PackageRequest.create(requests);
  for (let i = 0; i < requestDocs.length; i++) {
    const today = moment();
    today.hour('10');
    today.minute('9');
    today.second('7');
    today.millisecond('500');
    // put missions +-15 days from today
    // i + 10 is to make 'drone36' with id '583e6e9df4209e0aa5d94da6' to have missions in different months for testing
    today.add((i + 10) % 31 - 15, 'd');

    requestDocs[i].mission = missionDocs[i % missionDocs.length].id;
    requestDocs[i].status = _.values(RequestStatus)[i % _.values(RequestStatus).length];
    requestDocs[i].launchDate = new Date();
    yield requestDocs[i].save();

    const mindex = i % missionDocs.length;
    missionDocs[mindex].packageRequest = requestDocs[i].id;
    missionDocs[mindex].provider = requestDocs[i].provider;
    // add some missions to the pilot user
    if (i % (providerDocs.length + 1) !== providerDocs.length) {
      missionDocs[mindex].pilot = providerUserDocs[i % (providerDocs.length + 1)].id;
      // for provider user use same statuses as for providerRequests
      missionDocs[mindex].status = i % 2 ? MissionStatus.IN_PROGRESS : MissionStatus.COMPLETED;
    } else {
      missionDocs[mindex].pilot = userDocs[2].id; // pilot user
      // for pilot user use all possible statuses
      missionDocs[mindex].status = MissionStatus[_.keys(MissionStatus)[Math.floor(_.keys(MissionStatus).length * Math.random())]];
      if (missionDocs[mindex].status === MissionStatus.IN_PROGRESS) {
        // for in progress mission status checklist needs to be completed.
        missionDocs[mindex].status = MissionStatus.SCHEDULED;
      }
    }
    missionDocs[mindex].scheduledAt = today.add(1, 'h').toDate(); // +1 hour
    missionDocs[mindex].startedAt = today.add(2, 'h').toDate(); // +2 hours
    missionDocs[mindex].launchDate = today.toDate();
    if (missionDocs[mindex].status === MissionStatus.COMPLETED) {
      missionDocs[mindex].completedAt = today.add(3, 'h').toDate(); // +3 hours
    }
    missionDocs[mindex].whatToBeDelivered = requestDocs[i].whatToBeDelivered;
    missionDocs[mindex].weight = requestDocs[i].weight;
    yield missionDocs[mindex].save();
  }
  _.each(notifications, (n) => {
    n.user = userDocs[0].id; // setting all to first user for testing convenience
  });
  logger.info(`creating ${notifications.length} notifications`);
  yield Notification.create(notifications);

  _.each(positions, (p) => {
    p.droneId = droneDocs[0].id;
  });
  logger.info(`creating ${positions.length} dronePositions`);
  yield DronePosition.create(positions);

  logger.info(`creating ${noFlyZones.length} no fly zones`);
  yield NoFlyZone.create(noFlyZones);

  logger.info(`creating ${questions.length} questions`);
  const questionDocs = yield Question.create(questions);

  logger.info('data created successfully');
}).then(() => {
  logger.info('Done');
  process.exit();
}).catch((e) => {
  logger.error(e);
  process.exit();
});