expressCart/test/helper.js

const fs = require('fs');
const _ = require('lodash');
const moment = require('moment');
const supertest = require('supertest');
const app = require('../app.js');
const { newId } = require('../lib/common');
const { runIndexing } = require('../lib/indexing');

// Get test data to compare in tests
const rawTestData = fs.readFileSync('./bin/testdata.json', 'utf-8');
const jsonData = JSON.parse(rawTestData);

// Setup some global DB objects for comparison
const g = {
    db: {},
    config: {},
    products: {},
    discounts: {},
    customers: {},
    users: {},
    request: null,
    jsonData
};

const setup = (db) => {
    return Promise.all([
        db.cart.deleteMany({}, {}),
        db.users.deleteMany({}, {}),
        db.customers.deleteMany({}, {}),
        db.products.deleteMany({}, {}),
        db.discounts.deleteMany({}, {}),
        db.orders.deleteMany({}, {}),
        db.sessions.deleteMany({}, {})
    ])
    .then(() => {
        return Promise.all([
            db.users.insertMany(addApiKey(jsonData.users)),
            db.customers.insertMany(jsonData.customers),
            db.products.insertMany(fixProductDates(jsonData.products)),
            db.discounts.insertMany(fixDiscountDates(jsonData.discounts))
        ]);
    });
};

const runBefore = async () => {
    // Create a session
    g.request = supertest.agent(app);
    await new Promise(resolve => {
        app.on('appStarted', async () => {
            // Set some stuff now we have the app started
            g.config = app.config;
            g.db = app.db;

            await setup(g.db);

            // Get some data from DB to use in compares
            g.products = await g.db.products.find({}).toArray();
            g.customers = await g.db.customers.find({}).toArray();
            g.discounts = await g.db.discounts.find({}).toArray();
            g.users = await g.db.users.find({}).toArray();

            // Insert orders using product ID's
            _(jsonData.orders).each(async (order) => {
                order.orderProducts.push({
                    productId: g.products[0]._id,
                    title: g.products[0].productTitle,
                    quantity: 1,
                    totalItemPrice: g.products[0].productPrice,
                    options: {
                        size: '7.5'
                    },
                    productImage: g.products[0].productImage,
                    productComment: null
                });
                order.orderDate = new Date();
                await g.db.orders.insertOne(order);
            });

            // Index everything
            await runIndexing(app);

            resolve();
        });
    });
};

const fixProductDates = (products) => {
    let index = 0;
    products.forEach(() => {
        products[index].productAddedDate = new Date();
        index++;
    });
    return products;
};

const fixDiscountDates = (discounts) => {
    let index = 0;
    discounts.forEach(() => {
        let startDate = moment().subtract(1, 'days').toDate();
        let endDate = moment().add(7, 'days').toDate();
        const expiredStart = moment().subtract(14, 'days').toDate();
        const expiredEnd = moment().subtract(7, 'days').toDate();
        const futureStart = moment().add(7, 'days').toDate();
        const futureEnd = moment().add(14, 'days').toDate();

        // If code is expired, make sure the dates are correct
        if(discounts[index].code.substring(0, 7) === 'expired'){
            startDate = expiredStart;
            endDate = expiredEnd;
        }

        // If code is future, make sure the dates are correct
        if(discounts[index].code.substring(0, 6) === 'future'){
            startDate = futureStart;
            endDate = futureEnd;
        }

        // Set the expiry dates
        discounts[index].start = startDate;
        discounts[index].end = endDate;
        index++;
    });
    return discounts;
};

const addApiKey = (users) => {
    let index = 0;
    users.forEach(() => {
        users[index].apiKey = newId();
        index++;
    });
    return users;
};

module.exports = {
    runBefore,
    setup,
    g,
    fixProductDates,
    fixDiscountDates
};