fix on test (additional failure statement added)

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()