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)
 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)