Enforce id (prevent user-set value), fix isNewRecord

This commit is contained in:
Fabien Franzen 2014-09-05 16:35:01 +02:00
parent aea2f64181
commit aebf5e9e6b
4 changed files with 28 additions and 7 deletions

View File

@ -137,14 +137,16 @@ DataAccessObject.create = function (data, callback) {
}
var obj;
var idValue = getIdValue(this, data);
// if we come from save
if (data instanceof Model && !getIdValue(this, data)) {
if (data instanceof Model && !idValue) {
obj = data;
} else {
obj = new Model(data);
}
data = obj.toObject(true);
// validation required
obj.isValid(function (valid) {
if (valid) {
@ -315,7 +317,7 @@ DataAccessObject.findById = function find(id, cb) {
if (!getIdValue(this, data)) {
setIdValue(this, data, id);
}
obj = new this(data, {applySetters: false});
obj = new this(data, {applySetters: false, persisted: true});
}
cb(err, obj);
}.bind(this));
@ -732,7 +734,7 @@ DataAccessObject.find = function find(query, cb) {
this.getDataSource().connector.all(this.modelName, query, function (err, data) {
if (data && data.forEach) {
data.forEach(function (d, i) {
var obj = new self(d, {fields: query.fields, applySetters: false});
var obj = new self(d, {fields: query.fields, applySetters: false, persisted: true});
if (query && query.include) {
if (query.collect) {
@ -960,6 +962,7 @@ DataAccessObject.prototype.save = function (options, callback) {
return callback(err, inst);
}
inst._initProperties(data);
inst.__persisted = true;
updateDone.call(inst, function () {
saveDone.call(inst, function () {
callback(err, inst);
@ -1027,7 +1030,7 @@ DataAccessObject.updateAll = function (where, data, cb) {
};
DataAccessObject.prototype.isNewRecord = function () {
return !getIdValue(this.constructor, this);
return !this.__persisted;
};
/**
@ -1168,6 +1171,7 @@ DataAccessObject.prototype.updateAttributes = function updateAttributes(data, cb
inst._adapter().updateAttributes(model, getIdValue(inst.constructor, inst),
inst.constructor._forDB(typedData), function (err) {
if (!err) inst.__persisted = true;
done.call(inst, function () {
saveDone.call(inst, function () {
if(cb) cb(err, inst);

View File

@ -489,6 +489,9 @@ DataSource.prototype.setupDataAccess = function (modelClass, settings) {
var idType = this.connector.getDefaultIdType() || String;
idProp.type = idType;
modelClass.definition.properties[idName].type = idType;
if (settings.forceId) {
modelClass.validatesAbsenceOf(idName, { if: 'isNewRecord' });
}
}
if (this.connector.define) {
// pass control to connector

View File

@ -47,6 +47,7 @@ function ModelBaseClass(data, options) {
* @param {Object} options An object to control the instantiation
* @property {Boolean} applySetters Controls if the setters will be applied
* @property {Boolean} strict Set the instance level strict mode
* @property {Boolean} persisted Whether the instance has been persisted
* @private
*/
ModelBaseClass.prototype._initProperties = function (data, options) {
@ -67,7 +68,9 @@ ModelBaseClass.prototype._initProperties = function (data, options) {
if(strict === undefined) {
strict = ctor.definition.settings.strict;
}
this.__persisted = options.persisted === true;
if (ctor.hideInternalProperties) {
// Object.defineProperty() is expensive. We only try to make the internal
// properties hidden (non-enumerable) if the model class has the

View File

@ -16,7 +16,7 @@ describe('manipulation', function () {
age: {type: Number, index: true},
dob: Date,
createdAt: {type: Number, default: Date.now}
});
}, { forceId: true });
db.automigrate(done);
@ -50,6 +50,17 @@ describe('manipulation', function () {
person.should.be.an.instanceOf(Person);
should.not.exist(person.id);
});
it('should not allow user-defined value for the id of object', function (done) {
Person.create({ id: 123456 }, function (err, p) {
err.should.be.instanceof(ValidationError);
err.message.should.equal('The `Person` instance is not valid. Details: `id` can\'t be set.');
err.statusCode.should.equal(422);
p.should.be.instanceof(Person);
p.id.should.equal(123456);
done();
});
});
it('should work when called without callback', function (done) {
Person.afterCreate = function (next) {