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