def test_incorrect_item(self): """Unlisted commands should not be allowed (string whitelist).""" message = ReceivedMessage(b'WRONG_COMMAND\r\n') wrapped = filters.allow('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.allow('COMMAND')(self.handler) wrapped(self.client, message) self.handler.assert_called_once_with( client=self.client, message=message, )