unit tests for checking if worker is on leave and getting worker for anynomous user

smash/web/tests/models/test_worker.py

-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()))