Move per-thread processing stuff into Thread

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

index 64da1649b1209b2bb5b37f914e6f40f65aae89b7..bd26e3c9dfc5bc0ee4ec02eead21e34bdca42689 100644 (file)
--- a/glowfic_test.py
+++ b/glowfic_test.py
@@ -5,45 +5,62 @@
  # Free Software Foundation, version 3.
  
  
+from sys import stderr
  import unittest
  
-from bs4 import BeautifulSoup
-
+from fetch import FakeFetcher
  from images import FakeImageStore
  from glowfic import makeChunk, Thread
+from spec import Spec
  from texify import PandocTexifier
  
  
+def spec_for_testing(html: bytes) -> Spec:
+    return Spec('https://fake/test',
+                FakeFetcher({'https://fake/test': html,
+                             'https://fake/test?view=flat': html}),
+                FakeImageStore(),
+                lambda x: x,
+                lambda x: None,
+                PandocTexifier('pandoc'),
+                lambda x: x,
+                20,
+                b'',
+                None,
+                stderr.buffer,
+                lambda x: None)
+
+
  class TestSplit(unittest.TestCase):
  
      def testSplit1(self) -> None:
-        t = Thread(BeautifulSoup(b'''
+        t = Thread(spec_for_testing(b'''
              <html><body><div class="post-container post-post">
                The "post"
-            </div></body></html>''', 'html.parser'))
+            </div></body></html>'''))
          self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
                           [['The "post"']])
  
      def testSplit2(self) -> None:
-        t = Thread(BeautifulSoup(b'''
+        t = Thread(spec_for_testing(b'''
              <html><body>
                <div class="post-container post-post">The "post"</div>
                <div class="flat-post-replies">
                  <div class="post-container post-reply">The "reply"</div>
                </div>
-            </body></html>''', 'html.parser'))
+            </body></html>'''))
          self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
                           [['The "post"'], ['The "reply"']])
  
      def testSplit3(self) -> None:
-        t = Thread(BeautifulSoup(b'''
+        t = Thread(spec_for_testing(b'''
              <html><body>
                <div class="post-container post-post">The "post"</div>
                <div class="flat-post-replies">
                  <div class="post-container post-reply">1st reply</div>
                  <div class="post-container post-reply">2nd reply</div>
                </div>
-            </body></html>''', 'html.parser'))
+            </body></html>'''))
          self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
                           [['The "post"'], ['1st reply'], ['2nd reply']])
  
@@ -52,7 +69,7 @@ class TestMakeChunk(unittest.TestCase):
  
      def testEmptyContent(self) -> None:
          with open('testdata/empty-content.html', 'rb') as f:
-            t = Thread(BeautifulSoup(f, 'html.parser'))
+            t = Thread(spec_for_testing(f.read((9999))))
          c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore())
          self.assertEqual(
              c.icon,
@@ -75,7 +92,7 @@ class TestMakeChunk(unittest.TestCase):
  class TestThread(unittest.TestCase):
  
      def testTitle(self) -> None:
-        t = Thread(BeautifulSoup(b'''
+        t = Thread(spec_for_testing(b'''
              <html><body>
                <div class="content-header">
                  <span id="post-title">
@@ -83,9 +100,33 @@ class TestThread(unittest.TestCase):
                  </span>
                </div>
                <div class="post-container post-post">The "post"</div>
-            </body></html>''', 'html.parser'))
+            </body></html>'''))
          self.assertEqual(t.title(), 'Teh Story!')
  
+    def testNextThreadRelative(self) -> None:
+        t = Thread(spec_for_testing(b'''
+            <html><body>
+              <div class="post-navheader">
+                <a class="view-button-link" href="/posts/4567"><div class="view-button">Next Post &raquo;</div>
+                </a><a class="view-button-link" href="/posts/4321"><div class="view-button float-none">&laquo; Previous Post</div>
+                </a>
+              </div>
+              <div class="post-container post-post">The "post"</div>
+            </body></html>'''))
+        self.assertEqual(t.next_thread(), 'https://fake/posts/4567')
+
+    def testNextThreadAbsolute(self) -> None:
+        t = Thread(spec_for_testing(b'''
+            <html><body>
+              <div class="post-navheader">
+                <a class="view-button-link" href="https://elsewhere/posts/4567"><div class="view-button">Next Post &raquo;</div>
+                </a><a class="view-button-link" href="https://elsewhere/posts/4321"><div class="view-button float-none">&laquo; Previous Post</div>
+                </a>
+              </div>
+              <div class="post-container post-post">The "post"</div>
+            </body></html>'''))
+        self.assertEqual(t.next_thread(), 'https://elsewhere/posts/4567')
+
  
  if __name__ == '__main__':
      unittest.main()