Extensible, flag-controlled DOM filters

[paperdoorknob] / paperdoorknob_test.py

diff --git a/paperdoorknob_test.py b/paperdoorknob_test.py
index 14f6dd9cde0057e459c3fa7bd961c1fdf02fc436..aa50340ecfd7709ed3437b9ce006d22a5fe1804d 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,6 +47,7 @@ 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'),
             buf)
         paperdoorknob.process(spec)
             PandocTexifier('pandoc'),
             buf)
         paperdoorknob.process(spec)


@@ -62,7 +63,13 @@ 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,
+            buf)

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


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

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

                 PandocTexifier('pandoc'),
                 out)
             paperdoorknob.process(spec)
                 PandocTexifier('pandoc'),
                 out)
             paperdoorknob.process(spec)