X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/0668adb72b24796c0c017a3091042a4b2574e082..HEAD:/glowfic_test.py
diff --git a/glowfic_test.py b/glowfic_test.py
index 898aafb..d3fb2dd 100644
--- a/glowfic_test.py
+++ b/glowfic_test.py
@@ -4,10 +4,13 @@
# under the terms of the GNU General Public License as published by the
# Free Software Foundation, version 3.
-
+import dataclasses
+from io import BytesIO
from sys import stderr
import unittest
+from typing import Optional
+
from fetch import FakeFetcher
from images import FakeImageStore
from glowfic import makeChunk, Thread
@@ -15,7 +18,7 @@ from spec import Spec
from texify import PandocTexifier
-def spec_for_testing(html: bytes) -> Spec:
+def spec_for_testing(html: bytes, outbuf: Optional[BytesIO] = None) -> Spec:
return Spec('https://fake/test',
FakeFetcher({'https://fake/test': html,
'https://fake/test?view=flat': html}),
@@ -27,7 +30,7 @@ def spec_for_testing(html: bytes) -> Spec:
20,
b'',
None,
- stderr.buffer,
+ stderr.buffer if outbuf is None else outbuf,
lambda x: None)
@@ -143,5 +146,55 @@ class TestThread(unittest.TestCase):
self.assertEqual(t.next_thread(), 'https://elsewhere/posts/4567')
+class TestEmit(unittest.TestCase):
+
+ def testEmit(self) -> None:
+ buf = BytesIO()
+ Thread(spec_for_testing(b'''
+
+
+
+ ''', buf)).emit()
+ self.assertEqual(buf.getvalue(), rb'''\glowhead{}{}{}{}A
+\glowhead{}{}{}{}B
+''')
+
+ def testEmitTwoThreads(self) -> None:
+ buf = BytesIO()
+ spec = dataclasses.replace(spec_for_testing(b'', buf), fetcher=FakeFetcher({
+ 'https://fake/test': b'''
+ ''',
+ 'https://fake/test?view=flat': b'''
+
+
+ ''',
+ 'https://fake/page2': b'''''',
+ 'https://fake/page2?view=flat': b'''
+
+
+ ''',
+ }))
+ Thread(spec).emit()
+ self.assertEqual(buf.getvalue(), rb'''\glowhead{}{}{}{}A
+\glowhead{}{}{}{}B
+\glowhead{}{}{}{}C
+\glowhead{}{}{}{}D
+\glowhead{}{}{}{}E
+''')
+
+
if __name__ == '__main__':
unittest.main()