def test_incorrect_item(self):
"""Blacklisted commands should not be allowed (string blacklist)."""
message = ReceivedMessage(b'WRONG_COMMAND\r\n')
wrapped = filters.command_blacklist('WRONG_COMMAND')(self.handler)
wrapped(self.client, message)
self.assertFalse(self.handler.called)
def test_correct_item(self):
"""Un-blacklisted commands should be allowed (string blacklist)."""
message = ReceivedMessage(b'COMMAND\r\n')
wrapped = filters.command_blacklist('WRONG_COMMAND')(self.handler)
wrapped(self.client, message)
self.handler.assert_called_once_with(
client=self.client,
message=message,
)
