added tests for different worker_type in 'web.views.workers'

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)