javascript - 在运行 Mocha 测试之前等待路由和数据库初始化

Question

我有一个 NodeJS 应用程序，它连接到 Mongo 数据库，然后在建立连接后初始化所有路由。我想使用服务器和 supertest 来添加一些 Mocha 测试来测试请求。但是，要从 test.js 文件检索服务器，我需要 require() 主文件并从中获取服务器变量。然而，当我尝试使用它时，它提示该变量未定义，因为数据库尚未加载并且路由尚未初始化。

main.js(摘录)

// This is a custom mongo module that connects using my db string
// It stores the DB as a variable in another file, which is later retrieved by the route handlers using mongo.GetDB()

mongo.ConnectToDB(function(err, dbo){
    if(err) throw err;

    require("./api")(app);
    require("./admin")(app);
    require("./routes")(app);
});

// This obviously doesn't work, as it's outside of the callback
module.exports.server = app.listen(4500);

server.test.js(摘录)

describe("loading express", (done) => {
    let server;
    beforeEach(() => {
        // The server is returned, but none of the routes are included
        server = require("../main").server;
    });

    afterEach(() => {
        server.close();
    });

    // ...
});

在检索服务器变量并继续Mocha测试之前，有什么方法可以等待数据库和路由初始化吗？

Answer 1

让main.js导出一个Promise，例如:

module.exports.serverPromise = new Promise((resolve, reject) => {
  mongo.ConnectToDB(function(err, dbo){
    if(err) reject(err);
    require("./api")(app);
    require("./admin")(app);
    require("./routes")(app);

    resolve(app.listen(4500));
  });
});

然后，导入后，在Promise上调用.then后就可以使用了