realms-wiki/realms/modules/wiki/tests.py

import json
from nose.tools import *
from flask import url_for
from realms.modules.wiki.models import cname_to_filename, filename_to_cname
from realms.lib.test import BaseTest


class WikiBaseTest(BaseTest):
    def update_page(self, name, message=None, content=None):
        return self.client.post(url_for('wiki.page_write', name=name),
                                data=dict(message=message, content=content))

    def create_page(self, name, message=None, content=None):
        return self.client.post(url_for('wiki.page_write', name=name),
                                data=dict(message=message, content=content))


class UtilTest(WikiBaseTest):
    def test_cname_to_filename(self):
        eq_(cname_to_filename('test'), 'test.md')

    def test_filename_to_cname(self):
        eq_(filename_to_cname('test-1-2-3.md'), 'test-1-2-3')


class WikiTest(WikiBaseTest):
    def test_routes(self):
        self.assert_200(self.client.get(url_for("wiki.create")))
        self.create_page('test', message='test message', content='testing')

        for route in ['page', 'edit', 'history']:
            rv = self.client.get(url_for("wiki.%s" % route, name='test'))
            self.assert_200(rv, "wiki.%s: %s" % (route, rv.status_code))

        self.assert_200(self.client.get(url_for('wiki.index')))

    def test_write_page(self):
        self.assert_200(self.create_page('test', message='test message', content='testing'))

        rv = self.client.get(url_for('wiki.page', name='test'))
        self.assert_200(rv)

        self.assert_context('name', 'test')
        eq_(self.get_context_variable('page')['info']['message'], 'test message')
        eq_(self.get_context_variable('page')['data'], 'testing')

    def test_history(self):
        self.assert_200(self.client.get(url_for('wiki.history', name='test')))

    def test_delete_page(self):
        self.app.config['WIKI_LOCKED_PAGES'] = ['test']
        self.assert_403(self.client.delete(url_for('wiki.page_write', name='test')))
        self.app.config['WIKI_LOCKED_PAGES'] = []

        self.assert_200(self.client.delete(url_for('wiki.page_write', name='test')))

        rv = self.client.get(url_for('wiki.page', name='test'))
        self.assert_status(rv, 302)

    def test_revert(self):
        rv1 = self.create_page('test', message='test message', content='testing_old')
        self.update_page('test', message='test message', content='testing_new')
        data = json.loads(rv1.data)
        self.client.post(url_for('wiki.revert'), data=dict(name='test', commit=data['sha']))
        self.client.get(url_for('wiki.page', name='test'))
        eq_(self.get_context_variable('page')['data'], 'testing_old')
        self.assert_404(self.client.post(url_for('wiki.revert'), data=dict(name='test', commit='does not exist')))

        self.app.config['WIKI_LOCKED_PAGES'] = ['test']
        self.assert_403(self.client.post(url_for('wiki.revert'), data=dict(name='test', commit=data['sha'])))
        self.app.config['WIKI_LOCKED_PAGES'] = []

    def test_anon(self):
        rv1 = self.create_page('test', message='test message', content='testing_old')
        self.update_page('test', message='test message', content='testing_new')
        data = json.loads(rv1.data)
        self.app.config['ALLOW_ANON'] = False
        self.assert_403(self.update_page('test', message='test message', content='testing_again'))
        self.assert_403(self.client.post(url_for('wiki.revert'), data=dict(name='test', commit=data['sha'])))


class RelativePathTest(WikiTest):
    def configure(self):
        return dict(RELATIVE_PATH='wiki')