From f4172550acec088e1d51cb3ebea42cb25954316b Mon Sep 17 00:00:00 2001 From: ariannedee Date: Wed, 14 Oct 2020 14:24:41 -0700 Subject: [PATCH] Add another test --- .../apps/core/tests/test_step_completeness.py | 36 ++++++++++++------- 1 file changed, 24 insertions(+), 12 deletions(-) diff --git a/edivorce/apps/core/tests/test_step_completeness.py b/edivorce/apps/core/tests/test_step_completeness.py index 0f3d4e68..b1e46d5f 100644 --- a/edivorce/apps/core/tests/test_step_completeness.py +++ b/edivorce/apps/core/tests/test_step_completeness.py @@ -500,24 +500,36 @@ class ChildrenStepCompletenessTestCase(TestCase): def test_no_children(self): self.create_response('children_of_marriage', 'NO') - self.assertEqual(self.get_children_step_status(), 'Hidden') - self.assertEqual(self.get_children_step_status('your_children'), 'Hidden') - self.assertEqual(self.get_children_step_status('income_expenses'), 'Hidden') - self.assertEqual(self.get_children_step_status('facts'), 'Hidden') - self.assertEqual(self.get_children_step_status('payor_medical'), 'Hidden') - self.assertEqual(self.get_children_step_status('what_for'), 'Hidden') + self.assertEqual(self.get_children_step_status(), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('your_children'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('income_expenses'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('facts'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('payor_medical'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('what_for'), Status.HIDDEN) def test_only_grown_children(self): self.create_response('children_of_marriage', 'YES') self.create_response('has_children_under_19', 'NO') self.create_response('has_children_over_19', 'YES') self.create_response('children_financial_support', '["NO"]') - self.assertEqual(self.get_children_step_status(), 'Hidden') - self.assertEqual(self.get_children_step_status('your_children'), 'Hidden') - self.assertEqual(self.get_children_step_status('income_expenses'), 'Hidden') - self.assertEqual(self.get_children_step_status('facts'), 'Hidden') - self.assertEqual(self.get_children_step_status('payor_medical'), 'Hidden') - self.assertEqual(self.get_children_step_status('what_for'), 'Hidden') + self.assertEqual(self.get_children_step_status(), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('your_children'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('income_expenses'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('facts'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('payor_medical'), Status.HIDDEN) + self.assertEqual(self.get_children_step_status('what_for'), Status.HIDDEN) + + def test_has_children(self): + self.create_response('children_of_marriage', 'YES') + self.create_response('has_children_under_19', 'NO') + self.create_response('has_children_over_19', 'YES') + self.create_response('children_financial_support', '["Yes, other reason"]') + self.assertEqual(self.get_children_step_status(), Status.NOT_STARTED) + self.assertEqual(self.get_children_step_status('your_children'), Status.NOT_STARTED) + self.assertEqual(self.get_children_step_status('income_expenses'), Status.NOT_STARTED) + self.assertEqual(self.get_children_step_status('facts'), Status.NOT_STARTED) + self.assertEqual(self.get_children_step_status('payor_medical'), Status.NOT_STARTED) + self.assertEqual(self.get_children_step_status('what_for'), Status.NOT_STARTED) def test_children_details(self): substep = 'your_children'