From 8d229eaaa4660a97f1e6f89f7b31bce2dc0858e5 Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Wed, 29 Nov 2017 18:13:22 +0100
Subject: [PATCH] unit tests for rendering export webpage and exporting
 appointments without visit defiend

---
 smash/web/tests/view/test_export.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/smash/web/tests/view/test_export.py b/smash/web/tests/view/test_export.py
index d1394128..d6938476 100644
--- a/smash/web/tests/view/test_export.py
+++ b/smash/web/tests/view/test_export.py
@@ -1,9 +1,9 @@
 # coding=utf-8
 from django.urls import reverse
 
-from web.models import Appointment
+from web.models import Appointment, AppointmentTypeLink
 from web.tests import LoggedInTestCase
-from web.tests.functions import create_study_subject, create_appointment, create_visit
+from web.tests.functions import create_study_subject, create_appointment, create_visit, create_appointment_type
 from web.views.export import subject_to_row_for_fields, DROP_OUT_FIELD
 
 
@@ -13,6 +13,11 @@ class TestExportView(LoggedInTestCase):
         response = self.client.get(reverse('web.views.export_to_csv', kwargs={'data_type': "subjects"}))
         self.assertEqual(response.status_code, 200)
 
+    def test_render_export(self):
+        create_study_subject()
+        response = self.client.get(reverse('web.views.export'))
+        self.assertEqual(response.status_code, 200)
+
     def test_export_appointments_to_csv(self):
         create_appointment()
         response = self.client.get(reverse('web.views.export_to_csv', kwargs={'data_type': "appointments"}))
@@ -24,7 +29,11 @@ class TestExportView(LoggedInTestCase):
         self.assertEqual(response.status_code, 200)
 
     def test_export_appointments_to_excel(self):
-        create_appointment()
+        appointment = create_appointment()
+        appointment.visit = None
+        appointment.save()
+        AppointmentTypeLink.objects.create(appointment=appointment, appointment_type=create_appointment_type())
+
         response = self.client.get(reverse('web.views.export_to_excel', kwargs={'data_type': "appointments"}))
         self.assertEqual(response.status_code, 200)
 
-- 
GitLab