git.scottworley.com Git - paperdoorknob/blame_incremental

... / ...

Commit	Line	Data
	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	from sys import stderr
	9	import unittest
	10
	11	from fetch import FakeFetcher
	12	from images import FakeImageStore
	13	from glowfic import makeChunk, Thread
	14	from spec import Spec
	15	from texify import PandocTexifier
	16
	17
	18	def spec_for_testing(html: bytes) -> Spec:
	19	return Spec('https://fake/test',
	20	FakeFetcher({'https://fake/test': html,
	21	'https://fake/test?view=flat': html}),
	22	FakeImageStore(),
	23	lambda x: x,
	24	lambda x: None,
	25	PandocTexifier('pandoc'),
	26	lambda x: x,
	27	20,
	28	b'',
	29	None,
	30	stderr.buffer,
	31	lambda x: None)
	32
	33
	34	class TestSplit(unittest.TestCase):
	35
	36	def testSplit1(self) -> None:
	37	t = Thread(spec_for_testing(b'''
	38	<html><body><div class="post-container post-post">
	39	The "post"
	40	</div></body></html>'''))
	41	self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
	42	[['The "post"']])
	43
	44	def testSplit2(self) -> None:
	45	t = Thread(spec_for_testing(b'''
	46	<html><body>
	47	<div class="post-container post-post">The "post"</div>
	48	<div class="flat-post-replies">
	49	<div class="post-container post-reply">The "reply"</div>
	50	</div>
	51	</body></html>'''))
	52	self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
	53	[['The "post"'], ['The "reply"']])
	54
	55	def testSplit3(self) -> None:
	56	t = Thread(spec_for_testing(b'''
	57	<html><body>
	58	<div class="post-container post-post">The "post"</div>
	59	<div class="flat-post-replies">
	60	<div class="post-container post-reply">1st reply</div>
	61	<div class="post-container post-reply">2nd reply</div>
	62	</div>
	63	</body></html>'''))
	64	self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()],
	65	[['The "post"'], ['1st reply'], ['2nd reply']])
	66
	67
	68	class TestMakeChunk(unittest.TestCase):
	69
	70	def testEmptyContent(self) -> None:
	71	with open('testdata/empty-content.html', 'rb') as f:
	72	t = Thread(spec_for_testing(f.read((9999))))
	73	c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore())
	74	self.assertEqual(
	75	c.icon,
	76	'stored:https://d1anwqy6ci9o1i.cloudfront.net/' +
	77	'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png')
	78	assert c.character
	79	assert c.screen_name
	80	assert c.author
	81	self.assertEqual(list(c.character.stripped_strings), ['Keltham'])
	82	self.assertEqual(
	83	list(c.screen_name.stripped_strings), ['lawful chaotic'])
	84	self.assertEqual(list(c.author.stripped_strings), ['Iarwain'])
	85	self.assertEqual(str(c.content),
	86	'<div class="post-content"><p></p></div>')
	87
	88	self.assertEqual(
	89	PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n')
	90
	91	def testMinimal(self) -> None:
	92	t = Thread(spec_for_testing(b'''
	93	<html><body>
	94	<div class="post-container post-post">
	95	<div class="post-content">Just content</div>
	96	</div>
	97	</body></html>'''))
	98	c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore())
	99	self.assertIsNone(c.icon)
	100	self.assertIsNone(c.character)
	101	self.assertIsNone(c.screen_name)
	102	self.assertIsNone(c.author)
	103	self.assertEqual(str(c.content),
	104	'<div class="post-content">Just content</div>')
	105
	106
	107	class TestThread(unittest.TestCase):
	108
	109	def testTitle(self) -> None:
	110	t = Thread(spec_for_testing(b'''
	111	<html><body>
	112	<div class="content-header">
	113	<span id="post-title">
	114	<a href="/posts/1234">Teh Story!</a>
	115	</span>
	116	</div>
	117	<div class="post-container post-post">The "post"</div>
	118	</body></html>'''))
	119	self.assertEqual(t.title(), 'Teh Story!')
	120
	121	def testNextThreadRelative(self) -> None:
	122	t = Thread(spec_for_testing(b'''
	123	<html><body>
	124	<div class="post-navheader">
	125	<a class="view-button-link" href="/posts/4567"><div class="view-button">Next Post »</div>
	126	</a><a class="view-button-link" href="/posts/4321"><div class="view-button float-none">« Previous Post</div>
	127	</a>
	128	</div>
	129	<div class="post-container post-post">The "post"</div>
	130	</body></html>'''))
	131	self.assertEqual(t.next_thread(), 'https://fake/posts/4567')
	132
	133	def testNextThreadAbsolute(self) -> None:
	134	t = Thread(spec_for_testing(b'''
	135	<html><body>
	136	<div class="post-navheader">
	137	<a class="view-button-link" href="https://elsewhere/posts/4567"><div class="view-button">Next Post »</div>
	138	</a><a class="view-button-link" href="https://elsewhere/posts/4321"><div class="view-button float-none">« Previous Post</div>
	139	</a>
	140	</div>
	141	<div class="post-container post-post">The "post"</div>
	142	</body></html>'''))
	143	self.assertEqual(t.next_thread(), 'https://elsewhere/posts/4567')
	144
	145
	146	if __name__ == '__main__':
	147	unittest.main()