diff options
| -rw-r--r-- | libpathod/app.py | 1 | ||||
| -rw-r--r-- | libpathod/rparse.py | 6 | ||||
| -rw-r--r-- | test/test_rparse.py | 6 |
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(): |