Question

我陷入了一次测试中。在我的测试中，我需要检测客户端（浏览器）已取消或中止图像上传。我的问题是如何模拟连接中止的时间。在我的控制器中，我有，可以说模拟连接丢失，我的代码反应正常。但正如我所提到的，问题仍然是如何从测试中触发。我使用Mocha 3.2.0作为测试套件，在控制器端使用multiparty。

这是原始代码：

return new Promise((resolve, reject) => {
let form = new multiparty.Form();
let fileCount = 0;
form.on('part', function (part) {
    fileCount++;

    let tmpFile = path.join(os.tmpDir(), `${userId}_${timePrefix}_${path.basename(part.filename)}`);
    part.pipe(fs.createWriteStream(tmpFile));

    part.on('end', () => {
        resolve(fs.createReadStream(tmpFile));
    });

    part.on('error', function (error) {
        reject(error);
    });
});

form.on('error', (error) => {

  reject(new LogicalError(
      `Image input request is not valid`, errorCodes.VALIDATION.IMAGE_UPLOAD_DATA_INVALID);

});

form.parse(request);
}).then((imageStream) => {
//AWS S3 is hendled here...

现在我添加了一个新案例，我触发了客户端连接被取消。我使用过Throttle。节流模拟慢速上传，在超时时我触发连接丢失。

return new Promise((resolve, reject) => {
    let form = new multiparty.Form();
    let fileCount = 0;
    let throttle = new Throttle(1); //1 = 1 byte. 1000 = 1kb... all per second


    form.on('part', function (part) {
        fileCount++;

        let tmpFile = path.join(os.tmpDir(), `${userId}_${timePrefix}_${path.basename(part.filename)}`);
        //part.pipe(fs.createWriteStream(tmpFile));
        //Slows down the upload
        part.pipe(throttle).pipe(fs.createWriteStream(tmpFile));

        //trigger connection aborted
        setTimeout(function () {
            request.emit('aborted');
        },10);


        part.on('end', () => {
            resolve(fs.createReadStream(tmpFile));
        });

        part.on('error', function (error) {
            reject(error);
        });
    });

    form.on('error', (error) => {

        let err = null;

        if (error.message.includes('Request aborted')){
            err = new LogicalError(
                `Client aborted image upload process`, errorCodes.VALIDATION.IMAGE_UPLOAD_CLIENT_REQUEST_ABORTED);
        } else {
            err = new LogicalError(
                `Image input request is not valid`, errorCodes.VALIDATION.IMAGE_UPLOAD_DATA_INVALID);
        }
        reject(err);

    });

    form.parse(request);
}).then((imageStream) => {
//AWS S3 is hendled here...

此模拟按预期工作。代码识别正确的条件（请求中止）。当然，我不能离开这个油门和setTimeout。如何从测试中触发它？

这是当前测试，即上传图像，并以Controller为目标。使用supertest：

describe('ImageControllerTest', function () {
    it('should upload image with name for the given user', (done) => {
        request(app)
            .put('path-to-controller/test-image.jpg')
            .attach('file', imagePath.testImagePath)
            .set(cons.TOKEN.HEADERS)
            .expect((res) => {
               //series of expect statements.
            })
            .expect(200, done);
    });
});

在我的第二次尝试中（代码在上面），我通过这样做模拟了连接中止：

setTimeout(function () {
            request.emit('aborted');
        },10);

Multiparty会收听该事件，并做出相应的反应。

我怎样从测试中做到这一点？

Answer 1

简单的方法是 - 使用以下方法在子进程中创建客户端：

const fork = require('child_process').fork;
const path = require('path');
const child = fork(path.join(__dirname, './some-sub.js'));

当你需要触发取消时 - 只需关闭该过程：

child.close();

或者来自流程本身：

process.exit(0);

如何模拟客户端中止

1 个答案: