water-orm/test/common/interfaces/associations/hasMany/add.js

A monolith version of Standalone waterline ORM

var assert = require('assert'),
    _ = require('lodash');

describe('Association Interface', function() {

  describe('Has Many Association', function() {
    describe('association .add()', function() {

      describe('with an object', function() {

        /////////////////////////////////////////////////////
        // TEST SETUP
        ////////////////////////////////////////////////////

        var customer;

        before(function(done) {
          Associations.Customer.create({ name: 'hasMany add' }, function(err, model) {
            if(err) return done(err);

            customer = model;
            Associations.Payment.create({ amount: 1, a_customer: customer.id }, done);
          });
        });

        /////////////////////////////////////////////////////
        // TEST METHODS
        ////////////////////////////////////////////////////

        it('should create a new payment association', function(done) {
          customer.payments.add({ amount: 1337 });
          customer.save(function(err) {
            assert.ifError(err);

            // Look up the customer again to be sure the payment was added
            Associations.Customer.findOne(customer.id)
            .populate('payments')
            .exec(function(err, model) {
              assert.ifError(err);

              assert.strictEqual(model.payments.length, 2);
              assert.strictEqual(model.payments[1].amount, 1337);
              done();
            });
          });
        });
      });

      describe('with an id', function() {

        /////////////////////////////////////////////////////
        // TEST SETUP
        ////////////////////////////////////////////////////

        var customer, payment;

        before(function(done) {

          var records = [
            { name: 'hasMany add 1' },
            { name: 'hasMany add 2' }
          ];

          Associations.Customer.createEach(records, function(err, models) {
            if(err) return done(err);

            customer = models[0];
            Associations.Payment.create({ amount: 1, a_customer: models[1].id }, function(err, paymentModel) {
              if(err) return done(err);

              payment = paymentModel;
              done();
            });
          });
        });

        /////////////////////////////////////////////////////
        // TEST METHODS
        ////////////////////////////////////////////////////

        it('should link the payment to another association', function(done) {
          customer.payments.add(payment.id);
          customer.save(function(err) {
            assert.ifError(err);

            // Look up the customer again to be sure the payment was added
            Associations.Customer.findOne(customer.id)
            .populate('payments')
            .exec(function(err, data) {
              assert.ifError(err);

              assert.strictEqual(data.payments.length, 1);
              assert.strictEqual(data.payments[0].amount, 1);
              done();
            });
          });
        });
      });

    });
  });
});