示例#1
0
 def setUp(self):
     self.filename = 'fake'
     self.cluster = Cluster('testcluster')
     HPCStatsConf.__bases__ = (MockConfigParser, object)
     self.conf = HPCStatsConf(self.filename, self.cluster.name)
     self.conf.conf = CONFIG
     self.app = None
     self.db = None
     self.importer = BusinessCodeImporterCSV(self.app, self.db, self.conf)
     init_reqs()
    def factory(app, db, config):
        """This method returns the appropriate BusinessCodeImporter object
           depending on what is specified in configuration. In case of
           configuration error, HPCStatsConfigurationException is raised.
        """

        implem = config.get_default('globals', 'business', 'dummy')

        if implem == 'dummy':
            return BusinessCodeImporterDummy(app, db, config)
        elif implem == 'slurm':
            return BusinessCodeImporterSlurm(app, db, config)
        elif implem == 'csv':
            return BusinessCodeImporterCSV(app, db, config)
        else:
            raise HPCStatsConfigurationException( \
                    "BusinessCodeImporter %s is not implemented" \
                      % (implem))