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