def setUp(self):
self.path = tempfile.mkdtemp()
self.storage = YamlRepository(self.path)
def setUp(self):
self.path = tempfile.mkdtemp()
self.storage = YamlRepository(self.path)
class YamlRepositoryTests(unittest.TestCase):
def setUp(self):
self.path = tempfile.mkdtemp()
self.storage = YamlRepository(self.path)
def tearDown(self):
shutil.rmtree(self.path, ignore_errors=True)
del self.storage
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)
]
cluster_names = [c.name for c in clusters]
for cluster in clusters:
self.storage.save_or_update(cluster)
new_clusters = self.storage.get_all()
for cluster in new_clusters:
assert cluster.name in cluster_names
def test_get(self):
clusters = [FakeCluster("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_delete(self):
pass
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 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"
class YamlRepositoryTests(unittest.TestCase):
def setUp(self):
self.path = tempfile.mkdtemp()
self.storage = YamlRepository(self.path)
def tearDown(self):
shutil.rmtree(self.path, ignore_errors=True)
del self.storage
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)]
cluster_names = [c.name for c in clusters]
for cluster in clusters:
self.storage.save_or_update(cluster)
new_clusters = self.storage.get_all()
cluster_names = [c.name for c in clusters]
for cluster in new_clusters:
assert cluster.name in cluster_names
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_delete(self):
pass
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 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'