def factory(app, db, config, cluster):
"""This method returns the appropriate EventImporter object depending
on what is specified in configuration. In case of configuration
error, HPCStatsConfigurationException is raised.
"""
implem = config.get(cluster.name, 'events')
if implem == "slurm":
return EventImporterSlurm(app, db, config, cluster)
else:
raise HPCStatsConfigurationException( \
"EventImporter %s is not implemented" \
% (implem))
def setUp(self):
# setup conf
self.filename = 'fake'
self.cluster = Cluster('testcluster')
HPCStatsConf.__bases__ = (MockConfigParser, object)
self.conf = HPCStatsConf(self.filename, self.cluster)
self.conf.conf = CONFIG.copy()
# setup importer
self.db = HPCStatsDB(self.conf)
self.db.bind()
self.app = MockApp(self.db, self.conf, self.cluster)
self.importer = EventImporterSlurm(self.app, self.db, self.conf,
self.cluster)
init_reqs()
# setup logger
logging.setLoggerClass(HPCStatsLogger)
self.logger = logging.getLogger(__name__)
self.handler = MockLoggingHandler()
self.logger.addHandler(self.handler)
self.handler.reset()
HPCStatsLogger.set_error_mgr(HPCStatsErrorMgr(self.conf))