rested/test/fixtures/controller.js

{REST}ed; {REST}ful Enterprise Data-as-a-Service (DaaS)

require('mongodb');
const mongoose = require('mongoose');
const express = require('express');
const rested = require('../..');

let app;
let server;
const Schema = mongoose.Schema;

mongoose.set('useCreateIndex', true);
mongoose.set('useNewUrlParser', true);
mongoose.set('useUnifiedTopology', true);
mongoose.set('debug', (process.env.DEBUG || '').match(/mongoose/));

const Stores = new Schema({
  name: { type: String, required: true, unique: true },
  mercoledi: Boolean,
  voltaic: { type: Boolean, default: true },
  'hyphenated-field-name': { type: Boolean, default: true }
});

const Cheese = new Schema({
  name: { type: String, required: true, unique: true },
  color: { type: String, required: true, select: false },
  bother: { type: Number, required: true, default: 5 },
  molds: [String],
  life: { type: Number, default: 42 },
  arbitrary: [{
    goat: Boolean,
    champagne: String,
    llama: [Number]
  }]
});

const Beans = new Schema({ koji: Boolean });
const Deans = new Schema({ room: { type: Number, unique: true } });
const Liens = new Schema({ title: { type: String, default: 'Babrius' } });
const Fiends = new Schema({ average: Number });
const Unmades = new Schema({ mode: Number });

mongoose.model('store', Stores);
mongoose.model('cheese', Cheese);
mongoose.model('bean', Beans);
mongoose.model('dean', Deans);
mongoose.model('lien', Liens);
mongoose.model('fiend', Fiends);
mongoose.model('unmade', Unmades);
mongoose.model('timeentry', Cheese, 'cheeses').plural('timeentries');
mongoose.model('mean', Fiends, 'fiends').locking(true);
mongoose.model('bal', Stores, 'stores').plural('baloo');

module.exports = {
  app: () => app,
  server: () => server,
  deinit: done => Promise.all([server.close(), mongoose.disconnect()]).then(done),
  init: done => {
    mongoose.connect(global.__MONGO_URI__);

    // Stores controller
    let stores = rested.rest('store').findBy('name').select('-hyphenated-field-name -voltaic');

    stores.use('/binfo', (request, response, next) => response.json('Poncho!'));

    stores.use((request, response, next) => {
      response.set('X-Poncho', 'Poncho!');
      next();
    });

    stores.get('/info', (request, response, next) => {
      response.json('OK!');
    });

    stores.get('/:id/arbitrary', (request, response, next) => {
      response.json(request.params.id);
    });

    let cheesy = rested.rest('cheese').select('-_id color name').findBy('name');
    cheesy.operators('$push', 'molds arbitrary arbitrary.$.llama');
    cheesy.operators('$set', 'molds arbitrary.$.champagne');
    cheesy.operators('$pull', 'molds arbitrary.$.llama');

    rested.rest('timeentry').findBy('name').select('color');
    rested.rest('bean').methods('get', false);
    rested.rest('dean').findBy('room').methods('get', false);
    rested.rest('lien').select('-title').methods('delete', false);
    rested.rest('mean');
    rested.rest('bal').findBy('name');
    rested.rest('bal').fragment('linseed.oil');

    app = express();
    app.use('/api', rested());

    rested.rest('cheese').fragment('geese').handleErrors(false);

    app.use('/api-no-error-handler', rested());

    server = app.listen(done);
  },
  create: done =>
    // clear all first
    mongoose.model('store').deleteMany({}, error =>
      error ? done(error) : mongoose.model('cheese').deleteMany({}, error =>
        error ? done(error) : mongoose.model('store')
          .create(['Westlake', 'Corner'].map(name => ({ name: name })), error =>
            error ? done(error) : mongoose.model('lien').create({ title: 'Heraclitus' }, error => {
              if (error) return done(error);
              let cheeses = [
                { name: 'Cheddar', color: 'Yellow' },
                { name: 'Huntsman', color: 'Yellow, Blue, White' },
                {
                  name: 'Camembert', color: 'White', arbitrary: [
                    { goat: true, llama: [3, 4] },
                    { goat: false, llama: [1, 2] }
                  ]
                }
              ];
              mongoose.model('cheese').create(cheeses, done);
            })
          )))
};