def load_cloud_driver(is_fake=False): if is_fake: import_path = "pumphouse.fake.{}" else: import_path = "pumphouse.cloud.{}" cloud_driver = utils.load_class(import_path.format("Cloud")) identity_driver = utils.load_class(import_path.format("Identity")) return cloud_driver, identity_driver
def init_cloud_service(self, app): config = app.config LOG.info("Identity driver will be used: %s", config["IDENTITY_DRIVER"]) identity_driver = utils.load_class(config["IDENTITY_DRIVER"]) LOG.info("Cloud driver will be used: %s", config["CLOUD_DRIVER"]) cloud_driver = utils.load_class(config["CLOUD_DRIVER"]) LOG.info("Cloud initializer will be used: %s", config["CLOUD_SERVICE"]) cloud_config = config["CLOUDS"][self.target].copy() cloud_service = utils.load_class(config["CLOUD_SERVICE"]) plugins = config.get("PLUGINS", {}) service = cloud_service(cloud_config, plugins, self.target, cloud_driver, identity_driver) return service
def test_load_class_exc(self): with self.assertRaises(ImportError) as ec: utils.load_class(self.incorrect_import_path) self.assertEqual(ec.exception.args[0], "No module named dummy")
def test_load_class(self): self.assertEqual( utils.load_class(self.correct_import_path), DummyClass )