[paperdoorknob] / glowfic_test.py

# paperdoorknob: Print glowfic
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation, version 3.


import unittest

from bs4 import BeautifulSoup

from images import FakeImageStore
from glowfic import chunkDOMs, makeChunk
from texify import PandocTexifier


class TestSplit(unittest.TestCase):

    def testSplit1(self) -> None:
        soup = BeautifulSoup(b'''
            <html><body><div class="post-container post-post">
              The "post"
            </div></body></html>''', 'html.parser')
        self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
                         [['The "post"']])

    def testSplit2(self) -> None:
        soup = BeautifulSoup(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')
        self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
                         [['The "post"'], ['The "reply"']])

    def testSplit3(self) -> None:
        soup = BeautifulSoup(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')
        self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
                         [['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>')

        self.assertEqual(
            PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n')


if __name__ == '__main__':
    unittest.main()
Commit	Line	Data
e6adf6ce SW	1	# paperdoorknob: Print glowfic
	2	#
	3	# This program is free software: you can redistribute it and/or modify it
	4	# under the terms of the GNU General Public License as published by the
	5	# Free Software Foundation, version 3.
	6
	7
	8	import unittest
	9
	10	from bs4 import BeautifulSoup
	11
aa060d9b SW	12	from images import FakeImageStore
aa060d9b SW	13	from glowfic import chunkDOMs, makeChunk
62043b2b	14	from texify import PandocTexifier
e6adf6ce SW	15
	16
	17	class TestSplit(unittest.TestCase):
	18
	19	def testSplit1(self) -> None:
	20	soup = BeautifulSoup(b'''
	21	<html><body><div class="post-container post-post">
	22	The "post"
	23	</div></body></html>''', 'html.parser')
	24	self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
	25	[['The "post"']])
	26
	27	def testSplit2(self) -> None:
	28	soup = BeautifulSoup(b'''
	29	<html><body>
	30	<div class="post-container post-post">The "post"</div>
	31	<div class="flat-post-replies">
	32	<div class="post-container post-reply">The "reply"</div>
	33	</div>
	34	</body></html>''', 'html.parser')
	35	self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
	36	[['The "post"'], ['The "reply"']])
	37
	38	def testSplit3(self) -> None:
	39	soup = BeautifulSoup(b'''
	40	<html><body>
	41	<div class="post-container post-post">The "post"</div>
	42	<div class="flat-post-replies">
	43	<div class="post-container post-reply">1st reply</div>
	44	<div class="post-container post-reply">2nd reply</div>
	45	</div>
	46	</body></html>''', 'html.parser')
	47	self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
	48	[['The "post"'], ['1st reply'], ['2nd reply']])
	49
	50
aa060d9b SW	51	class TestMakeChunk(unittest.TestCase):
	52
	53	def testEmptyContent(self) -> None:
	54	with open('testdata/empty-content.html', 'rb') as f:
	55	soup = BeautifulSoup(f, 'html.parser')
	56	c = makeChunk(next(iter(chunkDOMs(soup))), FakeImageStore())
	57	self.assertEqual(
	58	c.icon,
	59	'stored:https://d1anwqy6ci9o1i.cloudfront.net/' +
	60	'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png')
37c47bc2 SW	61	assert c.character
	62	assert c.screen_name
	63	assert c.author
	64	self.assertEqual(list(c.character.stripped_strings), ['Keltham'])
	65	self.assertEqual(
	66	list(c.screen_name.stripped_strings), ['lawful chaotic'])
	67	self.assertEqual(list(c.author.stripped_strings), ['Iarwain'])
aa060d9b SW	68	self.assertEqual(str(c.content),
	69	'<div class="post-content"><p></p></div>')
	70
62043b2b SW	71	self.assertEqual(
	72	PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n')
	73
aa060d9b	74
e6adf6ce SW	75	if __name__ == '__main__':
e6adf6ce SW	76	unittest.main()