class TestPlan(unittest.TestCase): def setUp(self): self.plan = Plan() for i in range(4): self.plan.add(generate_definition('vpc', i)) def test_add(self): first_id = 'vpc.1' self.assertIn(first_id, self.plan) self.assertIsInstance(self.plan[first_id], BlueprintContext) def test_status(self): self.assertEqual(len(self.plan.list_submitted()), 0) self.assertEqual(len(self.plan.list_completed()), 0) self.assertEqual(len(self.plan.list_skipped()), 0) self.assertEqual(len(self.plan.list_pending()), 4) self.plan.submit('vpc.1') self.assertEqual(len(self.plan.list_submitted()), 1) self.assertEqual(len(self.plan.list_completed()), 0) self.assertEqual(len(self.plan.list_skipped()), 0) self.assertEqual(len(self.plan.list_pending()), 4) self.plan.complete('vpc.1') self.assertEqual(len(self.plan.list_submitted()), 0) self.assertEqual(len(self.plan.list_completed()), 1) self.assertEqual(len(self.plan.list_skipped()), 0) self.assertEqual(len(self.plan.list_pending()), 3) self.assertFalse(self.plan.completed) self.plan.skip('vpc.2') self.assertEqual(len(self.plan.list_submitted()), 0) self.assertEqual(len(self.plan.list_completed()), 1) self.assertEqual(len(self.plan.list_skipped()), 1) self.assertEqual(len(self.plan.list_pending()), 2) for i in range(4): self.plan.complete("vpc.%d" % i) self.assertTrue(self.plan.completed)
class TestPlan(unittest.TestCase): def setUp(self): self.plan = Plan() for i in range(4): self.plan.add(generate_definition('vpc', i)) def test_add(self): first_id = 'vpc.1' self.assertIn(first_id, self.plan) self.assertIsInstance(self.plan[first_id], BlueprintContext) def test_status(self): self.assertEqual(len(self.plan.list_submitted()), 0) self.assertEqual(len(self.plan.list_completed()), 0) self.assertEqual(len(self.plan.list_pending()), 4) self.plan.submit('vpc.1') self.assertEqual(len(self.plan.list_submitted()), 1) self.assertEqual(len(self.plan.list_completed()), 0) self.assertEqual(len(self.plan.list_pending()), 4) self.plan.complete('vpc.1') self.assertEqual(len(self.plan.list_submitted()), 0) self.assertEqual(len(self.plan.list_completed()), 1) self.assertEqual(len(self.plan.list_pending()), 3) self.assertFalse(self.plan.completed) for i in range(4): self.plan.complete("vpc.%d" % i) self.assertTrue(self.plan.completed)
def test_reset_after_dump(self, *args): plan = Plan(description="Test", sleep_time=0) previous_stack = None for i in range(5): overrides = { "variables": { "PublicSubnets": "1", "SshKeyName": "1", "PrivateSubnets": "1", "Random": "${noop something}", }, } if previous_stack: overrides["requires"] = [previous_stack.fqn] stack = Stack( definition=generate_definition("vpc", i, **overrides), context=self.context, ) previous_stack = stack plan.add( stack=stack, run_func=self._run_func, requires=stack.requires, ) plan.dump("test", context=self.context) self.assertEqual(len(plan.list_pending()), len(plan))
def test_plan_steps_listed_with_fqn(self): plan = Plan(description="Test", sleep_time=0) stack = Stack(definition=generate_definition("vpc", 1), context=self.context) plan.add(stack=stack, run_func=lambda x, y: (x, y)) steps = plan.list_pending() self.assertEqual(steps[0][0], stack.fqn)
def test_plan_steps_listed_with_fqn(self): plan = Plan(description='Test', sleep_time=0) stack = Stack(definition=generate_definition('vpc', 1), context=self.context) plan.add(stack=stack, run_func=lambda x, y: (x, y)) steps = plan.list_pending() self.assertEqual(steps[0][0], stack.fqn)
def test_reset_after_outline(self): plan = Plan(description="Test", sleep_time=0) previous_stack = None for i in range(5): overrides = {} if previous_stack: overrides["requires"] = [previous_stack.fqn] stack = Stack( definition=generate_definition("vpc", i, **overrides), context=self.context, ) previous_stack = stack plan.add( stack=stack, run_func=self._run_func, requires=stack.requires, ) plan.outline() self.assertEqual(len(plan.list_pending()), len(plan))
def test_reset_after_dump(self, *args): plan = Plan(description='Test', sleep_time=0) previous_stack = None for i in range(5): overrides = {} if previous_stack: overrides['requires'] = [previous_stack.fqn] stack = Stack( definition=generate_definition('vpc', i, **overrides), context=self.context, ) previous_stack = stack plan.add( stack=stack, run_func=self._run_func, requires=stack.requires, ) plan.dump('test') self.assertEqual(len(plan.list_pending()), len(plan))