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

attributes.test.js 2.54 KB
'use strict';

var chai = require('chai')
  , Sequelize = require('../../../index')
  , Promise = Sequelize.Promise
  , expect = chai.expect
  , Support = require(__dirname + '/../support');

describe(Support.getTestDialectTeaser('Model'), function() {
  describe('attributes', function() {
    describe('set', function() {
      it('should only be called once when used on a join model called with an association getter', function() {
        var self = this;
        self.callCount = 0;

        this.Student = this.sequelize.define('student', {
          no: {type: Sequelize.INTEGER, primaryKey: true},
          name: Sequelize.STRING
        }, {
          tableName: 'student',
          timestamps: false
        });

        this.Course = this.sequelize.define('course', {
          no: {type: Sequelize.INTEGER, primaryKey: true},
          name: Sequelize.STRING
        },{
          tableName: 'course',
          timestamps: false
        });

        this.Score = this.sequelize.define('score', {
          score: Sequelize.INTEGER,
          test_value: {
            type: Sequelize.INTEGER,
            set: function(v) {
              self.callCount++;
              this.setDataValue('test_value', v + 1);
            }
          }
        }, {
          tableName: 'score',
          timestamps: false
        });

        this.Student.hasMany(this.Course, {through: this.Score, foreignKey: 'StudentId'});
        this.Course.hasMany(this.Student, {through: this.Score, foreignKey: 'CourseId'});

        return this.sequelize.sync({force: true}).then(function() {
          return Promise.join(
            self.Student.create({no: 1, name: 'ryan'}),
            self.Course.create({no: 100, name: 'history'})
          ).spread(function(student, course) {
            return student.addCourse(course, {score: 98, test_value: 1000});
          }).then(function() {
            expect(self.callCount).to.equal(1);
            return self.Score.find({StudentId: 1, CourseId: 100}).then(function(score) {
              expect(score.test_value).to.equal(1001);
            });
          })
          .then(function() {
            return Promise.join(
              self.Student.build({no: 1}).getCourses({where: {no: 100}}),
              self.Score.find({StudentId: 1, CourseId: 100})
            );
          })
          .spread(function(courses, score) {
            expect(score.test_value).to.equal(1001);
            expect(courses[0].score.toJSON().test_value).to.equal(1001);
            expect(self.callCount).to.equal(1);
          });
        });
      });
    });
  });
});