Merge pull request #59 from Haggus/handle-file-limit

Handle 'limit' event when file is over fileSize

Author: richardgirges

lib/index.js

@@ -74,6 +74,11 @@ function processMultipart(options, req, res, next) {
     const buffers = [];
     let safeFileNameRegex = /[^\w-]/g;
 
+    file.on('limit', () => {
+      res.writeHead(413, {'Connection': 'close'});
+      res.end('File size limit has been reached');
+    });
+
     file.on('data', function(data) {
       buffers.push(data);
 

test/fileLimitUploads.spec.js

@@ -0,0 +1,36 @@
+'use strict';
+
+const path = require('path');
+const request = require('supertest');
+const server = require('./server');
+const app = server.setup({
+  limits: {fileSize: 200 * 1024} // set 200kb upload limit
+});
+const clearUploadsDir = server.clearUploadsDir;
+const fileDir = server.fileDir;
+
+describe('Test Single File Upload With File Size Limit', function() {
+  it(`upload 'basketball.png' (~154kb) with 200kb size limit`, function(done) {
+    let filePath = path.join(fileDir, 'basketball.png');
+
+    clearUploadsDir();
+
+    request(app)
+      .post('/upload/single')
+      .attach('testFile', filePath)
+      .expect(200)
+      .end(done);
+  });
+
+  it(`fail when uploading 'car.png' (~269kb) with 200kb size limit`, function(done) {
+    let filePath = path.join(fileDir, 'car.png');
+
+    clearUploadsDir();
+
+    request(app)
+      .post('/upload/single')
+      .attach('testFile', filePath)
+      .expect(413)
+      .end(done);
+  });
+});