如何为 Node.js 命令行工具添加单元测试 - 51Testing软件测试论坛

　// 来源于文档
　　import {expect, test} from '@oclif/test'
　　describe('auth:whoami', () => {
　　 test
　　 .nock('https://api.heroku.com', api => api
　　 .get('/account')
　　 // user is logged in, return their name
　　 .reply(200, {email: 'jeff@example.com'})
　　 )
　　 .stdout()
　　 .command(['auth:whoami'])
　　 .it('shows user email when logged in', ctx => {
　　 expect(ctx.stdout).to.equal('jeff@example.com\n')
　　 })
　　})

复制代码

这个例子使用 nock(nock，一个用来模拟 http 请求的包)模拟发送一个 http 请求，然后 mock 标准输出stdout，再执行真正的命令 auth:whoami，最后从ctx中获取stdout进行断言。这里引入的test 和 expect是封装的 oclif/fancy-test，而它又是基于 Mocha，简言之就是一个能更少写 setup/teardown 的链式调用单测库，expect 使用的是 Chai 语法。
　　看到如此简单的示例，我不禁满头问号，其主要依赖的 mock 方式是直接代理 http 请求，这固然是非常符合直觉的，因为每一个命令中确实都需要发送 http 请求，然而发送 http 请求有时可能也是一件山路十八弯的事，简要列举一些问题：
　　·直接 mock http 请求要求给出详细的路径，比如https://exmaple.com/api/User和get、post等方法，然后模拟返回值，非常死板。
　　· 对于封装了多层的 api 调用（例如 api 调用可能需要经过各种签名，还可能是通过 sdk 调用，最终暴露出来的已经是不知道转发了多少层的小小的接口），如果直接这样写单测的话，会很难找到具体需要调用哪个接口。且各个请求的路径可能完全一样，操作通过 Body / Header 中的某些字段区分，这就导致很难精准 mock。
　　· 无法做到连续 mock 多个 http 调用，这意味着下面的情况：

test
　　 .nock("https://example.com", api => api
　　 .post('/api/User')
　　 //...
　　 )
　　// 这里不能再接`.nock`了

复制代码

当然，它也并非一无是处。使用.stdout()来 mock 输出还是非常方便的。另外仔细地翻看各个仓库文档，可以发现它可以通过插桩代理支持 mock 用户输入（stdin）的，例如使用 cli-ux获取输入时，可以这么写：

test
　　 .stub(cli, "prompt", () => async () => mockUserName)
　　.stub(cli, "prompt", () => async () => mockPassword)
　　 .stdout()
　　 .command(["auth:login"])
　　 .it("should login successfully", (ctx) => {
　　 expect(ctx.stdout).toContain(`login successfully`);
　　 });

复制代码

简单来说，它缺失了最重要的能力：函数或模块级别的 mock ，而非单纯代理 http 请求。
　　jest
　　Jest 是一个非常流行的测试框架，并且它提供了优秀的函数&模块级别的 mock 能力，这恰好就是我们所需要的。利用它的代理函数和模块能力，可以像这样来模拟 api 的调用：
　　// 这也是那常见的一长串 {__esModule: true, ...originalModule, ...} 的简洁写法。

　jest.mock("../path/to/api", () => ({
　　 ...jest.requireActual("../path/to/api"),
　　 functionNeedToMock: jest.fn().mockResolvedValue({
　　 MockKey: 'mockValue'
　　 }),
　　 anotherFunctionNeedToMock: jest.fn().mockResolvedValue(true),
　　}));

复制代码

上面的这段代码就可以代理../path/to/api这个模块中的functionNeedToMock和anotherFunctionNeedToMock两个函数，而不修改其它的函数。
　　另外一个问题是如果需要代理的是位于Command 类之内的函数，例如在下面这个GetShoppingCartStatus类中：

// 修改自文档
　　import Command from '@oclif/command'
　　export class GetShoppingCartStatus extends Command {
　　 // 需要代理的函数
　　 async function checkLoginStatus() {/* ... */}
　　
　　 async run() {
　　 try {
　　 await checkLoginStatus()
　　 /* some other code */
　　 } catch (err) {
　　 if (err.statusCode === 401) {
　　 this.error('not logged in', {exit: 100})
　　 }
　　 throw err
　　 }
　　 }
　　}

复制代码

我们需要代理的是checkLoginStatus这个函数，那么我们就需要使用spyOn来“监听”GetShoppingCartStatus这个类的原型，示例：

const checkLoginStatus = jest
　　 .spyOn(GetShoppingCartStatus.prototype, "checkLoginStatus")
　　 .mockImplementation(() =>
　　 {/* ... */}
　　 );

复制代码

spyOn是一个非常强大的功能，它在 CLI 工具的单测中有一个更为重要的作用，那就是“监听”stdout。可以像这样来获取stdout的输出结果：

　let stdout;
　　beforeEach(() => {
　　 stdout = [];
　　 jest
　　 .spyOn(process.stdout, "write")
　　 .mockImplementation((val) => stdout.push(val));
　　});
　　afterEach(() => jest.restoreAllMocks());

复制代码

当需要断言的时候，就使用stdout就可以了。
　　总结
　　现在，我们已经明白了两种框架的优劣：
　　·@oclif/test直接代理 http 请求的方式有很大的局限性，但它链式的调用方式使得模拟用户输入和监听输出很方便；
　　· jest能够进行函数&模块级别的 mock，但用它捕获 stdout、模拟处理用户输入却显得繁琐。
　　各取长处，对于这样一个命令：

// 假设其命令为 run:upload
　　import Command from '@oclif/command'
　　import { cli } from 'cli-ux';
　　import { zipDir } from '../utils'
　　export class UploadFilesCommand extends Command {
　　 // 需要代理的函数
　　 async function checkLoginStatus() {/* ... */}
　　 // 用户输入
　　 static flags = {
　　 path: flags.string({ char: 'p', description: '文件路径'})
　　 }
　　 async run() {
　　 try {
　　 await checkLoginStatus()
　　 await cli.prompt('请输入文件路径', { required: true })
　　 await zipDir(/* ... */)
　　 /* some other code */
　　 } catch (err) {
　　 if (err.statusCode === 401) {
　　 this.error('not logged in', {exit: 100})
　　 }
　　 throw err
　　 }
　　 }
　　}

复制代码

我们可以这样利用这两个框架：

　// uploadFiles.test.js
　　import cli from "cli-ux";
　　import { test } from "@oclif/test";
　　// 代理 zipDir 函数
　　jest.mock("../path/to/utils", () => {
　　 return {
　　 zipDir: jest.fn().mockResolvedValue(true),
　　 };
　　});
　　// 代理 checkLoginStatus 函数
　　const checkLoginStatus = jest
　　 .spyOn(UploadFilesCommand.prototype, "checkLoginStatus")
　　 .mockImplementation(() => {/* ... */});
　　// 测试
　　describe('run:upload', () => {
　　 test
　　 .stub(cli,'prompt', () => async () => 'a/b/c') // 对 cli 插桩，由于是链式调用，如果有多个用户输入，可以插桩多次
　　 .stdout()
　　 .command(['run:upload'])
　　 .it('should upload files successfully', (ctx) => {
　　 expect(ctx.stdout).toEqual(/* or other assets */);
　　 });
　　});

复制代码

在这段代码中，我们既利用了 jest 模拟函数和模块的功能，也使用了 @oclif/test 简洁地处理输入输出，融合了两种框架的优点，我们就可以顺畅地进行单测编写了。