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")