使用模拟单元测试使用请求、管道和流的私有方法

Question

我想在下面的代码中对导出的方法进行单元测试。我想模拟私有方法中的值来控制返回的 Promise 的拒绝/解析。 client 是 node-postgres 已经连接到数据库的对象。

我知道我可以使用 proxyquire 来存根出所需的库，但我如何模拟链接方法 .on('error', ...)、.pipe(stream) 和 .on('end', ...) 以便我可以控制返回的值。

注意显示的导出方法是对真实方法的简化，导出 importDomain 是不可行的。

const copyFrom = require('pg-copy-streams').from
const request = require('request')
const Promise = require('bluebird')

// private
function importDomain (client, domain) {
    return new Promise((resolve, reject) => {
    let stream = client.query(copyFrom(`COPY ${domain.table} FROM STDIN;`))

    let req = request(`${domain.url}`)
    req.on('error', reject)
    req.pipe(stream)
       .on('error', reject)
       .on('end', resolve)
  })
}

// public
module.exports = (client) => {
  let domain = someFunctionReturningDomain()
  importDomain(client, domain)
}

Answer 1

为了对函数 importDomain 进行单元测试，您需要模拟请求。它是您的模块所必需的，因此应视为经过测试并正常工作。

由于您将客户端作为参数提供。它的查询方法和返回的流已经完全不受模块控制。

只需替换 req.pipe，您就可以控制传递给客户端创建的流的内容。