[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 glowfic import chunkDOMs


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']])


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
	12	from glowfic import chunkDOMs
	13
	14
	15	class TestSplit(unittest.TestCase):
	16
	17	def testSplit1(self) -> None:
	18	soup = BeautifulSoup(b'''
	19	<html><body><div class="post-container post-post">
	20	The "post"
	21	</div></body></html>''', 'html.parser')
	22	self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
	23	[['The "post"']])
	24
	25	def testSplit2(self) -> None:
	26	soup = BeautifulSoup(b'''
	27	<html><body>
	28	<div class="post-container post-post">The "post"</div>
	29	<div class="flat-post-replies">
	30	<div class="post-container post-reply">The "reply"</div>
	31	</div>
	32	</body></html>''', 'html.parser')
	33	self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
	34	[['The "post"'], ['The "reply"']])
	35
	36	def testSplit3(self) -> None:
	37	soup = BeautifulSoup(b'''
	38	<html><body>
	39	<div class="post-container post-post">The "post"</div>
	40	<div class="flat-post-replies">
	41	<div class="post-container post-reply">1st reply</div>
	42	<div class="post-container post-reply">2nd reply</div>
	43	</div>
	44	</body></html>''', 'html.parser')
	45	self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
	46	[['The "post"'], ['1st reply'], ['2nd reply']])
	47
	48
	49	if __name__ == '__main__':
	50	unittest.main()