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
)
