allow for additional transform functions for toJSON and sanitize

This commit is contained in:
Matteo Pagliazzi
2015-11-25 15:42:29 +01:00
parent 94fc1c9bef
commit d67836ee1f
3 changed files with 47 additions and 16 deletions

View File

@@ -52,8 +52,38 @@ describe('Base model plugin', () => {
let objToTransform = {ok: true, amPrivate: true};
let privatized = schema.options.toJSON.transform({}, objToTransform);
expect(objToTransform).to.have.property('ok');
expect(objToTransform).not.to.have.property('amPrivate');
expect(objToTransform.amPrivate).to.equal(undefined);
expect(privatized).to.have.property('ok');
expect(privatized).not.to.have.property('amPrivate');
});
it('accepts a further transform function for toJSON', () => {
let options = {
private: ['amPrivate'],
toJSONTransform: sandbox.stub().returns(true)
};
baseModel(schema, options);
let objToTransform = {ok: true, amPrivate: true};
let privatized = schema.options.toJSON.transform({}, objToTransform);
expect(privatized).to.equals(true);
expect(options.toJSONTransform).to.be.calledWith(objToTransform);
});
it('accepts a transform function for sanitize', () => {
let options = {
private: ['amPrivate'],
sanitizeTransform: sandbox.stub().returns(true)
};
baseModel(schema, options);
expect(schema.options.toJSON.transform).to.exist;
let objToSanitize = {ok: true, noUpdateForMe: true};
let sanitized = schema.statics.sanitize(objToSanitize);
expect(sanitized).to.equals(true);
expect(options.sanitizeTransform).to.be.calledWith(objToSanitize);
});
});