Ghost/core/test/unit/admin_spec.js

/*globals describe, beforeEach, afterEach, it*/
var fs      = require('fs-extra'),
    should  = require('should'),
    sinon   = require('sinon'),
    when    = require('when'),
    storage = require('../../server/storage'),

    // Stuff we are testing
    admin = require('../../server/controllers/admin');

describe('Admin Controller', function () {
    describe('upload', function () {

        var req, res, store;

        beforeEach(function () {
            req = {
                files: {
                    uploadimage: {
                        path: "/tmp/TMPFILEID"
                    }
                }
            };

            res = {
                send: function () {
                }
            };

            store = sinon.stub();
            store.save = sinon.stub().returns(when('URL'));
            store.exists = sinon.stub().returns(when(true));
            store.destroy = sinon.stub().returns(when());
            sinon.stub(storage, 'get_storage').returns(store);
        });

        afterEach(function () {
            storage.get_storage.restore();
        });

        describe('can not upload invalid file', function () {
            it('should return 415 for invalid file type', function () {
                res.send = sinon.stub();
                req.files.uploadimage.name = 'INVALID.FILE';
                req.files.uploadimage.type = 'application/octet-stream';
                admin.upload(req, res);
                res.send.calledOnce.should.be.true;
                res.send.args[0][0].should.equal(415);
                res.send.args[0][1].should.equal('Unsupported Media Type');
            });
        });

        describe('can not upload file with valid extension but invalid type', function () {
            it('should return 415 for invalid file type', function () {
                res.send = sinon.stub();
                req.files.uploadimage.name = 'INVALID.jpg';
                req.files.uploadimage.type = 'application/octet-stream';
                admin.upload(req, res);
                res.send.calledOnce.should.be.true;
                res.send.args[0][0].should.equal(415);
                res.send.args[0][1].should.equal('Unsupported Media Type');
            });
        });

        describe('valid file', function () {

            beforeEach(function () {
                req.files.uploadimage.name = 'IMAGE.jpg';
                req.files.uploadimage.type = 'image/jpeg';
                sinon.stub(fs, 'unlink').yields();
            });

            afterEach(function () {
                fs.unlink.restore();
            });

            it('can upload jpg', function (done) {
                sinon.stub(res, 'send', function (data) {
                    data.should.not.equal(415);
                    return done();
                });

                admin.upload(req, res);
            });

            it('cannot upload jpg with incorrect extension', function (done) {
                req.files.uploadimage.name = 'IMAGE.xjpg';
                sinon.stub(res, 'send', function (data) {
                    data.should.equal(415);
                    return done();
                });

                admin.upload(req, res);
            });

            it('can upload png', function (done) {
                req.files.uploadimage.name = 'IMAGE.png';
                req.files.uploadimage.type = 'image/png';
                sinon.stub(res, 'send', function (data) {
                    data.should.not.equal(415);
                    return done();
                });

                admin.upload(req, res);
            });

            it('can upload gif', function (done) {
                req.files.uploadimage.name = 'IMAGE.gif';
                req.files.uploadimage.type = 'image/gif';
                sinon.stub(res, 'send', function (data) {
                    data.should.not.equal(415);
                    return done();
                });

                admin.upload(req, res);
            });

            it('should send correct url', function (done) {
                sinon.stub(res, 'send', function (data) {
                    data.should.equal('URL');
                    return done();
                });

                admin.upload(req, res);
            });
        });
    });
});