diff --git a/smash/web/tests/test_view_visit.py b/smash/web/tests/test_view_visit.py index ced9caa763918e8978b39d92b77530df02efc0d0..cb55d6fca90b36f65899b92fcf5f38c465049f5b 100644 --- a/smash/web/tests/test_view_visit.py +++ b/smash/web/tests/test_view_visit.py @@ -4,8 +4,9 @@ from django.test import Client from django.test import TestCase from django.urls import reverse -from functions import create_subject, create_visit, create_appointment, create_user, create_appointment_type +from functions import create_visit, create_appointment, create_user, create_appointment_type from web.views.notifications import get_today_midnight_date +from web.forms import VisitDetailForm class VisitViewTests(TestCase): @@ -18,13 +19,30 @@ class VisitViewTests(TestCase): self.client.login(username=username, password=password) def test_visit_details_request(self): - subject = create_subject() - visit = create_visit(subject) + visit = create_visit() create_appointment(visit) response = self.client.get(reverse('web.views.visit_details', args=[visit.id])) self.assertEqual(response.status_code, 200) + def test_save_visit_details(self): + visit = create_visit() + create_appointment(visit) + + form_appointment = VisitDetailForm(instance=visit) + form_data = {} + for key, value in form_appointment.initial.items(): + if value is not None: + if isinstance(value, datetime.datetime): + form_data[key] = value.strftime("%Y-%m-%d") + else: + form_data[key] = value + + response = self.client.post( + reverse('web.views.visit_details', kwargs={'id': visit.id}), data=form_data) + self.assertEqual(response.status_code, 200) + self.assertFalse("error" in response.content) + def test_visit_list(self): create_visit()