Pretty-print the example.

nedtwigg · nedtwigg · commit 0383de217526 · 2024-06-27T18:53:51.000-07:00
diff --git a/python/example-pytest-selfie/tests/app_account_test.py b/python/example-pytest-selfie/tests/app_account_test.py
@@ -2,7 +2,8 @@
 from selfie_lib import expect_selfie
 
 from app import app, wait_for_incoming_email
-from tests.selfie_settings import web_selfie
+
+from .selfie_settings import web_selfie
 
 
 @pytest.fixture
@@ -13,14 +14,18 @@ def client():
 
 
 def test_homepage(client):
-    web_selfie(client.get("/")).to_be("""
-<html><body>
-  <h1>Please login</h1>
+    web_selfie(client.get("/")).to_be("""<html>
+ <body>
+  <h1>
+   Please login
+  </h1>
   <form action="/login" method="post">
-    <input type="text" name="email" placeholder="email">
-    <input type="submit" value="login">
+   <input name="email" placeholder="email" type="text"/>
+   <input type="submit" value="login"/>
   </form>
-</body></html>
+ </body>
+</html>
+
 ╔═ [status] ═╗
 200 OK""")
 
diff --git a/python/example-pytest-selfie/tests/selfie_settings.py b/python/example-pytest-selfie/tests/selfie_settings.py
@@ -1,3 +1,4 @@
+from bs4 import BeautifulSoup
 from selfie_lib import Camera, Snapshot, StringSelfie, expect_selfie
 from werkzeug.test import TestResponse
 
@@ -20,8 +21,21 @@ def _web_camera(response: TestResponse) -> Snapshot:
         return Snapshot.of(response.data.decode()).plus_facet("status", response.status)
 
 
-WEB_CAMERA = Camera.of(_web_camera)
+def _pretty_print_html(html: str) -> str:
+    return BeautifulSoup(html, "html.parser").prettify()
+
+
+def _pretty_print_lens(snapshot: Snapshot) -> Snapshot:
+    if "<html" in snapshot.subject.value_string():
+        return snapshot.plus_or_replace(
+            "", _pretty_print_html(snapshot.subject.value_string())
+        )
+    else:
+        return snapshot
+
+
+WEB_CAMERA = Camera.of(_web_camera).with_lens(_pretty_print_lens)
 
 
 def web_selfie(response: TestResponse) -> StringSelfie:
-    return expect_selfie(response, _web_camera)
+    return expect_selfie(response, WEB_CAMERA)
