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 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 __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 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 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
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
