Ghost/core/test/unit/apps/amp/router_spec.js

const should = require('should'),
    sinon = require('sinon'),
    path = require('path'),
    ampController = require('../../../../server/apps/amp/lib/router'),
    urlService = require('../../../../server/services/url'),
    helpers = require('../../../../server/services/routing/helpers'),
    common = require('../../../../server/lib/common'),
    testUtils = require('../../../utils'),
    configUtils = require('../../../utils/configUtils'),
    sandbox = sinon.sandbox.create();

// Helper function to prevent unit tests
// from failing via timeout when they
// should just immediately fail
function failTest(done) {
    return function (err) {
        done(err);
    };
}

describe('Unit - apps/amp/lib/router', function () {
    let res,
        req,
        defaultPath,
        rendererStub;

    beforeEach(function () {
        rendererStub = sandbox.stub();

        sandbox.stub(helpers, 'renderer').get(function () {
            return rendererStub;
        });

        res = {
            render: sandbox.spy(),
            locals: {
                context: ['amp', 'post']
            }
        };

        req = {
            route: {path: '/'},
            query: {r: ''},
            params: {},
            amp: {},
            body: {
                post: {
                    title: 'test'
                }
            }
        };

        defaultPath = path.join(configUtils.config.get('paths').appRoot, '/core/server/apps/amp/lib/views/amp.hbs');
    });

    afterEach(function () {
        sandbox.restore();
    });

    describe('fn: renderer', function () {
        it('should render default amp page when theme has no amp template', function (done) {
            helpers.renderer.callsFake(function (req, res, data) {
                res.routerOptions.defaultTemplate.should.eql(defaultPath);
                data.should.eql({post: {title: 'test'}});
                done();
            });

            ampController.renderer(req, res, failTest(done));
        });

        it('throws 404 when req.body has no post', function (done) {
            req.body = {};

            ampController.renderer(req, res, function (err) {
                (err instanceof common.errors.NotFoundError).should.be.true();
                helpers.renderer.called.should.be.false();
                done();
            });
        });

        it('throws 404 when req.body.post is a page', function (done) {
            req.body = {
                post: {
                    page: true
                }
            };

            ampController.renderer(req, res, function (err) {
                (err instanceof common.errors.NotFoundError).should.be.true();
                helpers.renderer.called.should.be.false();
                done();
            });
        });
    });

    describe('fn: getPostData', function () {
        let res, req, postLookupStub, post;

        beforeEach(function () {
            post = testUtils.DataGenerator.forKnex.createPost({slug: 'welcome'});

            res = {
                locals: {}
            };

            req = {};

            postLookupStub = sandbox.stub();

            sandbox.stub(helpers, 'postLookup').get(function () {
                return postLookupStub;
            });

            sandbox.stub(urlService, 'getPermalinkByUrl');
        });

        afterEach(function () {
            sandbox.restore();
        });

        it('should successfully get the post data from slug', function (done) {
            res.locals.relativeUrl = req.originalUrl = '/welcome/amp';

            urlService.getPermalinkByUrl.withArgs('/welcome/').returns('/:slug/');

            helpers.postLookup.withArgs('/welcome/', {permalinks: '/:slug/'}).resolves({
                post: post
            });

            ampController.getPostData(req, res, function () {
                req.body.post.should.be.eql(post);
                done();
            });
        });

        it('subdirectory: should successfully get the post data from slug', function (done) {
            req.originalUrl = '/blog/welcome/amp';
            res.locals.relativeUrl = '/welcome/amp';

            urlService.getPermalinkByUrl.withArgs('/welcome/').returns('/:slug/');

            helpers.postLookup.withArgs('/welcome/', {permalinks: '/:slug/'}).resolves({
                post: post
            });

            ampController.getPostData(req, res, function () {
                req.body.post.should.be.eql(post);
                done();
            });
        });

        it('should return error if postlookup returns NotFoundError', function (done) {
            res.locals.relativeUrl = req.originalUrl = '/welcome/amp';

            urlService.getPermalinkByUrl.withArgs('/welcome/').returns('/:slug/');

            helpers.postLookup.withArgs('/welcome/', {permalinks: '/:slug/'}).rejects(new common.errors.NotFoundError());

            ampController.getPostData(req, res, function (err) {
                (err instanceof common.errors.NotFoundError).should.be.true();
                done();
            });
        });
    });
});