Move get_title() to Thread

[paperdoorknob] / domfilter_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 domfilter import ApplyDOMFilters


class TestDOMFilters(unittest.TestCase):

    def setUp(self) -> None:
        self._strike = BeautifulSoup(b'''
            <div class="post-content">
              <p><span style="text-decoration: line-through;">Abandon hope and endure.</span></p>
              <p>No. Win.</p>
            </div>''', 'html.parser')

    def testStrike(self) -> None:
        ApplyDOMFilters('Strike', self._strike)
        s = self._strike.find_all('s')
        self.assertEqual(len(s), 1)
        self.assertEqual(list(s[0].stripped_strings),
                         ['Abandon hope and endure.'])


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