你如何在Node.js / Express中模拟MySQL(带node-orm2)?

时间:2013-09-05 02:33:37

标签: mysql node.js unit-testing express mocking

我使用带有https://github.com/dresende/node-orm2的node.js / express来使用我的MySQL数据库。

我是node.js世界的新手,到目前为止我很困难,我不知道如何对一个简单的函数进行单元测试(不是集成测试)。

这是我的server.js,加载我的用户模型(ORM)

var express = require('express'),
    orm = require('orm'),
    config = require('./config/config.js'),
    auth = require('./services/authentication'),
    helper = require('./middlewares/helper.js'), 
    friends = require('./modules/friends.js');

var app = express();

app.use(orm.express('mysql://' + config.mysql.username + ':' + config.mysql.pwd + '@' + config.mysql.host + ':' + config.mysql.port + '/' + config.mysql.db, {
    define: function(db, models, next) {
        db.load("./models/models.js", function(err) { // loaded!
            models.user = db.models.user;
        });
    }
}));

var middlewares = [auth.authenticate, helper.retrieveUser];

app.get('/friends', middlewares, friends.findActiveFriends);

app.listen(3000);
console.log('Listening on port 3000...');

这是用户模型:

module.exports = function (db, cb) {
    var User = db.define('user', {
        uid               : { type: 'number', rational: false, unique: true, required: true },
        first_name        : { type: 'text', size: 100, required: true },
        last_name         : { type: 'text', size: 100, required: true },
        picture           : { type: 'text', size: 255, required: false },
        email             : { type: 'text', size: 255, required: true }, 
        creation_date     : { type: 'date', time: true },
        modification_date : { type: 'date', time: true }
    }, {
        methods: {
            fullName: function () {
                return this.first_name + ' ' + this.last_name;
            }
        },
        hooks: {
            beforeCreate: function (next) { 
                if (this.creation_date == undefined) {
                    this.creation_date = new Date();
                }
                if (this.modification_date == undefined) {
                    this.modification_date = new Date();
                }
                return next();
            } 
        }
    });

    // CUSTOM FUNCTIONS
    User.getByUid = function(uid, callback) {
        this.find({ uid: uid }, function(err, users) {
            if(err) callback(err);
            if (users.length == 1) { 
                callback(null, users[0]); 
            } else {
               callback('No user found with uid=' + uid);
            }
        });
    }; 

    User.hasMany("friends", User, {
        status: { type: 'enum', values: ['pending', 'refused', 'active'] }
    }, {
        reverse: 'friendsrev', mergeId: 'user_id', mergeAssocId: 'friend_id'
    });

    return cb();
};

这是我在friends.js中找到活跃朋友的方法:

var _findActiveFriends = function(req, res) {
    req.currentUser.getFriends({
        status: 'active'
    }, function(err, friends) {
        if (err) throw err;
        res.send(JSON.stringify(friends));
    });
};

我想知道如何通过模拟数据库连接和请求来编写一个简单的测试(使用mocha和sinon.js?)。我需要模拟req.currentUser的值,这是ORM在中间件中返回的用户。

我只想运行单元测试,不要使用真正的数据库或进行一些HTTP调用。

感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

如果您想使用sinon.js嘲笑请求,您可以执行以下操作。

var sinon = require('sinon');
var friend = require('./friend');

it('some test', function(done) {
  var req = {
    currentUser: {
      // Add all the properties/functions that you are concerned with here. 
      // and you can/should wrap them around sinon.spy().
      // If you are not concerned with that function (i.e. you are not using it) 
      // then you can simply use sinon.stub() to return a stub function. 
    }
  };
  var res = {
    send: sinon.spy(function(obj) { 
      assert.ok(obj); // yes object exists 
      done(); // end of test
    };
  };
  var next = function() {};
  friend.findActiveFriend(req, res, next);
});

这样您就不应该连接到只测试friend.js的模型。

此外,由于我刚刚注意到您正在使用orm.express,您可能还想简单地使用您想要的存根函数模拟req.models,如上所述。