X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/55958ec0820a10472e5091b105d730fa38172390..36ae1d5f39e74dc6c6cc1cd253d029f86003c83f:/paperdoorknob_test.py diff --git a/paperdoorknob_test.py b/paperdoorknob_test.py index 5fe1779..7cb73b7 100644 --- a/paperdoorknob_test.py +++ b/paperdoorknob_test.py @@ -6,6 +6,7 @@ import unittest +import io import threading from http.server import BaseHTTPRequestHandler, HTTPServer import requests @@ -31,11 +32,15 @@ class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler): body = b'''
+
We don't want edit boxes
This is glowfic +
We don't want footers
+
We don't want edit boxes
You sure? +
We don't want footers
Pretty sure. @@ -95,14 +100,15 @@ class TestFetch(unittest.TestCase): paperdoorknob.fetch(f"http://localhost:{self._port()}", s, TIMEOUT) self.assertEqual(self._request_counter, 1) - def testFetchConents(self) -> None: + def testReplies(self) -> None: with requests.session() as s: - post = paperdoorknob.Post(paperdoorknob.fetch( - f"http://localhost:{self._port()}", s, TIMEOUT)) - self.assertEqual(post.text().text.strip(), "This is glowfic") - self.assertEqual([r.text.strip() for r in post.replies()], - ["You sure?", "Pretty sure."]) - self.assertEqual([r.text.strip() for r in post.entries()], + replies = paperdoorknob.replies( + paperdoorknob.clean( + paperdoorknob.fetch( + f"http://localhost:{self._port()}", + s, + TIMEOUT))) + self.assertEqual([r.text.strip() for r in replies], ["This is glowfic", "You sure?", "Pretty sure."]) def testFetchErrors(self) -> None: @@ -114,6 +120,18 @@ class TestFetch(unittest.TestCase): paperdoorknob.fetch( f"http://localhost:{self._port()}/server_error", s, TIMEOUT) + def testProcess(self) -> None: + with requests.session() as s: + buf = io.BytesIO() + paperdoorknob.process( + f"http://localhost:{self._port()}", + s, + TIMEOUT, + buf, + 'pandoc') + self.assertEqual(buf.getvalue(), + b'This is glowfic\nYou sure?\nPretty sure.\n') + if __name__ == '__main__': unittest.main()