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