不要怂,就是干,撸起袖子干!

find-or-create.test.js 1.56 KB
'use strict';

const chai = require('chai'),
  expect = chai.expect,
  Support = require(__dirname + '/../support'),
  current = Support.sequelize,
  cls = require('continuation-local-storage'),
  sinon = require('sinon'),
  stub = sinon.stub,
  Promise = require('bluebird');

describe(Support.getTestDialectTeaser('Model'), () => {

  describe('method findOrCreate', () => {

    before(() => {
      current.constructor.useCLS(cls.createNamespace('sequelize'));
    });

    after(() => {
      delete current.constructor._cls;
    });

    beforeEach(function() {
      this.User = current.define('User', {}, {
        name: 'John'
      });

      this.transactionStub = stub(this.User.sequelize, 'transaction');
      this.transactionStub.returns(new Promise(() => {}));

      this.clsStub = stub(current.constructor._cls, 'get');
      this.clsStub.returns({ id: 123 });
    });

    afterEach(function() {
      this.transactionStub.restore();
      this.clsStub.restore();
    });

    it('should use transaction from cls if available', function() {

      const options = {
        where: {
          name: 'John'
        }
      };

      this.User.findOrCreate(options);

      expect(this.clsStub.calledOnce).to.equal(true, 'expected to ask for transaction');
    });

    it('should not use transaction from cls if provided as argument', function() {

      const options = {
        where: {
          name: 'John'
        },
        transaction: { id: 123 }
      };

      this.User.findOrCreate(options);

      expect(this.clsStub.called).to.equal(false);
    });
  });
});