Loopback sample mocha unit test

sample.test.js

'use strict';

const loopback = require('loopback');
const expect  = require('chai').expect;
const request = require('supertest');


describe('Test relations update support', function() {

  // Create a new loopback app.
  const app = loopback()

  // Register the mixin
  app.loopback.modelBuilder.mixins.define('IncludeOnUpdate', require('../include-on-update'))
  app.loopback.modelBuilder.mixins.define('HasAndBelongsToManyUpdate', require('../hasAndBelongsToMany-update'))

  const City = loopback.PersistedModel.extend('City', {
    name: String,
  }, {
    plural: 'Cities',
    mixins: {
      IncludeOnUpdate: {},
    },
    relations: {
      country: {
        type: 'belongsTo',
        model: 'Country',
        foreignKey: 'countryId',
      },
    },
  })

  const Country = loopback.PersistedModel.extend('Country', {
    name: String,
  }, { plural: 'Countries'})

  const Person = loopback.PersistedModel.extend('Person', {
    name: String,
  }, {
    plural: 'People',
    mixins: {
      HasAndBelongsToManyUpdate: {},
      IncludeOnUpdate: {},
    },
    relations: {
      liveIn: {
        type: 'hasAndBelongsToMany',
        model: 'City'
      }
    }
  })


  app.model(City)
  app.model(Country)
  app.model(Person)

  const dbConnector = loopback.memory();

  City.attachTo(dbConnector);
  Country.attachTo(dbConnector);
  Person.attachTo(dbConnector);

  app.use('/api', loopback.rest());

  beforeEach(function clearMemoryConnector(done){
    dbConnector.automigrate(done);
  });

  it('create belongsTo', function() {
    return app.models.Country.create({name: 'France'})
    .then(country => {
      return request(app).post('/api/Cities')
        .send({ name: 'Paris', country: country })
        .expect(200)
        .set('Accept', 'application/json')
        .then(response => {
          expect(response.body.country).to.be.a('object');
          expect(response.body.id).to.equal(1);
        });
    });
  });

  it('update belongsTo', function() {
    return app.models.Country.create({name: 'Belgium'})
    .then(country => {
      return request(app).put('/api/Cities')
        .send({ name: 'Paris', country: country })
        .expect(200)
        .set('Accept', 'application/json')
        .then(response => {
          expect(response.body.country).to.be.a('object');
          expect(response.body.country).to.have.property('name').to.be.equal('Belgium');
          expect(response.body.id).to.equal(1);
        })
        .then(() => City.findById(1, {include: 'country'}))
        .then(city => {
          city = city.toJSON();
          expect(city.country).to.be.a('object');
          expect(city.country).to.have.property('name').to.be.equal('Belgium');
        })
    });
  });

  it('create hasAndBelongsToMany', function() {
    return app.models.City.create([{name: 'Paris'}, {name: 'New York'}])
    .then(cities => {
      return request(app).post('/api/People')
        .send({ name: 'Jean-Claude', liveIn: cities })
        .expect(200)
        .set('Accept', 'application/json')
        .then(response => {
          expect(response.body.liveIn).to.be.a('array');
          expect(response.body.liveIn).to.have.lengthOf(2);
          expect(response.body.id).to.equal(1);
        })
        .then(() => app.models.Person.findOne().then(person => person.liveIn.find()))
        .then(liveIn => {
          expect(liveIn).to.be.a('array');
          expect(liveIn).to.have.lengthOf(2);
        });
    });
  });

  it('update hasAndBelongsToMany', function() {
    return app.models.City.create([{name: 'Paris'}, {name: 'New York'}])
      .then(() => app.models.Person.create({ name: 'Yve' }))
      .then(person => {
        person = person.toJSON();
        person.liveIn = [{id: 1, name: 'Paris'}];
        return request(app).put('/api/People')
          .send(person)
          .expect(200)
          .set('Accept', 'application/json')
          .then(response => {
            expect(response.body.liveIn).to.be.a('array');
            expect(response.body.liveIn).to.have.lengthOf(1);
            expect(response.body.id).to.equal(1);
          })
          .then(() => app.models.Person.findOne().then(person => person.liveIn.find()))
          .then(liveInCountries => {
            expect(liveInCountries).to.be.a('array');
            expect(liveInCountries).to.have.lengthOf(1);
          });
      });
  });
});