X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/6409066b6abbe205fa64f844a01516f3f3de2553..afd3c3a13956819ceb32c28875f56cff7df8fe3f:/paperdoorknob_test.py diff --git a/paperdoorknob_test.py b/paperdoorknob_test.py index 83cf76a..ee4de04 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,7 +32,19 @@ 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. +
''' @@ -87,11 +100,16 @@ 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") + 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: with requests.session() as s: @@ -102,6 +120,19 @@ 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 \\emph{sure}?\nPretty sure.\n') + if __name__ == '__main__': unittest.main()