class ApplicationTester(object):
"""
Eases the testing of console applications.
"""
def __init__(self, application: Application) -> None:
self._application = application
self._application.auto_exits(False)
self._io = BufferedIO()
self._status_code = 0
@property
def application(self) -> Application:
return self._application
@property
def io(self): # type: () -> BufferedIO
return self._io
@property
def status_code(self): # type: () -> int
return self._status_code
def execute(
self,
args: Optional[str] = "",
inputs: Optional[str] = None,
interactive: Optional[bool] = None,
verbosity: Optional[Verbosity] = None,
decorated: bool = False,
supports_utf8: bool = True,
) -> int:
"""
Executes the command
"""
self._io.clear()
input = StringInput(args)
self._io.set_input(input)
self._io.decorated(decorated)
self._io.output.set_supports_utf8(supports_utf8)
self._io.error_output.set_supports_utf8(supports_utf8)
if inputs is not None:
self._io.input.set_stream(StringIO(inputs))
if interactive is not None:
self._io.interactive(interactive)
if verbosity is not None:
self._io.set_verbosity(verbosity)
self._status_code = self._application.run(
self._io.input,
self._io.output,
self._io.error_output,
)
return self._status_code
class CommandTester(object):
"""
Eases the testing of console commands.
"""
def __init__(self, command: Command) -> None:
self._command = command
self._io = BufferedIO()
self._inputs = []
self._status_code = None
@property
def command(self) -> Command:
return self._command
@property
def io(self) -> BufferedIO:
return self._io
@property
def status_code(self): # type: () -> int
return self._status_code
def execute(
self,
args: Optional[str] = "",
inputs: Optional[str] = None,
interactive: Optional[bool] = None,
verbosity: Optional[Verbosity] = None,
decorated: Optional[bool] = None,
supports_utf8: bool = True,
) -> int:
"""
Executes the command
"""
application = self._command.application
input = StringInput(args)
if application is not None and application.definition.has_argument("command"):
name = self._command.name
if " " in name:
# If the command is namespaced we rearrange
# the input to parse it as a single argument
argv = [application.name, self._command.name] + input._tokens
input = ArgvInput(argv)
else:
input = StringInput(name + " " + args)
self._io.set_input(input)
self._io.output.set_supports_utf8(supports_utf8)
self._io.error_output.set_supports_utf8(supports_utf8)
if inputs is not None:
self._io.input.set_stream(StringIO(inputs))
if interactive is not None:
self._io.interactive(interactive)
if verbosity is not None:
self._io.set_verbosity(verbosity)
if decorated is not None:
self._io.decorated(decorated)
self._status_code = self._command.run(self._io)
return self._status_code
