def get_cluster(self, cloud_provider=None, config=None, nodes=None):
if not cloud_provider:
cloud_provider = BotoCloudProvider("https://hobbes.gc3.uzh.ch/",
"nova", "a-key", "s-key")
if not config:
config = Configuration().get_config(self.path)
setup = Mock()
configurator = Configurator(config)
conf_login = configurator.cluster_conf['mycluster']['login']
repository = PickleRepository(self.storage_path)
cluster = Cluster(
name="mycluster",
cloud_provider=cloud_provider,
setup_provider=setup,
repository=repository,
user_key_name=conf_login['user_key_name'],
user_key_public=conf_login['user_key_public'],
user_key_private=conf_login['user_key_private'],
)
if not nodes:
nodes = {"compute": 2, "frontend": 1}
for kind, num in nodes.iteritems():
conf_kind = configurator.cluster_conf['mycluster']['nodes'][kind]
cluster.add_nodes(kind, num, conf_kind['image_id'],
conf_login['image_user'], conf_kind['flavor'],
conf_kind['security_group'])
return cluster
def test_get_frontend_node(self):
"""
Get frontend node
"""
config = Configuration().get_config(self.path)
ssh_to = "frontend"
config["mycluster"]["cluster"]["ssh_to"] = ssh_to
cluster = self.get_cluster(config=config)
cluster.ssh_to = ssh_to
frontend = cluster.get_frontend_node()
self.assertEqual(cluster.nodes['frontend'][0], frontend)
def test_dict_mixin(self):
"""Check that the node class can be seen as dictionary"""
config = Configuration().get_config(self.path)
ssh_to = "frontend"
config["mycluster"]["cluster"]["ssh_to"] = ssh_to
cluster = self.get_cluster(config=config)
cluster.ssh_to = ssh_to
frontend = cluster.get_frontend_node()
dcluster = dict(cluster)
self.assertEqual(dcluster['ssh_to'], ssh_to)
self.assertEqual(dcluster['nodes'].keys(), cluster.nodes.keys())
self.failUnlessRaises(KeyError, lambda x: x['_cloud_provider'],
dcluster)
self.assertEqual(cluster['_cloud_provider'], cluster._cloud_provider)
def setUp(self):
file, path = tempfile.mkstemp()
self.path = path
self.config = Configuration().get_config(self.path)