Only look within each chunk-dom for chunk fields

[paperdoorknob] / glowfic_test.py

diff --git a/glowfic_test.py b/glowfic_test.py
index d84733382a52a4f63692060e1b746d3bf9849796..9b802c180cf3463e659f892948b0c9a64d4f936e 100644 (file)
--- a/glowfic_test.py
+++ b/glowfic_test.py
@@ -9,7 +9,8 @@ import unittest
 
 from bs4 import BeautifulSoup
 
 
 from bs4 import BeautifulSoup
 

-from glowfic import chunkDOMs
+from images import FakeImageStore
+from glowfic import chunkDOMs, makeChunk

 
 
 class TestSplit(unittest.TestCase):
 
 
 class TestSplit(unittest.TestCase):


@@ -46,5 +47,26 @@ class TestSplit(unittest.TestCase):
                          [['The "post"'], ['1st reply'], ['2nd reply']])
 
 
                          [['The "post"'], ['1st reply'], ['2nd reply']])
 
 

+class TestMakeChunk(unittest.TestCase):
+
+    def testEmptyContent(self) -> None:
+        with open('testdata/empty-content.html', 'rb') as f:
+            soup = BeautifulSoup(f, 'html.parser')
+        c = makeChunk(next(iter(chunkDOMs(soup))), FakeImageStore())
+        self.assertEqual(
+            c.icon,
+            'stored:https://d1anwqy6ci9o1i.cloudfront.net/' +
+            'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png')
+        assert c.character
+        assert c.screen_name
+        assert c.author
+        self.assertEqual(list(c.character.stripped_strings), ['Keltham'])
+        self.assertEqual(
+            list(c.screen_name.stripped_strings), ['lawful chaotic'])
+        self.assertEqual(list(c.author.stripped_strings), ['Iarwain'])
+        self.assertEqual(str(c.content),
+                         '<div class="post-content"><p></p></div>')
+
+

 if __name__ == '__main__':
     unittest.main()
 if __name__ == '__main__':
     unittest.main()