def assert_ok_pidbox_response(self, replies): for reply in values(replies): if not reply['ok']: raise RuntimeError( 'Worker remote control command raised: {0!r}'.format( reply.get('error', reply))) return replies
def __enter__(self): # The __warningregistry__'s need to be in a pristine state for tests # to work properly. warnings.resetwarnings() for v in list(values(sys.modules)): # do not evaluate Django moved modules and other lazily # initialized modules. if v and not _is_magic_module(v): # use raw __getattribute__ to protect even better from # lazily loaded modules try: object.__getattribute__(v, '__warningregistry__') except AttributeError: pass else: object.__setattr__(v, '__warningregistry__', {}) self.warnings_manager = warnings.catch_warnings(record=True) self.warnings = self.warnings_manager.__enter__() warnings.simplefilter('always', self.expected) return self
def override_worker_setting(self, setting_name, new_value): old_value = list(values(self.setenv(setting_name, new_value)))[0] try: yield finally: self.setenv(setting_name, old_value)
def __call__(self, model): [event.connect_model(model) for event in values(self.events) if event] model.webhook_events = self return model
def contribute_to_model(self, model): # type: (type) -> type [event.connect_model(model) for event in values(self.events) if event] model.webhooks = self model.webhook_events = self # XXX remove for Thorn 2.0 return model