diff --git a/smash/web/tests/view/test_worker.py b/smash/web/tests/view/test_worker.py index 9c49f8d7a80bd1a7053fb468220e92204e43cc5c..bead17b6e0286145621d9418bdc8461242ea974b 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)