def test_get_service_by_role(self, get_cloudera_cluster):
class Ob(object):
def get_service(self, x):
return x
get_cloudera_cluster.return_value = Ob()
roles = [
'NAMENODE', 'DATANODE', 'SECONDARYNAMENODE', 'HDFS_GATEWAY',
'RESOURCEMANAGER', 'NODEMANAGER', 'JOBHISTORY', 'YARN_GATEWAY',
'OOZIE_SERVER', 'HIVESERVER2', 'HIVEMETASTORE', 'WEBHCAT',
'HUE_SERVER', 'SPARK_YARN_HISTORY_SERVER', 'SERVER', 'MASTER',
'REGIONSERVER'
]
resps = [
'hdfs01', 'hdfs01', 'hdfs01', 'hdfs01', 'yarn01', 'yarn01',
'yarn01', 'yarn01', 'oozie01', 'hive01', 'hive01', 'hive01',
'hue01', 'spark_on_yarn01', 'zookeeper01', 'hbase01', 'hbase01'
]
provider = cu.ClouderaUtilsV530()
cluster = mock.Mock()
for (role, resp) in zip(roles, resps):
self.assertEqual(
resp, provider.get_service_by_role(role, cluster=cluster))
with testtools.ExpectedException(ValueError):
provider.get_service_by_role('cat', cluster=cluster)
def __init__(self):
super(VersionHandler, self).__init__()
self.config_helper = config_helper.ConfigHelperV530()
self.cloudera_utils = cloudera_utils.ClouderaUtilsV530()
self.plugin_utils = plugin_utils.PluginUtilsV530()
self.deploy = deploy
self.edp_engine = edp_engine
self.validation = validation.ValidatorV530()
from oslo_log import log as logging
from sahara import conductor
from sahara import context
from sahara.plugins.cdh import abstractversionhandler as avm
from sahara.plugins.cdh.v5_3_0 import cloudera_utils as cu
from sahara.plugins.cdh.v5_3_0 import config_helper as c_helper
from sahara.plugins.cdh.v5_3_0 import deploy as dp
from sahara.plugins.cdh.v5_3_0 import edp_engine
from sahara.plugins.cdh.v5_3_0 import validation as vl
conductor = conductor.API
LOG = logging.getLogger(__name__)
CONF = cfg.CONF
CU = cu.ClouderaUtilsV530()
class VersionHandler(avm.AbstractVersionHandler):
def get_plugin_configs(self):
return c_helper.get_plugin_configs()
def get_node_processes(self):
return {
"CLOUDERA": ['CLOUDERA_MANAGER'],
"HDFS": [],
"NAMENODE": ['HDFS_NAMENODE'],
"DATANODE": ['HDFS_DATANODE'],
"SECONDARYNAMENODE": ['HDFS_SECONDARYNAMENODE'],
"YARN": [],
def __init__(self, cluster):
super(EdpOozieEngine, self).__init__(cluster)
self.cloudera_utils = cu.ClouderaUtilsV530()
