def test__replaces_missing_ask_with_none(self):
box = amp.AmpBox(_command=b"command")
self.patch(common, "gethostname").return_value = "host"
self.patch(common, "getpid").return_value = 1234
self.assertThat(common.make_command_ref(box), Equals(
"host:pid=1234:cmd=command:ask=none"))
def test__command_ref_includes_host_pid_command_and_ask_sequence(self):
host = factory.make_name("hostname")
pid = random.randint(99, 9999999)
cmd = factory.make_name("command").encode("ascii")
ask = str(random.randint(99, 9999999)).encode("ascii")
box = amp.AmpBox(_command=cmd, _ask=ask)
self.patch(common, "gethostname").return_value = host
self.patch(common, "getpid").return_value = pid
self.assertThat(
common.make_command_ref(box), Equals("%s:pid=%s:cmd=%s:ask=%s" % (
host, pid, cmd.decode("ascii"), ask.decode("ascii"))))
def test_unhandled_errors_do_not_cause_disconnection(self):
self.patch(common.log, "debug")
protocol = common.RPCProtocol()
protocol.makeConnection(StringTransport())
# Ensure that the superclass dispatchCommand() will fail.
dispatchCommand = self.patch(amp.AMP, "dispatchCommand")
dispatchCommand.side_effect = always_fail_with(ZeroDivisionError())
# Push a command box into the protocol.
seq = b"%d" % random.randrange(0, 2**32)
cmd = factory.make_string().encode("ascii")
box = amp.AmpBox(_ask=seq, _command=cmd)
with TwistedLoggerFixture() as logger:
protocol.ampBoxReceived(box)
# The transport is still connected.
self.expectThat(protocol.transport.disconnecting, Is(False))
# The error has been logged on the originating side of the AMP
# session, along with an explanatory message. The message includes a
# command reference.
cmd_ref = common.make_command_ref(box)
self.assertDocTestMatches(
"""\
Unhandled failure dispatching AMP command. This is probably a bug.
Please ensure that this error is handled within application code
or declared in the signature of the %s command. [%s]
Traceback (most recent call last):
...
""" % (cmd, cmd_ref),
logger.output,
)
# A simpler error message has been transmitted over the wire. It
# includes the same command reference as logged locally.
protocol.transport.io.seek(0)
observed_boxes_sent = amp.parse(protocol.transport.io)
expected_boxes_sent = [
amp.AmpBox(
_error=seq,
_error_code=amp.UNHANDLED_ERROR_CODE,
_error_description=(b"Unknown Error [%s]" %
cmd_ref.encode("ascii")),
)
]
self.assertThat(observed_boxes_sent, Equals(expected_boxes_sent))