def test_command_arguments(test):
"""Tests command argument parsing."""
process = sh.test()
assert process.command() == ["test"]
assert str(process) == "test"
assert bytes(process) == psys.b(str(process))
process = sh.complex_command_name()
assert process.command() == ["complex-command-name"]
assert str(process) == "complex-command-name"
assert bytes(process) == psys.b(str(process))
process = sh("complex command name")("arg1", "arg2")
assert process.command() == ["complex command name", "arg1", "arg2"]
assert str(process) == "'complex command name' arg1 arg2"
assert bytes(process) == psys.b(str(process))
process = sh.test(b"arg", b"space arg", b"carriage\rline", b"line\narg",
b"tab\targ", br"slash\arg", b"quote'arg", b'quote"arg',
psys.b("тест"), psys.b("тест тест"), "arg", "space arg",
"carriage\rline", "line\narg", "tab\targ", r"slash\arg",
"quote'arg", 'quote"arg', "тест", "тест тест", 3, 2**128,
2.0)
assert process.command() == [
"test", b"arg", b"space arg", b"carriage\rline", b"line\narg",
b"tab\targ", br"slash\arg", b"quote'arg", b'quote"arg',
psys.b("тест"),
psys.b("тест тест"), "arg", "space arg", "carriage\rline", "line\narg",
"tab\targ", r"slash\arg", "quote'arg", 'quote"arg', "тест",
"тест тест", "3", "340282366920938463463374607431768211456", "2.0"
]
assert str(process) == (
"test "
r"""arg 'space arg' 'carriage\rline' 'line\narg' 'tab\targ' 'slash\\arg' "quote'arg" 'quote"arg' \xd1\x82\xd0\xb5\xd1\x81\xd1\x82 '\xd1\x82\xd0\xb5\xd1\x81\xd1\x82 \xd1\x82\xd0\xb5\xd1\x81\xd1\x82' """
r"""arg 'space arg' 'carriage\rline' 'line\narg' 'tab\targ' 'slash\\arg' 'quote\'arg' 'quote"arg' тест 'тест тест' """
"3 340282366920938463463374607431768211456 2.0")
assert bytes(process) == psys.b(str(process))
process = sh.test("space arg", s="short_arg")
assert process.command() == ["test", "-s", "short_arg", "space arg"]
assert str(process) == "test -s short_arg 'space arg'"
assert bytes(process) == psys.b(str(process))
process = sh.test("arg", long_long_arg="long arg")
assert process.command() == ["test", "--long-long-arg", "long arg", "arg"]
assert str(process) == "test --long-long-arg 'long arg' arg"
assert bytes(process) == psys.b(str(process))
process = sh.test("arg", bool_arg=True)
assert process.command() == ["test", "--bool-arg", "arg"]
assert str(process) == "test --bool-arg arg"
assert bytes(process) == psys.b(str(process))
process = sh.test("arg", bool_arg=False)
assert process.command() == ["test", "arg"]
assert str(process) == "test arg"
assert bytes(process) == psys.b(str(process))
def __handle_request(self):
"""Handles a request."""
try:
request = json.loads(self._read_buffer.decode("utf-8"))
if (
"method" not in request or
any(type(key) is not str for key in request.keys()) or
any(type(value) is not str for value in request.values())
):
raise ValueError()
except (UnicodeDecodeError, ValueError):
LOG.error("%s: got an invalid request %s.", self, bytes(self._read_buffer))
self.close()
return
LOG.info("%s: request %s", self, request)
try:
reply = { "result": monitor.request.handle(request.pop("method"), request) }
except Exception as e:
(LOG.warning if isinstance(e, Error) else LOG.error)(
"%s: request failed: %s", self, e)
reply = { "error": str(e) if isinstance(e, Error) else "Internal error" }
response = json.dumps(reply).encode("utf-8")
if self._write(struct.pack(self.__message_size_format, len(response)) + response):
self.close()
def test_command_arguments(test):
"""Tests command argument parsing."""
process = sh.test()
assert process.command() == [ "test" ]
assert str(process) == "test"
assert bytes(process) == psys.b(str(process))
process = sh.complex_command_name()
assert process.command() == [ "complex-command-name" ]
assert str(process) == "complex-command-name"
assert bytes(process) == psys.b(str(process))
process = sh("complex command name")("arg1", "arg2")
assert process.command() == [ "complex command name", "arg1", "arg2" ]
assert str(process) == "'complex command name' arg1 arg2"
assert bytes(process) == psys.b(str(process))
process = sh.test(
b"arg", b"space arg", b"carriage\rline", b"line\narg", b"tab\targ", br"slash\arg", b"quote'arg", b'quote"arg', psys.b("тест"), psys.b("тест тест"),
"arg", "space arg", "carriage\rline", "line\narg", "tab\targ", r"slash\arg", "quote'arg", 'quote"arg', "тест", "тест тест",
3, 2 ** 128, 2.0
)
assert process.command() == [ "test",
b"arg", b"space arg", b"carriage\rline", b"line\narg", b"tab\targ", br"slash\arg", b"quote'arg", b'quote"arg', psys.b("тест"), psys.b("тест тест"),
"arg", "space arg", "carriage\rline", "line\narg", "tab\targ", r"slash\arg", "quote'arg", 'quote"arg', "тест", "тест тест",
"3", "340282366920938463463374607431768211456", "2.0"
]
assert str(process) == ("test "
r"""arg 'space arg' 'carriage\rline' 'line\narg' 'tab\targ' 'slash\\arg' "quote'arg" 'quote"arg' \xd1\x82\xd0\xb5\xd1\x81\xd1\x82 '\xd1\x82\xd0\xb5\xd1\x81\xd1\x82 \xd1\x82\xd0\xb5\xd1\x81\xd1\x82' """
r"""arg 'space arg' 'carriage\rline' 'line\narg' 'tab\targ' 'slash\\arg' 'quote\'arg' 'quote"arg' тест 'тест тест' """
"3 340282366920938463463374607431768211456 2.0"
)
assert bytes(process) == psys.b(str(process))
process = sh.test("space arg", s = "short_arg")
assert process.command() == [ "test", "-s", "short_arg", "space arg" ]
assert str(process) == "test -s short_arg 'space arg'"
assert bytes(process) == psys.b(str(process))
process = sh.test("arg", long_long_arg = "long arg")
assert process.command() == [ "test", "--long-long-arg", "long arg", "arg" ]
assert str(process) == "test --long-long-arg 'long arg' arg"
assert bytes(process) == psys.b(str(process))
process = sh.test("arg", bool_arg = True)
assert process.command() == [ "test", "--bool-arg", "arg" ]
assert str(process) == "test --bool-arg arg"
assert bytes(process) == psys.b(str(process))
process = sh.test("arg", bool_arg = False)
assert process.command() == [ "test", "arg" ]
assert str(process) == "test arg"
assert bytes(process) == psys.b(str(process))
def on_read(self):
"""Called when we are able to read."""
if self.__message_size is None:
if self._read(struct.calcsize(self.__message_size_format)):
self.__message_size, = struct.unpack(
self.__message_size_format, bytes(self._read_buffer))
self._clear_read_buffer()
if self.__message_size > _MAX_REQUEST_SIZE:
self.on_error("Too big message size has been gotten ({0}).",
self.__message_size)
else:
if self._read(self.__message_size):
self.__got_request = True
self.__handle_request()
