ImageStore

[paperdoorknob] / paperdoorknob_test.py

diff --git a/paperdoorknob_test.py b/paperdoorknob_test.py
index 14f6dd9cde0057e459c3fa7bd961c1fdf02fc436..49f1872842ea2e61503fd119942c68bfa0eb6bcc 100644 (file)
--- a/paperdoorknob_test.py
+++ b/paperdoorknob_test.py
@@ -13,6 +13,7 @@ import subprocess
 import paperdoorknob
 
 from testing.fakeserver import FakeGlowficServer
 import paperdoorknob
 
 from testing.fakeserver import FakeGlowficServer

+from domfilter import ApplyDOMFilters

 from fetch import DirectFetcher, FakeFetcher, Fetcher
 from spec import Spec
 from texify import DirectTexifier, PandocTexifier, VerifyingTexifier
 from fetch import DirectFetcher, FakeFetcher, Fetcher
 from spec import Spec
 from texify import DirectTexifier, PandocTexifier, VerifyingTexifier


@@ -31,11 +32,10 @@ class BaseTestProcess(ABC):
         raise NotImplementedError()
 
     def testReplies(self) -> None:
         raise NotImplementedError()
 
     def testReplies(self) -> None:

-        replies = paperdoorknob.replies(
-            paperdoorknob.clean(
-                paperdoorknob.parse(
-                    self.fetcher().fetch(
-                        self.url()))))
+        replies = list(paperdoorknob.replies(
+            paperdoorknob.parse(self.fetcher().fetch(self.url()))))
+        for r in replies:
+            ApplyDOMFilters('NoEdit,NoFooter', r)

         assert [r.text.strip() for r in replies] == [
             "This is glowfic",
             "You sure?",
         assert [r.text.strip() for r in replies] == [
             "This is glowfic",
             "You sure?",


@@ -47,10 +47,13 @@ class BaseTestProcess(ABC):
             self.url(),
             self.fetcher(),
             lambda x: x,
             self.url(),
             self.fetcher(),
             lambda x: x,

+            lambda x: ApplyDOMFilters('NoEdit,NoFooter', x),

             PandocTexifier('pandoc'),
             PandocTexifier('pandoc'),

+            'margin=20mm',

             buf)
         paperdoorknob.process(spec)
         assert buf.getvalue() == b'''\\documentclass{article}
             buf)
         paperdoorknob.process(spec)
         assert buf.getvalue() == b'''\\documentclass{article}

+\\usepackage[margin=20mm]{geometry}

 \\begin{document}
 This is glowfic
 You \\emph{sure}?
 \\begin{document}
 This is glowfic
 You \\emph{sure}?


@@ -62,7 +65,14 @@ Pretty sure.
         texifier = VerifyingTexifier(
             PandocTexifier('pandoc'), DirectTexifier())
         buf = io.BytesIO()
         texifier = VerifyingTexifier(
             PandocTexifier('pandoc'), DirectTexifier())
         buf = io.BytesIO()

-        spec = Spec(self.url(), self.fetcher(), lambda x: x, texifier, buf)
+        spec = Spec(
+            self.url(),
+            self.fetcher(),
+            lambda x: x,
+            lambda x: ApplyDOMFilters('NoEdit,NoFooter', x),
+            texifier,
+            None,
+            buf)

         paperdoorknob.process(spec)
 
     def testPDF(self) -> None:
         paperdoorknob.process(spec)
 
     def testPDF(self) -> None:


@@ -71,7 +81,9 @@ Pretty sure.
                 self.url(),
                 self.fetcher(),
                 lambda x: x,
                 self.url(),
                 self.fetcher(),
                 lambda x: x,

+                lambda x: ApplyDOMFilters('NoEdit,NoFooter', x),

                 PandocTexifier('pandoc'),
                 PandocTexifier('pandoc'),

+                None,

                 out)
             paperdoorknob.process(spec)
         subprocess.run(['pdflatex', 'test.tex'],
                 out)
             paperdoorknob.process(spec)
         subprocess.run(['pdflatex', 'test.tex'],