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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
|
# vim: ts=4:sw=4:expandtab
# This file is part of ReText
# Copyright: 2018-2024 Dmitry Shachnev
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from textwrap import dedent
from unittest import TestCase, skipIf
from markdown import markdown
from markdown.extensions.codehilite import CodeHiliteExtension
from markdown.extensions.fenced_code import FencedCodeExtension
from ReText.mdx_posmap import PosMapExtension
try:
from pymdownx.superfences import SuperFencesCodeExtension
except ImportError:
SuperFencesCodeExtension = None
class PosMapTest(TestCase):
maxDiff = None
extensionsPosMap = [
CodeHiliteExtension(),
FencedCodeExtension(),
PosMapExtension()
]
extensionsNoPosMap = [
CodeHiliteExtension(),
FencedCodeExtension()
]
def test_normalUse(self):
text = dedent("""\
# line 1
- line 3
- line 4
- line 5
line 7
line 8
code block, line 10
""")
html = markdown(text, extensions=[PosMapExtension()])
self.assertIn('<h1 data-posmap="1">line 1</h1>', html)
self.assertIn('<ul data-posmap="5">', html)
self.assertIn('<p data-posmap="8">', html)
self.assertIn('<pre data-posmap="10"><code>code block, line 10', html)
self.assertNotIn("posmapmarker", html)
def test_highlightC(self):
text = dedent("""\
```c
#include <stdio.h>
int main(int argc, char **argv)
{
printf("Hello, world!\\n");
}
```""")
html = markdown(text, extensions=self.extensionsPosMap)
expected = markdown(text, extensions=self.extensionsNoPosMap)
self.assertIn('<div class="codehilite">', html)
self.assertMultiLineEqual(html, expected)
def test_highlightEmptyC(self):
text = dedent("""\
```c
```""")
html = markdown(text, extensions=self.extensionsPosMap)
expected = markdown(text, extensions=self.extensionsNoPosMap)
self.assertIn('<div class="codehilite">', html)
self.assertMultiLineEqual(html, expected)
def test_highlightPython(self):
text = dedent("""\
```python
if __name__ == "__main__":
print("Hello, world!")
```""")
html = markdown(text, extensions=self.extensionsPosMap)
expected = markdown(text, extensions=self.extensionsNoPosMap)
self.assertIn('<div class="codehilite">', html)
self.assertMultiLineEqual(html, expected)
def test_highlightEmptyPython(self):
text = dedent("""\
```python
```""")
html = markdown(text, extensions=self.extensionsPosMap)
expected = markdown(text, extensions=self.extensionsNoPosMap)
self.assertIn('<div class="codehilite">', html)
self.assertMultiLineEqual(html, expected)
def test_traditionalCodeBlock(self):
text = dedent("""\
:::python
if __name__ == "__main__":
print("Hello, world!")
a paragraph following the code block, line 5
""")
extensions = [CodeHiliteExtension(), PosMapExtension()]
html = markdown(text, extensions=extensions)
self.assertNotIn('posmapmarker', html)
self.assertIn('<div class="codehilite">', html)
self.assertIn('<p data-posmap="5">', html)
@skipIf(SuperFencesCodeExtension is None,
"pymdownx module is not available")
def test_superFences(self):
text = dedent("""\
```bash
tee ~/test << EOF
A
B
C
EOF
```""")
extensions = [SuperFencesCodeExtension(), PosMapExtension()]
html = markdown(text, extensions=extensions)
self.assertNotIn("posmapmarker", html)
expected = markdown(text, extensions=[SuperFencesCodeExtension()])
self.assertMultiLineEqual(html, expected)
@skipIf(SuperFencesCodeExtension is None,
"pymdownx module is not available")
def test_superFencesInList(self):
text = dedent("""\
1. List item 1
```python
import sys
sys.stdout.write("Hello, world!")
sys.stdout.write("One more call.")
```
2. List item 2
""")
extensions = [SuperFencesCodeExtension(), PosMapExtension()]
html = markdown(text, extensions=extensions)
self.assertIn('<ol data-posmap="12">', html)
html = html.replace(' data-posmap="12"', "")
self.assertNotIn("posmapmarker", html)
expected = markdown(text, extensions=[SuperFencesCodeExtension()])
self.assertMultiLineEqual(html, expected)
|
