From ee916a270b9c51db54c9fd0c7896366283db25cf Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Wed, 29 Nov 2017 17:55:38 +0100
Subject: [PATCH] unit tests for checking if worker is on leave and getting
worker for anynomous user
---
smash/web/tests/models/test_worker.py | 22 ++++++++++++++++++----
1 file changed, 18 insertions(+), 4 deletions(-)
diff --git a/smash/web/tests/models/test_worker.py b/smash/web/tests/models/test_worker.py
index 9b193326..e43ae0bb 100644
--- a/smash/web/tests/models/test_worker.py
+++ b/smash/web/tests/models/test_worker.py
@@ -1,9 +1,11 @@
-from django.contrib.auth.models import User
-from django.test import Client
-from django.test import TestCase
+import datetime
-from web.models import Worker
+from django.contrib.auth.models import User, AnonymousUser
+from django.test import Client, TestCase
+
+from web.models import Holiday, Worker
from web.tests.functions import create_worker
+from web.views.notifications import get_today_midnight_date
class WorkerModelTests(TestCase):
@@ -46,3 +48,15 @@ class WorkerModelTests(TestCase):
self.fail("Exception expected")
except TypeError:
pass
+
+ def test_is_on_leave(self):
+ worker = create_worker()
+ self.assertFalse(worker.is_on_leave())
+ Holiday(person=worker,
+ datetime_start=get_today_midnight_date() + datetime.timedelta(days=-2),
+ datetime_end=get_today_midnight_date() + datetime.timedelta(days=2)).save()
+
+ self.assertTrue(worker.is_on_leave())
+
+ def test_get_by_user_for_anonymous(self):
+ self.assertIsNone(Worker.get_by_user(AnonymousUser()))
--
GitLab