mirror of
				https://github.com/NodeBB/NodeBB.git
				synced 2025-10-31 02:55:58 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			433 lines
		
	
	
		
			16 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			433 lines
		
	
	
		
			16 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| 'use strict';
 | |
| 
 | |
| var async = require('async');
 | |
| var	assert = require('assert');
 | |
| var nconf = require('nconf');
 | |
| var path = require('path');
 | |
| var request = require('request');
 | |
| 
 | |
| var db = require('./mocks/databasemock');
 | |
| var categories = require('../src/categories');
 | |
| var topics = require('../src/topics');
 | |
| var user = require('../src/user');
 | |
| var groups = require('../src/groups');
 | |
| var privileges = require('../src/privileges');
 | |
| var meta = require('../src/meta');
 | |
| var socketUser = require('../src/socket.io/user');
 | |
| var helpers = require('./helpers');
 | |
| var file = require('../src/file');
 | |
| var image = require('../src/image');
 | |
| 
 | |
| describe('Upload Controllers', function () {
 | |
| 	var tid;
 | |
| 	var cid;
 | |
| 	var pid;
 | |
| 	var adminUid;
 | |
| 	var regularUid;
 | |
| 
 | |
| 	before(function (done) {
 | |
| 		async.series({
 | |
| 			category: function (next) {
 | |
| 				categories.create({
 | |
| 					name: 'Test Category',
 | |
| 					description: 'Test category created by testing script',
 | |
| 				}, next);
 | |
| 			},
 | |
| 			adminUid: function (next) {
 | |
| 				user.create({ username: 'admin', password: 'barbar' }, next);
 | |
| 			},
 | |
| 			regularUid: function (next) {
 | |
| 				user.create({ username: 'regular', password: 'zugzug' }, next);
 | |
| 			},
 | |
| 		}, function (err, results) {
 | |
| 			if (err) {
 | |
| 				return done(err);
 | |
| 			}
 | |
| 			adminUid = results.adminUid;
 | |
| 			regularUid = results.regularUid;
 | |
| 			cid = results.category.cid;
 | |
| 
 | |
| 			topics.post({ uid: adminUid, title: 'test topic title', content: 'test topic content', cid: results.category.cid }, function (err, result) {
 | |
| 				if (err) {
 | |
| 					return done(err);
 | |
| 				}
 | |
| 				tid = result.topicData.tid;
 | |
| 				pid = result.postData.pid;
 | |
| 				groups.join('administrators', adminUid, done);
 | |
| 			});
 | |
| 		});
 | |
| 	});
 | |
| 
 | |
| 	describe('regular user uploads', function () {
 | |
| 		var jar;
 | |
| 		var csrf_token;
 | |
| 
 | |
| 		before(function (done) {
 | |
| 			helpers.loginUser('regular', 'zugzug', function (err, _jar, _csrf_token) {
 | |
| 				assert.ifError(err);
 | |
| 				jar = _jar;
 | |
| 				csrf_token = _csrf_token;
 | |
| 				privileges.global.give(['groups:upload:post:file'], 'registered-users', done);
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload an image to a post', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(body && body.status && body.response && body.response.images);
 | |
| 				assert(Array.isArray(body.response.images));
 | |
| 				assert(body.response.images[0].url);
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload an image to a post and then delete the upload', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.strictEqual(res.statusCode, 200);
 | |
| 				assert(body && body.status && body.response && body.response.images);
 | |
| 				assert(Array.isArray(body.response.images));
 | |
| 				assert(body.response.images[0].url);
 | |
| 				var name = body.response.images[0].url.replace(nconf.get('relative_path') + nconf.get('upload_url'), '');
 | |
| 				socketUser.deleteUpload({ uid: regularUid }, { uid: regularUid, name: name }, function (err) {
 | |
| 					assert.ifError(err);
 | |
| 					db.getSortedSetRange('uid:' + regularUid + ':uploads', 0, -1, function (err, uploads) {
 | |
| 						assert.ifError(err);
 | |
| 						assert.equal(uploads.includes(name), false);
 | |
| 						done();
 | |
| 					});
 | |
| 				});
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow deleting if path is not correct', function (done) {
 | |
| 			socketUser.deleteUpload({ uid: adminUid }, { uid: regularUid, name: '../../bkconfig.json' }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-path]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow deleting if path is not correct', function (done) {
 | |
| 			socketUser.deleteUpload({ uid: adminUid }, { uid: regularUid, name: '/files/../../bkconfig.json' }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-path]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should resize and upload an image to a post', function (done) {
 | |
| 			var oldValue = meta.config.resizeImageWidth;
 | |
| 			meta.config.resizeImageWidth = 10;
 | |
| 			meta.config.resizeImageWidthThreshold = 10;
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(body && body.status && body.response && body.response.images);
 | |
| 				assert(Array.isArray(body.response.images));
 | |
| 				assert(body.response.images[0].url);
 | |
| 				assert(body.response.images[0].url.match(/\/assets\/uploads\/files\/\d+-test-resized\.png/));
 | |
| 				meta.config.resizeImageWidth = oldValue;
 | |
| 				meta.config.resizeImageWidthThreshold = 1520;
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 
 | |
| 		it('should upload a file to a post', function (done) {
 | |
| 			var oldValue = meta.config.allowedFileExtensions;
 | |
| 			meta.config.allowedFileExtensions = 'png,jpg,bmp,html';
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/503.html'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				meta.config.allowedFileExtensions = oldValue;
 | |
| 				assert.ifError(err);
 | |
| 				assert.strictEqual(res.statusCode, 200);
 | |
| 				assert(body && body.status && body.response && body.response.images);
 | |
| 				assert(Array.isArray(body.response.images));
 | |
| 				assert(body.response.images[0].url);
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail to upload image to post if image dimensions are too big', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/toobig.jpg'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.strictEqual(res.statusCode, 500);
 | |
| 				assert(body && body.status && body.status.message);
 | |
| 				assert.strictEqual(body.status.message, 'Input image exceeds pixel limit');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail to upload image to post if image is broken', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/brokenimage.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.strictEqual(res.statusCode, 500);
 | |
| 				assert(body && body.status && body.status.message);
 | |
| 				assert(body.status.message.startsWith('pngload_buffer: non-recoverable state'));
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail if file is not an image', function (done) {
 | |
| 			image.isFileTypeAllowed(path.join(__dirname, '../test/files/notanimage.png'), function (err) {
 | |
| 				assert.strictEqual(err.message, 'Input file contains unsupported image format');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail if file is not an image', function (done) {
 | |
| 			image.isFileTypeAllowed(path.join(__dirname, '../test/files/notanimage.png'), function (err) {
 | |
| 				assert.strictEqual(err.message, 'Input file contains unsupported image format');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail if file is not an image', function (done) {
 | |
| 			image.size(path.join(__dirname, '../test/files/notanimage.png'), function (err) {
 | |
| 				assert.strictEqual(err.message, 'Input file contains unsupported image format');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail if file is missing', function (done) {
 | |
| 			image.size(path.join(__dirname, '../test/files/doesnotexist.png'), function (err) {
 | |
| 				assert.strictEqual(err.message, 'Input file is missing');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		// it('should fail if topic thumbs are disabled', function (done) {
 | |
| 		// 	helpers.uploadFile(nconf.get('url') + '/api/topic/thumb/upload', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 		// 		assert.ifError(err);
 | |
| 		// 		assert.strictEqual(res.statusCode, 404);
 | |
| 		// 		console.log(body);
 | |
| 		// 		assert(body && body.status && body.status.code);
 | |
| 		// 		assert.strictEqual(body.status.code, '[[error:topic-thumbnails-are-disabled]]');
 | |
| 		// 		done();
 | |
| 		// 	});
 | |
| 		// });
 | |
| 
 | |
| 		// it('should fail if file is not image', function (done) {
 | |
| 		// 	meta.config.allowTopicsThumbnail = 1;
 | |
| 		// 	helpers.uploadFile(nconf.get('url') + '/api/topic/thumb/upload', path.join(__dirname, '../test/files/503.html'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 		// 		assert.ifError(err);
 | |
| 		// 		assert.equal(res.statusCode, 500);
 | |
| 		// 		assert.equal(body.error, '[[error:invalid-file]]');
 | |
| 		// 		done();
 | |
| 		// 	});
 | |
| 		// });
 | |
| 
 | |
| 		// it('should upload topic thumb', function (done) {
 | |
| 		// 	meta.config.allowTopicsThumbnail = 1;
 | |
| 		// 	helpers.uploadFile(nconf.get('url') + '/api/topic/thumb/upload', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 		// 		assert.ifError(err);
 | |
| 		// 		assert.equal(res.statusCode, 200);
 | |
| 		// 		assert(Array.isArray(body));
 | |
| 		// 		assert(body[0].path);
 | |
| 		// 		assert(body[0].url);
 | |
| 		// 		done();
 | |
| 		// 	});
 | |
| 		// });
 | |
| 
 | |
| 		it('should not allow non image uploads', function (done) {
 | |
| 			socketUser.updateCover({ uid: 1 }, { uid: 1, file: { path: '../../text.txt' } }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-data]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow non image uploads', function (done) {
 | |
| 			socketUser.updateCover({ uid: 1 }, { uid: 1, imageData: 'data:text/html;base64,PHN2Zy9vbmxvYWQ9YWxlcnQoMik+' }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-image]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow svg uploads', function (done) {
 | |
| 			socketUser.updateCover({ uid: 1 }, { uid: 1, imageData: 'data:image/svg;base64,PHN2Zy9vbmxvYWQ9YWxlcnQoMik+' }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-image]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow non image uploads', function (done) {
 | |
| 			socketUser.uploadCroppedPicture({ uid: 1 }, { uid: 1, file: { path: '../../text.txt' } }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-data]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow non image uploads', function (done) {
 | |
| 			socketUser.uploadCroppedPicture({ uid: 1 }, { uid: 1, imageData: 'data:text/html;base64,PHN2Zy9vbmxvYWQ9YWxlcnQoMik+' }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-image]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should not allow svg uploads', function (done) {
 | |
| 			socketUser.uploadCroppedPicture({ uid: 1 }, { uid: 1, imageData: 'data:image/svg;base64,PHN2Zy9vbmxvYWQ9YWxlcnQoMik+' }, function (err) {
 | |
| 				assert.equal(err.message, '[[error:invalid-image]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should delete users uploads if account is deleted', function (done) {
 | |
| 			var jar;
 | |
| 			var uid;
 | |
| 			var url;
 | |
| 			var file = require('../src/file');
 | |
| 
 | |
| 			async.waterfall([
 | |
| 				function (next) {
 | |
| 					user.create({ username: 'uploader', password: 'barbar' }, next);
 | |
| 				},
 | |
| 				function (_uid, next) {
 | |
| 					uid = _uid;
 | |
| 					helpers.loginUser('uploader', 'barbar', next);
 | |
| 				},
 | |
| 				function (jar, csrf_token, next) {
 | |
| 					helpers.uploadFile(nconf.get('url') + '/api/post/upload', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, next);
 | |
| 				},
 | |
| 				function (res, body, next) {
 | |
| 					assert(body && body.status && body.response && body.response.images);
 | |
| 					assert(Array.isArray(body.response.images));
 | |
| 					assert(body.response.images[0].url);
 | |
| 					url = body.response.images[0].url;
 | |
| 
 | |
| 					user.delete(1, uid, next);
 | |
| 				},
 | |
| 				function (userData, next) {
 | |
| 					var filePath = path.join(nconf.get('upload_path'), url.replace('/assets/uploads', ''));
 | |
| 					file.exists(filePath, next);
 | |
| 				},
 | |
| 				function (exists, next) {
 | |
| 					assert(!exists);
 | |
| 					done();
 | |
| 				},
 | |
| 			], done);
 | |
| 		});
 | |
| 	});
 | |
| 
 | |
| 	describe('admin uploads', function () {
 | |
| 		var jar;
 | |
| 		var csrf_token;
 | |
| 
 | |
| 		before(function (done) {
 | |
| 			helpers.loginUser('admin', 'barbar', function (err, _jar, _csrf_token) {
 | |
| 				assert.ifError(err);
 | |
| 				jar = _jar;
 | |
| 				csrf_token = _csrf_token;
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload site logo', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/uploadlogo', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(Array.isArray(body));
 | |
| 				assert.equal(body[0].url, nconf.get('relative_path') + '/assets/uploads/system/site-logo.png');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail to upload invalid file type', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/category/uploadpicture', path.join(__dirname, '../test/files/503.html'), { params: JSON.stringify({ cid: cid }) }, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(body.error, '[[error:invalid-image-type, image/png, image/jpeg, image/pjpeg, image/jpg, image/gif, image/svg+xml]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail to upload category image with invalid json params', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/category/uploadpicture', path.join(__dirname, '../test/files/test.png'), { params: 'invalid json' }, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(body.error, '[[error:invalid-json]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload category image', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/category/uploadpicture', path.join(__dirname, '../test/files/test.png'), { params: JSON.stringify({ cid: cid }) }, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(Array.isArray(body));
 | |
| 				assert.equal(body[0].url, nconf.get('relative_path') + '/assets/uploads/category/category-1.png');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload default avatar', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/uploadDefaultAvatar', path.join(__dirname, '../test/files/test.png'), { }, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert.equal(body[0].url, nconf.get('relative_path') + '/assets/uploads/system/avatar-default.png');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload og image', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/uploadOgImage', path.join(__dirname, '../test/files/test.png'), { }, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert.equal(body[0].url, nconf.get('relative_path') + '/assets/uploads/system/og-image.png');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload favicon', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/uploadfavicon', path.join(__dirname, '../test/files/favicon.ico'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(Array.isArray(body));
 | |
| 				assert.equal(body[0].url, '/assets/uploads/system/favicon.ico');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload touch icon', function (done) {
 | |
| 			var touchiconAssetPath = '/assets/uploads/system/touchicon-orig.png';
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/uploadTouchIcon', path.join(__dirname, '../test/files/test.png'), {}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(Array.isArray(body));
 | |
| 				assert.equal(body[0].url, touchiconAssetPath);
 | |
| 				meta.config['brand:touchIcon'] = touchiconAssetPath;
 | |
| 				request(nconf.get('url') + '/apple-touch-icon', function (err, res, body) {
 | |
| 					assert.ifError(err);
 | |
| 					assert.equal(res.statusCode, 200);
 | |
| 					assert(body);
 | |
| 					done();
 | |
| 				});
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should upload regular file', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/upload/file', path.join(__dirname, '../test/files/test.png'), {
 | |
| 				params: JSON.stringify({
 | |
| 					folder: 'system',
 | |
| 				}),
 | |
| 			}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 200);
 | |
| 				assert(Array.isArray(body));
 | |
| 				assert.equal(body[0].url, '/assets/uploads/system/test.png');
 | |
| 				assert(file.existsSync(path.join(nconf.get('upload_path'), 'system', 'test.png')));
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 
 | |
| 		it('should fail to upload regular file in wrong directory', function (done) {
 | |
| 			helpers.uploadFile(nconf.get('url') + '/api/admin/upload/file', path.join(__dirname, '../test/files/test.png'), {
 | |
| 				params: JSON.stringify({
 | |
| 					folder: '../../system',
 | |
| 				}),
 | |
| 			}, jar, csrf_token, function (err, res, body) {
 | |
| 				assert.ifError(err);
 | |
| 				assert.equal(res.statusCode, 500);
 | |
| 				assert.strictEqual(body.error, '[[error:invalid-path]]');
 | |
| 				done();
 | |
| 			});
 | |
| 		});
 | |
| 	});
 | |
| });
 |