attributes.test.js
3.31 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
'use strict';
const chai = require('chai'),
Sequelize = require('../../../index'),
Promise = Sequelize.Promise,
expect = chai.expect,
Support = require(__dirname + '/../support');
describe(Support.getTestDialectTeaser('Model'), () => {
describe('attributes', () => {
describe('set', () => {
it('should only be called once when used on a join model called with an association getter', function() {
const 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(v) {
self.callCount++;
this.setDataValue('test_value', v + 1);
}
}
}, {
tableName: 'score',
timestamps: false
});
this.Student.belongsToMany(this.Course, {through: this.Score, foreignKey: 'StudentId'});
this.Course.belongsToMany(this.Student, {through: this.Score, foreignKey: 'CourseId'});
return this.sequelize.sync({force: true}).then(() => {
return Promise.join(
self.Student.create({no: 1, name: 'ryan'}),
self.Course.create({no: 100, name: 'history'})
).spread((student, course) => {
return student.addCourse(course, { through: {score: 98, test_value: 1000}});
}).then(() => {
expect(self.callCount).to.equal(1);
return self.Score.find({ where: { StudentId: 1, CourseId: 100 } }).then(score => {
expect(score.test_value).to.equal(1001);
});
})
.then(() => {
return Promise.join(
self.Student.build({no: 1}).getCourses({where: {no: 100}}),
self.Score.find({ where: { StudentId: 1, CourseId: 100 } })
);
})
.spread((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);
});
});
});
it('allows for an attribute to be called "toString"', function () {
const Person = this.sequelize.define('person', {
name: Sequelize.STRING,
nick: Sequelize.STRING
}, {
timestamps: false
});
return this.sequelize.sync({force: true})
.then(() => Person.create({name: 'Jozef', nick: 'Joe'}))
.then(() => Person.findOne({
attributes: [
'nick',
['name', 'toString']
],
where: {
name: 'Jozef'
}
}))
.then(person => {
expect(person.dataValues['toString']).to.equal('Jozef');
expect(person.get('toString')).to.equal('Jozef');
});
});
});
});
});