def code(self, node):
import html
assert self.snippet == None, "code tag placed in the scene twice"
text = deindent(node.decode_contents())
try:
escape = bool(node.attrs.get('escape', True))
except:
escape = True
if escape:
text = html.unescape(text)
self.snippet = Snippet(text, lang=node.attrs.get('lang',''))
self._push_snippet()
def test_get_formatter(self):
snippet = Snippet({}, mock.sentinel.tokens)
self.assertIsInstance(snippet._get_formatter(), HtmlFormatter)
def test_has_formatted_source(self, get_formatter):
snippet = Snippet({}, mock.sentinel.tokens)
snippet.get_formatted_source()
self.assertEqual(get_formatter.return_value.format.call_count, 1)
self.assertIs(get_formatter.return_value.format.call_args[0][0], mock.sentinel.tokens)
