Test emit()

[paperdoorknob] / glowfic_test.py
diff --git a/glowfic_test.py b/glowfic_test.py

index bd26e3c9dfc5bc0ee4ec02eead21e34bdca42689..befd0b57a584d6a18624dbd48b7352b17dafa053 100644 (file)
--- a/glowfic_test.py
+++ b/glowfic_test.py
@@ -4,10 +4,12 @@
  # under the terms of the GNU General Public License as published by the
  # Free Software Foundation, version 3.
  
-
+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 +17,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 +29,7 @@ def spec_for_testing(html: bytes) -> Spec:
                  20,
                  b'',
                  None,
-                stderr.buffer,
+                stderr.buffer if outbuf is None else outbuf,
                  lambda x: None)
  
  
@@ -88,6 +90,21 @@ class TestMakeChunk(unittest.TestCase):
          self.assertEqual(
              PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n')
  
+    def testMinimal(self) -> None:
+        t = Thread(spec_for_testing(b'''
+            <html><body>
+              <div class="post-container post-post">
+                <div class="post-content">Just content</div>
+              </div>
+            </body></html>'''))
+        c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore())
+        self.assertIsNone(c.icon)
+        self.assertIsNone(c.character)
+        self.assertIsNone(c.screen_name)
+        self.assertIsNone(c.author)
+        self.assertEqual(str(c.content),
+                         '<div class="post-content">Just content</div>')
+
  
  class TestThread(unittest.TestCase):
  
@@ -128,5 +145,25 @@ 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'''
+            <html><body>
+              <div class="post-container post-post">
+                <div class="post-content">A</div>
+              </div>
+              <div class="flat-post-replies">
+                <div class="post-container post-reply">
+                  <div class="post-content">B</div>
+                </div>
+              </div>
+            </body></html>''', buf)).emit()
+        self.assertEqual(buf.getvalue(), rb'''\glowhead{}{}{}{}A
+\glowhead{}{}{}{}B
+''')
+
+
  if __name__ == '__main__':
      unittest.main()