From e63bb8cde5a62c7b0ad0b6e6577e8e1a78e822f5 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Wed, 13 Dec 2017 14:41:46 +1300 Subject: commands: add a Path argument type This is just an alias for str, and in this patch is used mostly to give an appropriate type in help strings. More to come. Fixes #2198 --- test/mitmproxy/test_command.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'test') diff --git a/test/mitmproxy/test_command.py b/test/mitmproxy/test_command.py index cb9dc4ed..abb09cea 100644 --- a/test/mitmproxy/test_command.py +++ b/test/mitmproxy/test_command.py @@ -35,6 +35,9 @@ class TAddon: def choose(self, arg: TChoice) -> typing.Sequence[str]: # type: ignore return ["one", "two", "three"] + def path(self, arg: command.Path) -> None: + pass + class TestCommand: def test_varargs(self): @@ -97,6 +100,7 @@ def test_typename(): assert command.typename(typing.Sequence[str], False) == "[str]" assert command.typename(TChoice, False) == "choice" + assert command.typename(command.Path, False) == "path" class DummyConsole: @@ -156,6 +160,10 @@ def test_parsearg(): tctx.master.commands, "invalid", TChoice, ) + assert command.parsearg( + tctx.master.commands, "foo", command.Path + ) == "foo" + class TDec: @command.command("cmd1") -- cgit v1.2.3