1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
|
#
# Copyright (C) 2010 Wikkid Developers.
#
# This software is licensed under the GNU Affero General Public License
# version 3 (see the file LICENSE).
"""Tests for wikkid.formatter.registry."""
from textwrap import dedent
from wikkid.formatter.registry import get_wiki_formatter
from wikkid.tests import TestCase
class TestGetWikiFormtter(TestCase):
"""Tests for get_wiki_formatter."""
def test_default_missing(self):
self.assertRaises(
KeyError,
get_wiki_formatter,
'some content',
'missing')
def test_default_exists(self):
content, formatter = get_wiki_formatter(
'some content', 'rest')
self.assertEqual('some content', content)
self.assertEqual(
'RestructuredTextFormatter', formatter.__class__.__name__)
def test_specify_missing(self):
content, formatter = get_wiki_formatter(
dedent("""\
# missing
some content
"""), 'rest')
self.assertEqual('# missing\nsome content\n', content)
self.assertEqual(
'RestructuredTextFormatter', formatter.__class__.__name__)
def test_specify_extra_whitespace(self):
content, formatter = get_wiki_formatter(
dedent("""\
#\t\tpygments
some content
"""), 'rest')
self.assertEqual('some content\n', content)
self.assertEqual('PygmentsFormatter', formatter.__class__.__name__)
def test_specify_extra_params(self):
content, formatter = get_wiki_formatter(
dedent("""\
# pygments extra params
some content
"""), 'rest')
self.assertEqual('some content\n', content)
self.assertEqual('PygmentsFormatter', formatter.__class__.__name__)
def test_specify_pygments_case_insensitive(self):
content, formatter = get_wiki_formatter(
dedent("""\
# pygments
some content
"""), 'rest')
self.assertEqual('some content\n', content)
self.assertEqual('PygmentsFormatter', formatter.__class__.__name__)
|