def create_cluster_async(choices):
"""
Asynchronous create cluster task.
"""
new_yarn_cluster = YarnCluster(choices)
MASTER_IP, servers, password = new_yarn_cluster.create_yarn_cluster()
task_result = {"master_IP": MASTER_IP, "master_VM_password": password}
return task_result
def test_create_yarn_cluster(self):
# arrange
self.opts['yarn'] = True
expected_masterip = '127.0.0.1'
expected_vm_dict = {1: 'f vm'}
c_yarn_cluster = YarnCluster(self.opts)
del(self.opts['yarn'])
# 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))
def test_create_yarn_cluster(self):
# arrange
opts = {'name': 'Test', 'clustersize': 2, 'cpu_master': 2,
'ram_master': 4096, 'disk_master': 5, 'cpu_slave': 2,
'ram_slave': 2048, 'disk_slave': 5, 'token': self.token,
'disk_template': 'ext_vlmc', 'image': 'Debian Base',
'auth_url': self.auth_url, 'yarn': True}
expected_masterip = '127.0.0.1'
expected_vm_dict = {1: 'f vm'}
c_yarn_cluster = YarnCluster(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))
def create_cluster_async(choices):
"""
Asynchronous create cluster task.
"""
new_yarn_cluster = YarnCluster(choices)
MASTER_IP, servers, password, cluster_id = new_yarn_cluster.create_yarn_cluster(
)
task_result = {
"master_IP": MASTER_IP,
"master_VM_password": password,
"cluster_id": cluster_id
}
return task_result
