X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/136277e30143cd1219c896bb4980027ac3c6dbe1..55958ec0820a10472e5091b105d730fa38172390:/paperdoorknob_test.py?ds=inline
diff --git a/paperdoorknob_test.py b/paperdoorknob_test.py
index ab13eed..5fe1779 100644
--- a/paperdoorknob_test.py
+++ b/paperdoorknob_test.py
@@ -28,7 +28,21 @@ class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):
return 200
def do_GET(self) -> None:
- body = b'
This is glowfic'
+ body = b'''
+
+
+ This is glowfic
+
+
+
+ You sure?
+
+
+ Pretty sure.
+
+
+
+'''
self.send_response(self._response_code())
self.send_header("Content-type", "text/html")
self.send_header("Content-Length", str(len(body)))
@@ -83,11 +97,13 @@ class TestFetch(unittest.TestCase):
def testFetchConents(self) -> None:
with requests.session() as s:
- doc = paperdoorknob.fetch(
- f"http://localhost:{self._port()}", s, TIMEOUT)
- body = doc.body
- assert body
- self.assertEqual(body.text, "This is glowfic")
+ 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()],
+ ["This is glowfic", "You sure?", "Pretty sure."])
def testFetchErrors(self) -> None:
with requests.session() as s: