From 2c9140d8d24f82fa92c322abbce3e4e30ec75204 Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@uni.lu>
Date: Wed, 24 Oct 2018 11:31:34 +0200
Subject: [PATCH] added tests for different worker_type in 'web.views.workers'

---
 smash/web/tests/view/test_worker.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/smash/web/tests/view/test_worker.py b/smash/web/tests/view/test_worker.py
index 9c49f8d7..bead17b6 100644
--- a/smash/web/tests/view/test_worker.py
+++ b/smash/web/tests/view/test_worker.py
@@ -5,7 +5,7 @@ from django_common.auth_backends import User
 
 from web.forms import WorkerForm
 from web.models import Worker
-from web.models.worker_study_role import WORKER_STAFF, ROLE_CHOICES_DOCTOR, WORKER_VOUCHER_PARTNER
+from web.models.worker_study_role import WORKER_STAFF, ROLE_CHOICES_DOCTOR, WORKER_HEALTH_PARTNER, WORKER_VOUCHER_PARTNER
 from web.tests import create_worker
 from web.tests.functions import create_language, create_location, create_availability, format_form_field
 from .. import LoggedInTestCase
@@ -16,10 +16,18 @@ logger = logging.getLogger(__name__)
 class WorkerViewTests(LoggedInTestCase):
     def test_render_workers_list_request(self):
         create_worker()
-
         response = self.client.get(reverse('web.views.workers'))
         self.assertEqual(response.status_code, 200)
 
+    def test_render_worker_type_request(self):
+        create_worker()
+        response = self.client.get(reverse('web.views.workers', kwargs={'worker_type': WORKER_STAFF}))
+        self.assertEqual(response.status_code, 200)
+        response = self.client.get(reverse('web.views.workers', kwargs={'worker_type': WORKER_HEALTH_PARTNER}))
+        self.assertEqual(response.status_code, 200)
+        response = self.client.get(reverse('web.views.workers', kwargs={'worker_type': WORKER_VOUCHER_PARTNER}))
+        self.assertEqual(response.status_code, 200)
+
     def test_render_add_worker_request(self):
         response = self.client.get(reverse('web.views.worker_add', kwargs={'worker_type': WORKER_STAFF}))
         self.assertEqual(response.status_code, 200)
-- 
GitLab