const assert = require('assert'); const sinon = require('sinon'); const proxyquire = require('proxyquire'); const conf = require('../server/config.js') conf.notLocalHost = false; let redisStub = {}; let exists = sinon.stub(); let hget = sinon.stub(); let hmset = sinon.stub(); let expire = sinon.stub(); let del = sinon.stub(); redisStub.createClient = function() { return { on: sinon.spy(), exists: exists, hget: hget, hmset: hmset, expire: expire, del: del } } let fsStub = {}; fsStub.statSync = sinon.stub(); fsStub.createReadStream = sinon.stub(); fsStub.createWriteStream = sinon.stub(); fsStub.unlinkSync = sinon.stub(); let logStub = {}; logStub.info = sinon.stub(); logStub.error = sinon.stub(); const storage = proxyquire('../server/storage', { 'redis': redisStub, 'fs': fsStub, './log.js': function() { return logStub } }); describe('Testing Exists from local filesystem', function() { it('Exists returns true when file exists', function() { exists.callsArgWith(1, null, 1); return storage.exists('test') .then((reply) => assert(reply)); }) it('Exists returns false when file does not exist', function() { exists.callsArgWith(1, null, 0); return storage.exists('test') .then((reply) => assert(!reply)); }) }); describe('Testing Filename from local filesystem', function() { it('Filename returns properly if id exists', function() { hget.callsArgWith(2, null, 'Filename.moz'); return storage.filename('test') .then(reply => assert(1)) .catch(reply => assert.fail()) }) it('Filename fails if id does not exist', function() { hget.callsArgWith(2, null, 'Filename.moz'); return storage.filename('test') .then(reply => assert.fail()) .catch(reply => assert(1)) }) }) describe('Testing Length from local filesystem', function() { it('Filesize returns properly if id exists', function() { fsStub.statSync.returns({size: 10}); return storage.length('Filename.moz') .then(reply => assert(1)) .catch(reply => assert.fail()) }) it('Filesize fails if the id does not exist', function() { fsStub.statSync.returns(null); return storage.length('Filename.moz') .then(reply => assert.fail()) .catch(reply => assert(1)) }) }) describe('Testing Get from local filesystem', function() { it('Get returns properly if id exists', function() { fsStub.createReadStream.returns(1); if (storage.get('Filename.moz')) { assert(1) } else { assert.fail(); } }) it('Get fails if the id does not exist', function() { fsStub.createReadStream.returns(null); if (storage.get('Filename.moz')) { assert.fail(); } else { assert(1); } }) }) describe('Testing Set to local filesystem', function() { it('Successfully writes the file to the local filesystem', function() { let stub = sinon.stub(); stub.withArgs('close', sinon.match.any).callsArgWithAsync(1) stub.withArgs('error', sinon.match.any).returns(1); fsStub.createWriteStream.returns({ on: stub }) return storage.set('test', { pipe: sinon.stub() }, 'Filename.moz', 'moz.la') .then(reply => { assert(reply.uuid); assert.equal(reply.url, 'moz.la'); }) .catch(reply => assert.fail()) }) it('Fails when the file is not properly written to the local filesystem', function() { let stub = sinon.stub(); stub.withArgs('error', sinon.match.any).callsArgWithAsync(1) stub.withArgs('close', sinon.match.any).returns(1); fsStub.createWriteStream.returns({ on: stub }) return storage.set('test', {pipe: sinon.stub()}, 'Filename.moz', 'moz.la') .then(reply => assert.fail()) .catch(reply => assert(1)) }) }) describe('Testing Delete from local filesystem', function() { it('Deletes properly if id exists', function() { hget.callsArgWith(2, null, '123'); fsStub.unlinkSync.returns(1); return storage.delete('Filename.moz', '123') .then(reply => assert(reply)) .catch(reply => assert.fail()) }) it('Delete fails if id does not exist', function() { hget.callsArgWith(2, null, null); return storage.delete('Filename.moz', '123') .then(reply => assert.fail()) .catch(reply => assert(1)) }) it('Delete fails if the delete token does not match', function() { hget.callsArgWith(2, null, null); return storage.delete('Filename.moz', '123') .then(reply => assert.fail()) .catch(reply => assert(1)) }) }) describe('Testing Forced Delete from local filesystem', function() { it('Deletes properly if id exists', function() { fsStub.unlinkSync.returns(1); return storage.forceDelete('Filename.moz') .then(reply => assert(reply)) }) it('Deletes fails if id does not exist, but no reject is called', function() { fsStub.unlinkSync.returns(0); return storage.forceDelete('Filename.moz') .then(reply => assert(!reply)) }) })