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，您就可以控制传递给客户端创建的流的内容。

单元测试使用模拟

1 个答案: