Make previews safer by stripping pauses.

3 files changed, 13 insertions, 0 deletions

diff --git a/libpathod/app.py b/libpathod/app.py
index 5de69337..111fafe5 100644
--- a/libpathod/app.py
+++ b/libpathod/app.py
@@ -90,6 +90,7 @@ def preview():
         return render("preview.html", **args)
 
     s = cStringIO.StringIO()
+    r.preview_safe()
     r.serve(s, check=app.config["pathod"].check_size)
     args["output"] = utils.escape_unprintables(s.getvalue())
     return render("preview.html", **args)
diff --git a/libpathod/rparse.py b/libpathod/rparse.py
index e56135ea..d60a78d3 100644
--- a/libpathod/rparse.py
+++ b/libpathod/rparse.py
@@ -527,6 +527,12 @@ class Message:
         l += len(self.body)
         return l
 
+    def preview_safe(self):
+        """
+            Modify this message to be safe for previews.
+        """
+        self.actions = [i for i in self.actions if i[1] != "pause"]
+
     def effective_length(self, actions):
         """
             Calculate the length of the base message with all applied actions.
diff --git a/test/test_rparse.py b/test/test_rparse.py
index 922c8fc6..dadf5bc1 100644
--- a/test/test_rparse.py
+++ b/test/test_rparse.py
@@ -493,6 +493,12 @@ class TestResponse:
         r.actions = actions
         testlen(r, actions)
 
+    def test_render(self):
+        r = rparse.parse_response({}, "400:p0,100:dr")
+        assert r.actions[0][1] == "pause"
+        r.preview_safe()
+        assert not r.actions[0][1] == "pause"
+
 
 
 def test_read_file():