diff --git a/src/backend/expungeservice/models/case.py b/src/backend/expungeservice/models/case.py index 5ee858cb5..ce3cabfaa 100644 --- a/src/backend/expungeservice/models/case.py +++ b/src/backend/expungeservice/models/case.py @@ -29,6 +29,6 @@ def closed(self): @staticmethod def _set_birth_year(info): if len(info) > 1: - return int(info[1]) + return int(info[1].split('/')[-1]) else: return '' diff --git a/src/backend/tests/models/test_case.py b/src/backend/tests/models/test_case.py index 8ca8aafba..a5230cbd7 100644 --- a/src/backend/tests/models/test_case.py +++ b/src/backend/tests/models/test_case.py @@ -42,7 +42,8 @@ def test_it_returns_true_for_an_inactive_case(self): assert self.case.closed() is True -class TestBirthYearDefaultValue(unittest.TestCase): +class TestBirthYearInitializesGivenMultipleValues(unittest.TestCase): + def setUp(self): self.case = CaseFactory.build() @@ -51,3 +52,15 @@ def test_birth_year_defaults_to_empty_string(self): case = CaseFactory.save(self.case) assert case.birth_year == '' + + def test_it_assigns_birth_year_when_given_the_year(self): + self.case['info'] = ['John Doe', '1979'] + case = CaseFactory.save(self.case) + + assert case.birth_year == 1979 + + def test_it_assigns_birth_year_when_given_the_month_day_year_format(self): + self.case['info'] = ['John Doe', '12/21/1979'] + case = CaseFactory.save(self.case) + + assert case.birth_year == 1979