8

我想在下面的代码中对导出的方法进行单元测试。我想模拟私有方法中的值来控制返回的 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)
}
4

1 回答 1

1

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

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

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

于 2017-03-09T08:04:36.033 回答