def test_create_yarn_cluster(self): # arrange expected_masterip = '127.0.0.1' expected_vm_dict = [{'name': 'f_vm', 'adminPass': '******'}, {'name': 'f_2 vm', 'adminPass': '******'}] c_yarn_cluster = YarnCluster(self.opts) # act returned_masterip, returned_vm_dict = c_yarn_cluster.create_yarn_cluster() # assert self.assertTupleEqual((expected_masterip, expected_vm_dict), (returned_masterip, returned_vm_dict))
raise ClientError(msg,error_fatal) if not skip_cluster_create: try: cluster_options = check_cluster_options(cluster_options,dsl,token,map_dsl_to_cluster) except ClientError,e: msg = str(e.args[0]) action_stop(token, id, current_task, msg) raise e # cluster section # only need to import create cluster if we are going to be making a cluster from backend.create_cluster import YarnCluster state_msg = 'Started experiment cluster creation' action_continue(token, id, current_task, state_msg) c_cluster = YarnCluster(cluster_options) MASTER_IP, servers, password, cluster_id = c_cluster.create_yarn_cluster() if cluster_id > 0: state_msg = 'Experiment cluster created successfully' action_continue(token, id, current_task, state_msg) cluster_id_for_replay = cluster_id else: state_msg = 'Experiment cluster creation failed' action_stop(token, id, current_task, state_msg) raise ClientError(state_msg, error_fatal) # actions section actions = dsl_data_dict.get('actions',[]) if len(actions)>0 and cluster_id_for_replay is not None: try: actions = check_actions(actions, dsl, token) except ClientError,e: