ForgeWiki/forgewiki/tests/test_wiki2markdown.py

import mock

from forgewiki.command.wiki2markdown.extractors import MySQLExtractor
from forgewiki.command.wiki2markdown.loaders import MediawikiLoader
from alluratest.controller import setup_basic_test
from allura import model as M
from forgewiki import model as WM
from allura.lib import helpers as h

import pylons
pylons.c = pylons.tmpl_context
from pylons import c as context


class TestMySQLExtractor(object):

    def setUp(self):
        setup_basic_test()
        self.options = mock.Mock()
        self.options.dump_dir = '/tmp/w2m_test'

        # monkey-patch MySQLExtractor for test
        def pages(self):
            # yield (page_id, page_data)
            yield 1, {'title': 'Test title', 'text': 'Test Text'}
            yield 2, {'title': 'Main_Page', 'text': 'Main_page text'}
            yield 3, {'title': 'Test', 'text': ''}

        MySQLExtractor._pages = pages
        self.extractor = MySQLExtractor(self.options)

    def test_extract_pages(self):
        self.extractor.extract_pages()

        with open('/tmp/w2m_test/pages/1.json', 'r') as f:
            json_page = f.read()
        assert json_page == '{"text": "Test Text", "title": "Test title"}'

        with open('/tmp/w2m_test/pages/2.json', 'r') as f:
            json_page = f.read()
        assert json_page == '{"text": "Main_page text", "title": "Main_Page"}'

        with open('/tmp/w2m_test/pages/3.json', 'r') as f:
            json_page = f.read()
        assert json_page == '{"text": "", "title": "Test"}'


class TestMediawikiLoader(object):

    def setUp(self):
        setup_basic_test()
        self.options = mock.Mock()
        # need test project with installed wiki app
        self.options.nbhd = 'Adobe'
        self.options.project = '--init--'

        nbhd = M.Neighborhood.query.get(name=self.options.nbhd)
        h.set_context(self.options.project, 'wiki', neighborhood=nbhd)

        # monkey-patch MediawikiLoader for test
        def pages(self):
            yield {'title': 'Test title', 'text': "'''bold''' ''italics''"}
            yield {'title': 'Main', 'text': "main"}
            yield {'title': 'Test', 'text': 'test'}

        MediawikiLoader._pages = pages
        self.loader = MediawikiLoader(self.options)

    def get_page(self, title):
        return WM.Page.query.get(app_config_id=context.app.config._id,
                                 title=title)

    def test_load_pages(self):
        self.loader.load_pages()
        page = self.get_page('Test title')
        assert '**bold** _italics_' in page.text

        page = self.get_page('Main')
        assert 'main' in page.text

        page = self.get_page('Test')
        print page.text, len(page.text)
        assert 'test' in page.text