From 8b52ea248e690df0d9892e9c5d9f88eda2486275 Mon Sep 17 00:00:00 2001 From: Henrique Date: Sun, 17 Nov 2019 11:26:20 -0500 Subject: [PATCH] Added coverage for the changes made --- test/mitmproxy/test_command.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/mitmproxy/test_command.py b/test/mitmproxy/test_command.py index f5a641a87..eb3857bf5 100644 --- a/test/mitmproxy/test_command.py +++ b/test/mitmproxy/test_command.py @@ -367,6 +367,8 @@ def test_simple(): c.add("one.two", a.cmd1) assert(c.commands["one.two"].help == "cmd1 help") assert(c.execute("one.two foo") == "ret foo") + assert(c.execute("one.two \"foo\"") == "ret foo") + assert(c.execute("one.two \"foo bar\"") == "ret \"foo bar\"") assert(c.call("one.two", "foo") == "ret foo") with pytest.raises(exceptions.CommandError, match="Unknown"): c.execute("nonexistent") @@ -382,6 +384,8 @@ def test_simple(): c.execute(r"\'") with pytest.raises(exceptions.CommandError, match="Unknown"): c.execute(r"\"") + with pytest.raises(exceptions.CommandError, match="Unknown"): + c.execute(r"\"") c.add("empty", a.empty) c.execute("empty")