def test_invoke_primitive_str(self):
@test_utils.mock
def do_test(self, str_var: str):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(str_var='str'):
invoker.invoke(self, cmdline=CommandLine('test str'))
def test_invoke_primitive_int(self):
@test_utils.mock
def do_test(self, int_var: int):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(int_var=42):
invoker.invoke(self, cmdline=CommandLine('test 42'))
def test_invoke_mixed_free_named(self):
@test_utils.mock
def do_test(self, first: str, second: int):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(first='first', second=2):
invoker.invoke(self, cmdline=CommandLine('test first second=2'))
def test_invoke_list(self):
@test_utils.mock
def do_test(self, arg: List[int]):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(arg=[3, 42]):
invoker.invoke(self, cmdline=CommandLine('test arg=[3,42]'))
def test_invoke_primitive_int(self):
@test_utils.mock
def do_test(self, int_var: int):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(int_var=42):
invoker.invoke(self, cmdline=CommandLine('test 42'))
def test_invoke_primitive_str(self):
@test_utils.mock
def do_test(self, str_var: str):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(str_var='str'):
invoker.invoke(self, cmdline=CommandLine('test str'))
def test_construct_tuple(self):
@test_utils.mock
def do_test(self, arg: Tuple[float, str]):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(arg=(float(3), 'foo')):
invoker.invoke(self, cmdline=CommandLine('test arg=(3,foo)'))
def test_construct_tuple(self):
@test_utils.mock
def do_test(self,
arg: Tuple[float, str]):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(arg=(float(3), 'foo')):
invoker.invoke(self, cmdline=CommandLine('test arg=(3,foo)'))
def test_invoke_list(self):
@test_utils.mock
def do_test(self,
arg: List[int]):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(arg=[3, 42]):
invoker.invoke(self, cmdline=CommandLine('test arg=[3,42]'))
def test_invoke_free_args(self):
@test_utils.mock
def do_test(self,
first: str,
second: int):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(first='first',
second=2):
invoker.invoke(self, cmdline=CommandLine('test first 2'))
def test_invoke_class(self):
class ClassConstructor(str):
pass
@test_utils.mock
def do_test(self, cls_var: ClassConstructor):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(cls_var=ClassConstructor('cls')):
invoker.invoke(self, cmdline=CommandLine('test cls'))
def test_invoke_class(self):
class ClassConstructor(str):
pass
@test_utils.mock
def do_test(self,
cls_var: ClassConstructor):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(cls_var=ClassConstructor('cls')):
invoker.invoke(self, cmdline=CommandLine('test cls'))
def test_invoke_mixed_free_named(self):
@test_utils.mock
def do_test(self,
first: str,
second: int):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(first='first',
second=2):
invoker.invoke(self, cmdline=CommandLine('test first second=2'))
with do_test.expect_no_calls(), self.assertRaises(InvalidInput):
invoker.invoke(self, cmdline=CommandLine('test first=first 2'))
def default(self, cmdline):
"""
Interprets CMDLINE as a command and executes it.
"""
try:
if not cmdline:
return self.emptyline()
invoker = CommandInvoker(self._get_all_commands())
return invoker.invoke(self, cmdline=CommandLine(cmdline))
# it's a bit too ruthless to terminate on every single broken command
# pylint: disable=broad-except
except Exception as e:
self._last_exception = sys.exc_info()
print('%s (try "get_error" for details)' % e)
else:
self._last_exception = None
def default(self, cmdline):
"""
Interprets CMDLINE as a command and executes it.
"""
try:
if not cmdline:
return self.emptyline()
invoker = CommandInvoker(self._get_all_commands())
return invoker.invoke(self, cmdline=CommandLine(cmdline))
# it's a bit too ruthless to terminate on every single broken command
# pylint: disable=broad-except
except Exception as e:
self._last_exception = sys.exc_info()
print('%s (try "get_error" for details)' % e)
else:
self._last_exception = None
def test_construct_union(self):
@test_utils.mock
def do_test(self, arg: Union[float, str]):
pass
cmds = CommandsDict()
cmds['test'] = Command('test', do_test)
invoker = CommandInvoker(cmds)
with do_test.expect_call(arg=3.14):
invoker.invoke(self, cmdline=CommandLine('test arg=3.14'))
with do_test.expect_call(arg='test_arg'):
invoker.invoke(self, cmdline=CommandLine('test arg=test_arg'))