import json
import os
import re
import tempfile
import unittest
from claude_logbook import webpage
PAYLOAD_RE = re.compile(
r'', re.S)
LOG_NAME = {"id": "abc", "p": "/proj", "u": 1, "c": [{"r": "u", "x": "hola"}]}
class TestPayload(unittest.TestCase):
def test_escapes_the_closing_tag(self):
raw = webpage.encode_payload({"s": [{"x": "mirá este de acá"}]})
self.assertNotIn(" de acá")
def test_does_not_escape_to_ascii(self):
self.assertIn("ñ", webpage.encode_payload({"s": [{"x": "año"}]}))
def test_separates_sessions_from_memories(self):
payload = webpage.build_payload([LOG_NAME], [{"name": "algo"}])
self.assertEqual(payload["s"], [LOG_NAME])
self.assertEqual(payload["m"], [{"name": "algo"}])
def test_no_memories_still_has_the_key(self):
self.assertEqual(webpage.build_payload([LOG_NAME])["m"], [])
class TestRender(unittest.TestCase):
def test_replaces_the_marker(self):
html = webpage.render([LOG_NAME], template="__DATA__")
self.assertNotIn("__DATA__", html)
self.assertIn('"id":"abc"', html)
def test_fails_if_the_template_has_no_marker(self):
with self.assertRaises(webpage.TemplateError):
webpage.render([LOG_NAME], template="sin marcador")
def test_fails_if_the_marker_is_repeated(self):
with self.assertRaises(webpage.TemplateError):
webpage.render([LOG_NAME], template="__DATA__ y __DATA__")
def test_transcript_with_html_does_not_cut_the_script(self):
# The case that motivates the escape: a session that talked about this very
# generator has "" and "__DATA__" inside the text.
dangerous = dict(LOG_NAME, c=[{"r": "u", "x": "poné ![]()
y __DATA__"}])
html = webpage.render([dangerous])
blocks = PAYLOAD_RE.findall(html)
self.assertEqual(len(blocks), 1)
round_trip = json.loads(blocks[0])
self.assertEqual(round_trip["s"][0]["c"][0]["x"],
"poné ![]()
y __DATA__")
class TestTemplate(unittest.TestCase):
def setUp(self):
self.html = webpage.template_text()
def test_packaged_template_has_a_single_marker(self):
self.assertEqual(self.html.count(webpage.MARKER), 1)
def test_is_a_complete_document(self):
# Sin doctype ni charset, un file:// se abre en quirks mode y con la
# system encoding: accents come out broken.
self.assertTrue(self.html.lstrip().startswith(""))
self.assertIn('', self.html)
self.assertIn('name="viewport"', self.html)
self.assertTrue(self.html.rstrip().endswith("