unit tests added

smash/web/tests/api_views/test_voucher.py

 # coding=utf-8
 import datetime
+import json
 import logging
 
 from django.urls import reverse
@@ -194,3 +195,10 @@ class TestVoucherApi(LoggedInWithWorkerTestCase):
         voucher = self.voucher
 
         self.check_voucher_filtered([["", "X"]], [voucher])
+
+    def test_get_columns(self):
+        response = self.client.get(reverse('web.api.vouchers.columns'))
+        self.assertEqual(response.status_code, 200)
+
+        columns = json.loads(response.content)['columns']
+        self.assertTrue(len(columns) > 0)

smash/web/tests/api_views/test_worker.py

@@ -4,10 +4,12 @@ import json
 from django.test import RequestFactory
 from django.urls import reverse
 
+from web.api_views.worker import availabilities
 from web.models import Availability
 from web.models.constants import TUESDAY_AS_DAY_OF_WEEK
-from web.api_views.worker import availabilities
+from web.models.worker_study_role import WORKER_STAFF
 from web.tests import LoggedInWithWorkerTestCase
+from web.tests.functions import create_voucher_partner
 
 
 class TestWorkerApi(LoggedInWithWorkerTestCase):
@@ -52,6 +54,12 @@ class TestWorkerApi(LoggedInWithWorkerTestCase):
         self.assertEqual(response.status_code, 200)
         self.assertTrue(self.worker.first_name in response.content)
 
+    def test_voucher_partners(self):
+        voucher_partner = create_voucher_partner()
+        response = self.client.get(reverse('web.api.workers', kwargs={'worker_role': WORKER_STAFF}))
+        self.assertEqual(response.status_code, 200)
+        self.assertTrue(voucher_partner.first_name in response.content)
+
     def test_empty_availabilities(self):
         factory = RequestFactory()
         request = factory.get('/api/workers?start_date=2017-10-20&end_date=2017-11-20')