对Mongoose模型的虚拟属性进行拼接
有没有办法存根Mongoose模型的虚拟属性?
假设Problem
是一个模型类,difficulty
是一个虚拟属性。 delete Problem.prototype.difficulty
返回false,属性仍然存在,所以我不能用我想要的任何值替换它。
我也试过了
var p = new Problem();
delete p.difficulty;
p.difficulty = Problem.INT_EASY;
它没用。
将undefined分配给Problem.prototype.difficulty
或使用sinon.stub(Problem.prototype, 'difficulty').returns(Problem.INT_EASY);
会抛出异常“TypeError:无法读取未定义的属性'范围',同时执行
var p = new Problem();
sinon.stub(p, 'difficulty').returns(Problem.INT_EASY);
会抛出错误“TypeError:尝试将字符串属性难度包装为函数”。
我的想法已经不多了。帮帮我!谢谢!
回答如下:所有酒店的mongoose internally uses Object.defineProperty
。由于它们被定义为不可配置,你不能delete它们,你也不能re-configure它们。
但是,您可以做的是覆盖模型的get
和set
方法,这些方法用于获取和设置任何属性:
var p = new Problem();
p.get = function (path, type) {
if (path === 'difficulty') {
return Problem.INT_EASY;
}
return Problem.prototype.get.apply(this, arguments);
};
或者,使用sinon.js的完整示例:
var mongoose = require('mongoose');
var sinon = require('sinon');
var problemSchema = new mongoose.Schema({});
problemSchema.virtual('difficulty').get(function () {
return Problem.INT_HARD;
});
var Problem = mongoose.model('Problem', problemSchema);
Problem.INT_EASY = 1;
Problem.INT_HARD = 2;
var p = new Problem();
console.log(p.difficulty);
sinon.stub(p, 'get').withArgs('difficulty').returns(Problem.INT_EASY);
console.log(p.difficulty);