From 3059b6001a4e412e6840fa8cf67d9fde01d80853 Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Tue, 5 Jun 2018 14:47:55 +0200
Subject: [PATCH] unit tests added
---
smash/web/tests/api_views/test_voucher.py | 8 ++++++++
smash/web/tests/api_views/test_worker.py | 10 +++++++++-
2 files changed, 17 insertions(+), 1 deletion(-)
diff --git a/smash/web/tests/api_views/test_voucher.py b/smash/web/tests/api_views/test_voucher.py
index 6060a06b..adbb1fd5 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 bc245f0d..c146d560 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')
--
GitLab