def test_with_global_context(self):
with Diagram(name=os.path.join(self.name, "with_global_context"),
show=False) as d1:
self.assertIsNone(getcluster())
with Cluster():
self.assertIsNotNone(getcluster())
self.assertEqual(d1, getcluster())
def test_with_global_context(self):
with Diagram(name=os.path.join(self.name, 'with_global_context'),
show=False):
self.assertIsNone(getcluster())
with Cluster():
self.assertIsNotNone(getcluster())
self.assertIsNone(getcluster())
def test_with_nested_cluster(self):
with Diagram(name=self.name, show=False):
self.assertIsNone(getcluster())
with Cluster() as c1:
self.assertEqual(c1, getcluster())
with Cluster() as c2:
self.assertEqual(c2, getcluster())
self.assertEqual(c1, getcluster())
self.assertIsNone(getcluster())
def test_with_nested_cluster(self):
with Diagram(name=os.path.join(self.name, "with_nested_cluster"),
show=False) as d1:
self.assertIsNone(getcluster())
with Cluster() as c1:
self.assertEqual(c1, getcluster())
with Cluster() as c2:
self.assertEqual(c2, getcluster())
self.assertEqual(c1, getcluster())
self.assertEqual(d1, getcluster())
def test_node_as_cluster(self):
with Diagram(name=os.path.join(self.name, "test_node_as_cluster"),
show=False) as d1:
node1 = Node("node1")
with EC2("node-as-cluster") as c1:
node2 = Node("node2")
self.assertEqual(c1, getcluster())
