javascript - 定义助手链功能

我经常在授权下执行chai请求：

chai.request(baseUrl).get(`/resource`).set('authorization', `Bearer ${token}`)

我想在一个称为withAuth的函数中对自动生成进行分解，因此我可以更方便地重用它：

chai.request(baseUrl).get(`/resource`).withAuth()

但是我不确定该怎么做？
我试过了：

let withAuth = () => {set('authorization', `Bearer ${token}`)}

但这是行不通的。

最佳答案

使用Assertion原型的addChainableMethod：

chai.Assertion.addChainableMethod('withAuth',(request) => request.set('authorization', `Bearer ${token}`));