[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.


from sys import stderr
import unittest

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('test',
                FakeFetcher({'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(spec_for_testing(b'''
            <html><body><div class="post-container post-post">
              The "post"
            </div></body></html>'''))
        self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
                         [['The "post"']])

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

    def testSplit3(self) -> None:
        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>'''))
        self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
                         [['The "post"'], ['1st reply'], ['2nd reply']])


class TestMakeChunk(unittest.TestCase):

    def testEmptyContent(self) -> None:
        with open('testdata/empty-content.html', 'rb') as f:
            t = Thread(spec_for_testing(f.read((9999))))
        c = makeChunk(next(iter(t.chunkDOMs())), 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')


class TestThread(unittest.TestCase):

    def testTitle(self) -> None:
        t = Thread(spec_for_testing(b'''
            <html><body>
              <div class="content-header">
                <span id="post-title">
                  <a href="/posts/1234">Teh Story!</a>
                </span>
              </div>
              <div class="post-container post-post">The "post"</div>
            </body></html>'''))
        self.assertEqual(t.title(), 'Teh Story!')


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
19ed28f3	8	from sys import stderr
e6adf6ce SW	9	import unittest
e6adf6ce SW	10
19ed28f3	11	from fetch import FakeFetcher
aa060d9b	12	from images import FakeImageStore
94027099	13	from glowfic import makeChunk, Thread
19ed28f3	14	from spec import Spec
62043b2b	15	from texify import PandocTexifier
e6adf6ce SW	16
e6adf6ce SW	17
19ed28f3 SW	18	def spec_for_testing(html: bytes) -> Spec:
	19	return Spec('test',
	20	FakeFetcher({'test?view=flat': html}),
	21	FakeImageStore(),
	22	lambda x: x,
	23	lambda x: None,
	24	PandocTexifier('pandoc'),
	25	lambda x: x,
	26	20,
	27	b'',
	28	None,
	29	stderr.buffer,
	30	lambda x: None)
	31
	32
e6adf6ce SW	33	class TestSplit(unittest.TestCase):
	34
	35	def testSplit1(self) -> None:
19ed28f3	36	t = Thread(spec_for_testing(b'''
e6adf6ce SW	37	<html><body><div class="post-container post-post">
e6adf6ce SW	38	The "post"
19ed28f3	39	</div></body></html>'''))
94027099	40	self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
e6adf6ce SW	41	[['The "post"']])
	42
	43	def testSplit2(self) -> None:
19ed28f3	44	t = Thread(spec_for_testing(b'''
e6adf6ce SW	45	<html><body>
	46	<div class="post-container post-post">The "post"</div>
	47	<div class="flat-post-replies">
	48	<div class="post-container post-reply">The "reply"</div>
	49	</div>
19ed28f3	50	</body></html>'''))
94027099	51	self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
e6adf6ce SW	52	[['The "post"'], ['The "reply"']])
	53
	54	def testSplit3(self) -> None:
19ed28f3	55	t = Thread(spec_for_testing(b'''
e6adf6ce SW	56	<html><body>
	57	<div class="post-container post-post">The "post"</div>
	58	<div class="flat-post-replies">
	59	<div class="post-container post-reply">1st reply</div>
	60	<div class="post-container post-reply">2nd reply</div>
	61	</div>
19ed28f3	62	</body></html>'''))
94027099	63	self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
e6adf6ce SW	64	[['The "post"'], ['1st reply'], ['2nd reply']])
	65
	66
aa060d9b SW	67	class TestMakeChunk(unittest.TestCase):
	68
	69	def testEmptyContent(self) -> None:
	70	with open('testdata/empty-content.html', 'rb') as f:
19ed28f3	71	t = Thread(spec_for_testing(f.read((9999))))
94027099	72	c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore())
aa060d9b SW	73	self.assertEqual(
	74	c.icon,
	75	'stored:https://d1anwqy6ci9o1i.cloudfront.net/' +
	76	'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png')
37c47bc2 SW	77	assert c.character
	78	assert c.screen_name
	79	assert c.author
	80	self.assertEqual(list(c.character.stripped_strings), ['Keltham'])
	81	self.assertEqual(
	82	list(c.screen_name.stripped_strings), ['lawful chaotic'])
	83	self.assertEqual(list(c.author.stripped_strings), ['Iarwain'])
aa060d9b SW	84	self.assertEqual(str(c.content),
	85	'<div class="post-content"><p></p></div>')
	86
62043b2b SW	87	self.assertEqual(
	88	PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n')
	89
aa060d9b	90
21e82200 SW	91	class TestThread(unittest.TestCase):
	92
	93	def testTitle(self) -> None:
19ed28f3	94	t = Thread(spec_for_testing(b'''
21e82200 SW	95	<html><body>
	96	<div class="content-header">
	97	<span id="post-title">
	98	<a href="/posts/1234">Teh Story!</a>
	99	</span>
	100	</div>
	101	<div class="post-container post-post">The "post"</div>
19ed28f3	102	</body></html>'''))
21e82200 SW	103	self.assertEqual(t.title(), 'Teh Story!')
	104
	105
e6adf6ce SW	106	if __name__ == '__main__':
e6adf6ce SW	107	unittest.main()