def load(self, fp):
dcluster = yaml.load(fp)
from elasticluster import Cluster
cluster = Cluster(**dcluster)
cluster.repository = self
return cluster
def load(self, fp):
dcluster = yaml.load(fp)
from elasticluster import Cluster
cluster = Cluster(**dcluster)
cluster.repository = self
return cluster
def load(self, fp):
data = yaml.load(fp)
if not data:
raise ClusterError("Empty yml file: {0}.".format(fp.name))
from elasticluster import Cluster
cluster = Cluster(**data)
cluster.repository = self
return cluster
def load(self, fp):
dcluster = json.load(fp)
dcluster = dcluster if dcluster else {}
from elasticluster import Cluster
cluster = Cluster(**dcluster)
cluster.repository = self
return cluster
def load(self, fp):
dcluster = yaml.load(fp)
if not dcluster:
raise ClusterNotFound('create a cluster')
from elasticluster import Cluster
cluster = Cluster(**dcluster)
cluster.repository = self
return cluster
def load(self, fp):
dcluster = json.load(fp)
dcluster = dcluster if dcluster else {}
from elasticluster import Cluster
cluster = Cluster(**dcluster)
cluster.repository = self
return cluster
def load(self, fp):
dcluster = yaml.load(fp)
if not dcluster:
raise ClusterNotFound("create a cluster")
from elasticluster import Cluster
cluster = Cluster(**dcluster)
cluster.repository = self
return cluster
def test_saving_cluster_with_nodes(self):
cluster = Cluster(
name="test1", cloud_provider=None, setup_provider=None, user_key_name="key", repository=self.storage
)
cluster.add_node(
kind="foo", image_id="123", image_user="******", flavor="m1.tiny", security_group="default", name="foo123"
)
self.storage.save_or_update(cluster)
new = self.storage.get(cluster.name)
assert "foo" in cluster.nodes
assert len(cluster.nodes["foo"]) > 0
assert isinstance(cluster.nodes["foo"][0], Node)
assert cluster.nodes["foo"][0].name == "foo123"
def test_saving_cluster_with_nodes(self):
cluster = Cluster(name='test1',
cloud_provider=None,
setup_provider=None,
user_key_name='key',
repository=self.storage)
cluster.add_node(kind='foo', image_id='123',
image_user='******', flavor='m1.tiny',
security_group='default', name='foo123')
self.storage.save_or_update(cluster)
new = self.storage.get(cluster.name)
nt.assert_true('foo' in cluster.nodes)
nt.assert_true(cluster.nodes['foo'], 1)
nt.assert_true(cluster.nodes['foo'][0].name, 'foo123')
def test_delete(self):
cluster = Cluster('test1')
self.storage.save_or_update(cluster)
assert cluster.name in self.storage.clusters
self.storage.delete(cluster)
assert cluster.name not in self.storage.clusters
def test_save_and_delete(self):
cluster = Cluster('test1')
self.storage.save_or_update(cluster)
clusterpath = os.path.join(self.path, 'test1.pickle')
assert os.path.exists(clusterpath)
self.storage.delete(cluster)
assert not os.path.exists(clusterpath)
def test_get(self):
clusters = [Cluster('test_%d' % i) for i in range(10)]
for cluster in clusters:
self.storage.save_or_update(cluster)
new_clusters = [self.storage.get(cluster.name) for cluster in clusters]
for cluster in new_clusters:
assert cluster in clusters
def test_get_all(self):
clusters = [Cluster(name='test_%d' % i,
cloud_provider=None,
setup_provider=None,
user_key_name='key') for i in range(10)]
for cluster in clusters:
self.storage.save_or_update(cluster)
new_clusters = self.storage.get_all()
for cluster in new_clusters:
nt.assert_true(cluster.name in [c.name for c in clusters])
def test_save_and_delete(self):
cluster = Cluster(name='test1',
cloud_provider=None,
setup_provider=None,
user_key_name='key')
self.storage.save_or_update(cluster)
clusterpath = os.path.join(self.path, 'test1.yaml')
assert os.path.exists(clusterpath)
self.storage.delete(cluster)
assert not os.path.exists(clusterpath)
def store_is_of_type(self, store_type):
"""Check store type %s""" % store_type
storage = MultiDiskRepository(self.path, default_store=store_type)
cluster = Cluster(name='test1',
cloud_provider=None,
setup_provider=None,
user_key_name='key',
repository=storage)
storage.save_or_update(cluster)
assert os.path.exists(
os.path.join(self.path, cluster.name + '.' + store_type))
def test_save_and_delete(self):
cluster = Cluster(name='test1',
cloud_provider=None,
setup_provider=None,
user_key_name='key')
self.storage.save_or_update(cluster)
clusterpath = os.path.join(self.path, 'test1.json')
nt.assert_true(os.path.exists(clusterpath))
self.storage.delete(cluster)
nt.assert_false(os.path.exists(clusterpath))
def load(self, fp):
data = json.load(fp)
from elasticluster import Cluster
cluster = Cluster(**data)
cluster.repository = self
return cluster
def load(self, fp):
data = json.load(fp)
from elasticluster import Cluster
cluster = Cluster(**data)
cluster.repository = self
return cluster
