diff --git a/smash/web/tests/api_views/test_voucher.py b/smash/web/tests/api_views/test_voucher.py index 6060a06b34b09e9893305cf2b98d877d32b50094..adbb1fd521058559ef751e5d77b7b521037e2901 100644 --- a/smash/web/tests/api_views/test_voucher.py +++ b/smash/web/tests/api_views/test_voucher.py @@ -1,5 +1,6 @@ # 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) diff --git a/smash/web/tests/api_views/test_worker.py b/smash/web/tests/api_views/test_worker.py index bc245f0d432e478a0d3763b2e8d37d373ea4a114..c146d560acf741436cef1ba93667a994940d8fdb 100644 --- a/smash/web/tests/api_views/test_worker.py +++ b/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')