diff --git a/api/tests/test_views.py b/api/tests/test_views.py new file mode 100644 index 0000000..3b3d6e5 --- /dev/null +++ b/api/tests/test_views.py @@ -0,0 +1,37 @@ +from django.test import TestCase +from django.test.client import RequestFactory +from django.urls import reverse + +from api.views import reminders, case + + +class caseViewTest(TestCase): + def setUp(self): + self.url = reverse('case') + self.factory = RequestFactory() + + def test_case(self): + params = { + "year": "2014", + "county":"tulsa", + "case_num": "5203" + } + request = self.factory.get(self.url, params) + response = case(request) + self.assertEqual(response.status_code, 200) + + +class remindersViewTest(TestCase): + def setUp(self): + self.url = reverse('reminders') + self.factory = RequestFactory() + + def test_reminders(self): + params = { + "case_num": "CF-2014-5203", + "phone_num":"19182615259", + "arraignment_datetime": "2019-09-17T08:00:00" + } + request = self.factory.post(self.url, params) + response = reminders(request) + self.assertEqual(response.status_code, 201) diff --git a/api/views.py b/api/views.py index 543e0ca..f351b92 100644 --- a/api/views.py +++ b/api/views.py @@ -34,7 +34,7 @@ def case(request): f'year {year}, county {county}, case number {case_num}') return JsonResponse({'error': err_msg}) arraignment_datetime = parse_datetime_from_oscn_event_string( - arraignment_event.Event + arraignment_event ) return JsonResponse({ @@ -86,11 +86,14 @@ def reminders(request): def find_arraignment_or_return_False(events): for event in events: - if "arraignment" in event.Docket.lower(): - return event + if "description" in event.keys(): + if event['description'] == 'ARRAIGNMENT': + return event return False def parse_datetime_from_oscn_event_string(event): - event = event.replace('ARRAIGNMENT', '').rstrip() - return datetime.strptime(event, "%A, %B %d, %Y at %I:%M %p") + if 'date' in event.keys(): + date_str = event['date'] + return datetime.strptime(date_str, "%A, %B %d, %Y at %I:%M %p") + return False diff --git a/requirements.txt b/requirements.txt index 5f57260..5b91cfe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ Django==2.1.11 gunicorn==19.9.0 -oscn==0.0.0.32 +oscn==0.0.0.60 python-decouple==3.1 twilio==6.24.0 django-heroku==0.3.1