From a8c3c87e400420f712646d392b019e3273a0bfba Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Wed, 13 Dec 2017 17:32:43 +0100
Subject: [PATCH] unit test for rendering new voucher partner page

---
 smash/web/tests/view/test_worker.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/smash/web/tests/view/test_worker.py b/smash/web/tests/view/test_worker.py
index ecf3d303..6aba5ed6 100644
--- a/smash/web/tests/view/test_worker.py
+++ b/smash/web/tests/view/test_worker.py
@@ -4,7 +4,7 @@ from django.urls import reverse
 
 from web.forms import WorkerForm
 from web.models import Worker
-from web.models.worker_study_role import WORKER_STAFF, ROLE_CHOICES_DOCTOR
+from web.models.worker_study_role import WORKER_STAFF, ROLE_CHOICES_DOCTOR, 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
@@ -20,11 +20,13 @@ class WorkerViewTests(LoggedInTestCase):
         self.assertEqual(response.status_code, 200)
 
     def test_render_add_worker_request(self):
-        create_worker()
-
         response = self.client.get(reverse('web.views.worker_add', kwargs={'worker_type': WORKER_STAFF}))
         self.assertEqual(response.status_code, 200)
 
+    def test_render_add_worker_request_for_voucher_partner(self):
+        response = self.client.get(reverse('web.views.worker_add', kwargs={'worker_type': WORKER_VOUCHER_PARTNER}))
+        self.assertEqual(response.status_code, 200)
+
     def test_render_worker_added_request(self):
         language = create_language()
         location = create_location()
-- 
GitLab