def test_incorrect_item(self):
"""Unlisted commands should not be allowed (string whitelist)."""
message = ReceivedMessage(b'WRONG_COMMAND\r\n')
wrapped = filters.command_whitelist('COMMAND')(self.handler)
wrapped(self.client, message)
self.assertFalse(self.handler.called)
def test_correct_item(self):
"""Whitelisted commands should be allowed (string whitelist)."""
message = ReceivedMessage(b'COMMAND\r\n')
wrapped = filters.command_whitelist('COMMAND')(self.handler)
wrapped(self.client, message)
self.handler.assert_called_once_with(
client=self.client,
message=message,
)
