def _warn_if_reactor_is_not_running(cls): if reactor.running or util.in_unittest(): return msg = 'connection attempted without running Twisted reactor' detail = ('The reconnect mechanism is based on having a running Twisted reactor. ' 'Either ensure that Twisted is running, don\'t expect reconnecting to happen or ' 'implement a checker thread.') warning = exceptions.PipedWarning(msg, detail) warnings.warn(warning)
def _plugin_error_handler(package): f = failure.Failure() f.trap(ImportError) logger = log.error # set the logger to info when running unit tests in order to avoid flooding if util.in_unittest(): logger = log.info logger('Failed loading plugins from package %r: %s(%s)' % (package.__name__, f.type.__name__, f.getErrorMessage()))
def configure(self, runtime_environment): self.runtime_environment = runtime_environment self.dependency_manager = runtime_environment.dependency_manager self.persisted_contexts = runtime_environment.get_configuration_value('persisted_contexts', dict()) # State that we can provide all configured contexts. resource_manager = runtime_environment.resource_manager for context_name, context in self.persisted_contexts.items(): resource_manager.register('persisted_context.%s' % context_name, provider=self) # Hook up the shutdown-event if not util.in_unittest(): reactor.addSystemEventTrigger('before', 'shutdown', self.persist_contexts)
def _load_config(self, config, visited_files): """ Handles the configuration object, including referenced configurations as specified. Also handles replacement of special options like runmodes. """ current_configuration_file = visited_files[-1] # if a configuration file does not contain anything, yaml.load returns None, # but we consider an empty configuration file to be an empty dict instead if config is None: config = dict() self._fail_if_configuration_is_invalid(config, current_configuration_file) includes = config.get('includes', dict()) config = self._load_includes(config, includes, visited_files) # see if we have any special options for this mode if 'runmodes' in config: # currently we only support "unittest" as the runmode: if util.in_unittest(): config = util.merge_dicts(config, config['runmodes'].get('unittest', dict()), replace_primitives=True) return config
def test_trial_detection(self): self.assertTrue(util.in_unittest(), 'Failed to detect that we are running trial')