""" Python-Markdown Extension Regression Tests ========================================== A collection of regression tests to confirm that the included extensions continue to work as advertised. This used to be accomplished by doctests. """ from __future__ import unicode_literals import unittest import markdown class TestExtensionClass(unittest.TestCase): """ Test markdown.extensions.Extension. """ def setUp(self): class TestExtension(markdown.extensions.Extension): config = { 'foo': ['bar', 'Description of foo'], 'bar': ['baz', 'Description of bar'] } self.ext = TestExtension() self.ExtKlass = TestExtension def testGetConfig(self): self.assertEqual(self.ext.getConfig('foo'), 'bar') def testGetConfigDefault(self): self.assertEqual(self.ext.getConfig('baz'), '') self.assertEqual(self.ext.getConfig('baz', default='missing'), 'missing') def testGetConfigs(self): self.assertEqual(self.ext.getConfigs(), {'foo': 'bar', 'bar': 'baz'}) def testGetConfigInfo(self): self.assertEqual(dict(self.ext.getConfigInfo()), dict([('foo', 'Description of foo'), ('bar', 'Description of bar')]) def testSetConfig(self): self.ext.setConfig('foo', 'baz') self.assertEqual(self.ext.getConfigs(), {'foo': 'baz', 'bar': 'baz'}) def testSetConfigWithBadKey(self): # self.ext.setConfig('bad', 'baz) ==> KeyError self.assertRaises(KeyError, self.ext.setConfig, 'bad', 'baz') def testConfigAsArgListOnInit(self): ext = self.ExtKlass([('foo', 'baz'), ('bar', 'blah')]) self.assertEqual(ext.getConfigs(), {'foo': 'baz', 'bar': 'blah'}) def testConfigAsArgDictOnInit(self): ext = self.ExtKlass({'foo': 'baz', 'bar': 'blah', 'bar': 'blah'}) self.assertEqual(ext.getConfigs(), {'foo': 'baz', 'bar': 'blah'}) def testConfigAsKwargListOnInit(self): ext = self.ExtKlass(configs=[('foo', 'baz'), ('bar', 'blah')]) self.assertEqual(ext.getConfigs(), {'foo': 'baz', 'bar': 'blah'}) def testConfigAsKwargDictOnInit(self): ext = self.ExtKlass(configs={'foo': 'baz', 'bar': 'blah'}) self.assertEqual(ext.getConfigs(), {'foo': 'baz', 'bar': 'blah'}) def testConfigAsKwargsOnInit(self): ext = self.ExtKlass(foo='baz', bar='blah') self.assertEqual(ext.getConfigs(), {'foo': 'baz', 'bar': 'blah'}) class TestAbbr(unittest.TestCase): """ Test abbr extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['abbr']) def testSimpleAbbr(self): """ Test Abbreviations. """ text = 'Some text with an ABBR and a REF. Ignore REFERENCE and ref.' + \ '\n\n*[ABBR]: Abbreviation\n' + \ '*[REF]: Abbreviation Reference' self.assertEqual(self.md.convert(text), '

Some text with an ABBR ' 'and a REF. Ignore ' 'REFERENCE and ref.

') def testNestedAbbr(self): """ Test Nested Abbreviations. """ text = '[ABBR](/foo) and _ABBR_\n\n' + \ '*[ABBR]: Abreviation' self.assertEqual(self.md.convert(text), '

ABBR ' 'and ABBR

') class TestCodeHilite(unittest.TestCase): """ Test codehilite extension. """ def setUp(self): self.has_pygments = True try: import pygments except ImportError: self.has_pygments = False def testBasicCodeHilite(self): text = '\t# A Code Comment' md = markdown.Markdown(extensions=['codehilite']) if self.has_pygments: self.assertEqual(md.convert(text), '
' '
# A Code Comment\n'
                '
') else: self.assertEqual(md.convert(text), '
# A Code Comment'
                '
') def testLinenumsTrue(self): text = '\t# A Code Comment' md = markdown.Markdown(extensions=['codehilite(linenums=True)']) if self.has_pygments: # Differant versions of pygments output slightly different markup. # So we use 'startwith' and test just enough to confirm that # pygments received and processed linenums. self.assertTrue(md.convert(text).startswith( '
')) else: self.assertEqual(md.convert(text), '
# A Code Comment'
                '
') def testLinenumsFalse(self): text = '\t#!Python\n\t# A Code Comment' md = markdown.Markdown(extensions=['codehilite(linenums=False)']) if self.has_pygments: self.assertEqual(md.convert(text), '
' '
# A Code Comment\n'
                '
') else: self.assertEqual(md.convert(text), '
# A Code Comment'
                '
') def testLinenumsNone(self): text = '\t# A Code Comment' md = markdown.Markdown(extensions=['codehilite(linenums=None)']) if self.has_pygments: self.assertEqual(md.convert(text), '
' '
# A Code Comment\n'
                '
') else: self.assertEqual(md.convert(text), '
# A Code Comment'
                '
') def testLinenumsNoneWithShebang(self): text = '\t#!Python\n\t# A Code Comment' md = markdown.Markdown(extensions=['codehilite(linenums=None)']) if self.has_pygments: # Differant versions of pygments output slightly different markup. # So we use 'startwith' and test just enough to confirm that # pygments received and processed linenums. self.assertTrue(md.convert(text).startswith( '
')) else: self.assertEqual(md.convert(text), '
# A Code Comment'
                '
') def testLinenumsNoneWithColon(self): text = '\t:::Python\n\t# A Code Comment' md = markdown.Markdown(extensions=['codehilite(linenums=None)']) if self.has_pygments: self.assertEqual(md.convert(text), '
' '
# A Code Comment\n'
                '
') else: self.assertEqual(md.convert(text), '
# A Code Comment'
                '
') def testHighlightLinesWithColon(self): # Test with hl_lines delimited by single or double quotes. text0 = '\t:::Python hl_lines="2"\n\t#line 1\n\t#line 2\n\t#line 3' text1 = "\t:::Python hl_lines='2'\n\t#line 1\n\t#line 2\n\t#line 3" for text in (text0, text1): md = markdown.Markdown(extensions=['codehilite']) if self.has_pygments: self.assertEqual(md.convert(text), '
'
                    '#line 1\n'
                    '#line 2\n'
                    '#line 3\n'
                    '
') else: self.assertEqual(md.convert(text), '
'
                    '#line 1\n'
                    '#line 2\n'
                    '#line 3
') class TestFencedCode(unittest.TestCase): """ Test fenced_code extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['fenced_code']) self.has_pygments = True try: import pygments except ImportError: self.has_pygments = False def testBasicFence(self): """ Test Fenced Code Blocks. """ text = ''' A paragraph before a fenced code block: ~~~ Fenced code block ~~~''' self.assertEqual(self.md.convert(text), '

A paragraph before a fenced code block:

\n' '
Fenced code block\n'
            '
') def testSafeFence(self): """ Test Fenced Code with safe_mode. """ text = '~~~\nCode\n~~~' self.md.safeMode = 'replace' self.assertEqual(self.md.convert(text), '
Code\n'
            '
') def testNestedFence(self): """ Test nested fence. """ text = ''' ~~~~~~~~ ~~~~ ~~~~~~~~''' self.assertEqual(self.md.convert(text), '
\n'
            '~~~~\n'
            '
') def testFencedLanguage(self): """ Test Language Tags. """ text = ''' ~~~~{.python} # Some python code ~~~~''' self.assertEqual(self.md.convert(text), '
# Some python code\n'
            '
') def testFencedBackticks(self): """ Test Code Fenced with Backticks. """ text = ''' ````` # Arbitrary code ~~~~~ # these tildes will not close the block `````''' self.assertEqual(self.md.convert(text), '
# Arbitrary code\n'
        '~~~~~ # these tildes will not close the block\n'
        '
') def testFencedCodeWithHighlightLines(self): """ Test Fenced Code with Highlighted Lines. """ text = ''' ```hl_lines="1 3" line 1 line 2 line 3 ```''' md = markdown.Markdown(extensions=[ 'codehilite(linenums=None,guess_lang=False)', 'fenced_code']) if self.has_pygments: self.assertEqual(md.convert(text), '
'
                'line 1\n'
                'line 2\n'
                'line 3\n'
                '
') else: self.assertEqual(md.convert(text), '
line 1\n'
                'line 2\n'
                'line 3
') def testFencedLanguageAndHighlightLines(self): """ Test Fenced Code with Highlighted Lines. """ text0 = ''' ```.python hl_lines="1 3" #line 1 #line 2 #line 3 ```''' text1 = ''' ~~~{.python hl_lines='1 3'} #line 1 #line 2 #line 3 ~~~''' for text in (text0, text1): md = markdown.Markdown(extensions=[ 'codehilite(linenums=None,guess_lang=False)', 'fenced_code']) if self.has_pygments: self.assertEqual(md.convert(text), '
'
                    '#line 1\n'
                    '#line 2\n'
                    '#line 3\n'
                    '
') else: self.assertEqual(md.convert(text), '
#line 1\n'
                    '#line 2\n'
                    '#line 3
') class TestHeaderId(unittest.TestCase): """ Test HeaderId Extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['headerid']) def testBasicHeaderId(self): """ Test Basic HeaderID """ text = "# Some Header #" self.assertEqual(self.md.convert(text), '

Some Header

') def testUniqueFunc(self): """ Test 'unique' function. """ from markdown.extensions.headerid import unique ids = set(['foo']) self.assertEqual(unique('foo', ids), 'foo_1') self.assertEqual(ids, set(['foo', 'foo_1'])) def testUniqueIds(self): """ Test Unique IDs. """ text = '#Header\n#Header\n#Header' self.assertEqual(self.md.convert(text), '

Header

\n' '

Header

\n' '

Header

') def testBaseLevel(self): """ Test Header Base Level. """ text = '#Some Header\n## Next Level' self.assertEqual(markdown.markdown(text, ['headerid(level=3)']), '

Some Header

\n' '

Next Level

') def testHeaderInlineMarkup(self): """ Test Header IDs with inline markup. """ text = '#Some *Header* with [markup](http://example.com).' self.assertEqual(self.md.convert(text), '

Some Header with ' 'markup.

') def testHtmlEntities(self): """ Test HeaderIDs with HTML Entities. """ text = '# Foo & bar' self.assertEqual(self.md.convert(text), '

Foo & bar

') def testRawHtml(self): """ Test HeaderIDs with raw HTML. """ text = '# Foo Bar Baz.' self.assertEqual(self.md.convert(text), '

Foo Bar Baz.

') def testNoAutoIds(self): """ Test HeaderIDs with no auto generated IDs. """ text = '# Some Header\n# Another Header' self.assertEqual(markdown.markdown(text, ['headerid(forceid=False)']), '

Some Header

\n' '

Another Header

') def testHeaderIdWithMetaData(self): """ Test Header IDs with MetaData extension. """ text = '''header_level: 2 header_forceid: Off # A Header''' self.assertEqual(markdown.markdown(text, ['headerid', 'meta']), '

A Header

') def testHeaderIdWithAttr_List(self): """ Test HeaderIDs with Attr_List extension. """ text = '# Header1 {: #foo }\n# Header2 {: .bar }' self.assertEqual(markdown.markdown(text, ['headerid', 'attr_list']), '

Header1

\n' '

Header2

') # Switch order extensions are loaded - should be no change in behavior. self.assertEqual(markdown.markdown(text, ['attr_list', 'headerid']), '

Header1

\n' '

Header2

') class TestMetaData(unittest.TestCase): """ Test MetaData extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['meta']) def testBasicMetaData(self): """ Test basic metadata. """ text = '''Title: A Test Doc. Author: Waylan Limberg John Doe Blank_Data: The body. This is paragraph one.''' self.assertEqual(self.md.convert(text), '

The body. This is paragraph one.

') self.assertEqual(self.md.Meta, {'author': ['Waylan Limberg', 'John Doe'], 'blank_data': [''], 'title': ['A Test Doc.']}) def testMissingMetaData(self): """ Test document without Meta Data. """ text = ' Some Code - not extra lines of meta data.' self.assertEqual(self.md.convert(text), '
Some Code - not extra lines of meta data.\n'
            '
') self.assertEqual(self.md.Meta, {}) def testMetaDataWithoutNewline(self): """ Test doocument with only metadata and no newline at end.""" text = 'title: No newline' self.assertEqual(self.md.convert(text), '') self.assertEqual(self.md.Meta, {'title': ['No newline']}) class TestWikiLinks(unittest.TestCase): """ Test Wikilinks Extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['wikilinks']) self.text = "Some text with a [[WikiLink]]." def testBasicWikilinks(self): """ Test [[wikilinks]]. """ self.assertEqual(self.md.convert(self.text), '

Some text with a ' 'WikiLink.

') def testWikilinkWhitespace(self): """ Test whitespace in wikilinks. """ self.assertEqual(self.md.convert('[[ foo bar_baz ]]'), '

foo bar_baz

') self.assertEqual(self.md.convert('foo [[ ]] bar'), '

foo bar

') def testSimpleSettings(self): """ Test Simple Settings. """ self.assertEqual(markdown.markdown(self.text, ['wikilinks(base_url=/wiki/,end_url=.html,html_class=foo)']), '

Some text with a ' 'WikiLink.

') def testComplexSettings(self): """ Test Complex Settings. """ md = markdown.Markdown( extensions = ['wikilinks'], extension_configs = {'wikilinks': [ ('base_url', 'http://example.com/'), ('end_url', '.html'), ('html_class', '') ]}, safe_mode = True) self.assertEqual(md.convert(self.text), '

Some text with a ' 'WikiLink.

') def testWikilinksMetaData(self): """ test MetaData with Wikilinks Extension. """ text = """wiki_base_url: http://example.com/ wiki_end_url: .html wiki_html_class: Some text with a [[WikiLink]].""" md = markdown.Markdown(extensions=['meta', 'wikilinks']) self.assertEqual(md.convert(text), '

Some text with a ' 'WikiLink.

') # MetaData should not carry over to next document: self.assertEqual(md.convert("No [[MetaData]] here."), '

No MetaData ' 'here.

') def testURLCallback(self): """ Test used of a custom URL builder. """ from markdown.extensions.wikilinks import WikiLinkExtension def my_url_builder(label, base, end): return '/bar/' md = markdown.Markdown(extensions=[WikiLinkExtension(build_url=my_url_builder)]) self.assertEqual(md.convert('[[foo]]'), '

foo

') class TestAdmonition(unittest.TestCase): """ Test Admonition Extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['admonition']) def testRE(self): RE = self.md.parser.blockprocessors['admonition'].RE tests = [ ('!!! note', ('note', None)), ('!!! note "Please Note"', ('note', 'Please Note')), ('!!! note ""', ('note', '')), ] for test, expected in tests: self.assertEqual(RE.match(test).groups(), expected) class TestTOC(unittest.TestCase): """ Test TOC Extension. """ def setUp(self): self.md = markdown.Markdown(extensions=['toc']) def testMarker(self): """ Test TOC with a Marker. """ text = '[TOC]\n\n# Header 1\n\n## Header 2' self.assertEqual(self.md.convert(text), '
\n' '\n' '
\n' '

Header 1

\n' '

Header 2

') def testNoMarker(self): """ Test TOC without a Marker. """ text = '# Header 1\n\n## Header 2' self.assertEqual(self.md.convert(text), '

Header 1

\n' '

Header 2

') self.assertEqual(self.md.toc, '
\n' '\n' '
\n')