def test_check_all_resources(self):
# arrange
c_yarn_cluster = YarnCluster(self.opts)
expected = 0
# act
returned = c_yarn_cluster.check_all_resources()
# assert
self.assertTrue(True) # temporarily short-ciruit
def test_check_disk_valid_sufficient(self):
# arrange
c_yarn_cluster = YarnCluster(self.opts)
expected = 0
# act
returned = c_yarn_cluster.check_disk_valid()
# assert
self.assertEqual(expected, returned)
def test_check_network_quotas_sufficient(self):
# arrange
c_yarn_cluster = YarnCluster(self.opts)
expected = 0 # success
# act
returned = c_yarn_cluster.check_network_quotas()
# assert
self.assertEqual(expected, returned)
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))
def test_check_cpu_valid_sufficient(self):
# arrange
prev_clustersize = self.opts['cluster_size']
self.opts['cluster_size'] = 2
c_yarn_cluster = YarnCluster(self.opts)
expected = 0 # usage: 5, limit: 9 (4 remaining), requested: 4, expected result success
# act
returned = c_yarn_cluster.check_cpu_valid()
self.opts['cluster_size'] = prev_clustersize
# assert
self.assertEqual(expected, returned)
def test_check_disk_valid_exceeded(self):
# arrange
prev_clustersize = self.opts['cluster_size']
self.opts['cluster_size'] = 100
c_yarn_cluster = YarnCluster(self.opts)
expected = error_quotas_cyclades_disk
# act
with self.assertRaises(ClientError) as context:
c_yarn_cluster.check_disk_valid()
self.opts['cluster_size'] = prev_clustersize
# assert
the_exception = context.exception
self.assertEqual(expected, the_exception.status)
