def test_parser_called_with_kwarg(self):
"""
The parser is called with `message=message`.
It's not really a big deal, but it's slightly more flexible if the
parser is called with kwargs.
"""
parser = mock.Mock(return_value={})
wrapped = apply_message_parser(parser)(self.handler)
wrapped(client=self.client, message=self.message)
parser.assert_called_once_with(message=self.message)
def test_result_passed(self):
"""Dictionary returned from parser passed as kwargs."""
wrapped = apply_message_parser(parser_taking_message)(self.handler)
wrapped(client=self.client, message=self.message)
self.handler.assert_called_once_with(
client=self.client,
message=self.message,
key='value',
)
