Skip to content

Commit

Permalink
Adding back Cluster.reload() test to system tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
dhermes committed Jul 28, 2015
1 parent 98a878f commit 93bec1a
Showing 1 changed file with 37 additions and 3 deletions.
40 changes: 37 additions & 3 deletions system_tests/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,27 @@ def setUpModule():
EXISTING_CLUSTERS[:] = clusters


class TestClient(unittest2.TestCase):
class TestClusterAdminAPI(unittest2.TestCase):

@classmethod
def setUpClass(cls):
# Create a cluster which will remain throughout tests.
cluster = CLIENT.cluster(TEST_ZONE, TEST_CLUSTER_ID,
display_name=TEST_CLUSTER_ID)
cluster.create()
cls._cluster = cluster

@classmethod
def tearDownClass(cls):
cls._cluster.delete()
del cls._cluster

def setUp(self):
self.clusters_to_delete = []

def tearDown(self):
for cluster in self.clusters_to_delete:
cluster.delete()

def test_list_zones(self):
zones = CLIENT.list_zones()
Expand All @@ -55,6 +75,20 @@ def test_list_zones(self):
def test_list_clusters(self):
clusters, failed_zones = CLIENT.list_clusters()
self.assertEqual(failed_zones, [])
self.assertEqual(len(clusters), len(EXISTING_CLUSTERS))
# We have added one new cluster in `setUpClass`.
self.assertEqual(len(clusters), len(EXISTING_CLUSTERS) + 1)
for cluster in clusters:
self.assertTrue(cluster in EXISTING_CLUSTERS)
cluster_existence = (cluster in EXISTING_CLUSTERS or
cluster == self._cluster)
self.assertTrue(cluster_existence)

def test_reload(self):
# Use same arguments as self._cluster (created in `setUpClass`).
cluster = CLIENT.cluster(TEST_ZONE, TEST_CLUSTER_ID)
# Make sure metadata unset before reloading.
cluster.display_name = None
cluster.serve_nodes = None

cluster.reload()
self.assertEqual(cluster.display_name, self._cluster.display_name)
self.assertEqual(cluster.serve_nodes, self._cluster.serve_nodes)

0 comments on commit 93bec1a

Please sign in to comment.