From 9f44db56f2da55723fcf21e9c8d2fd0bf3eeae03 Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@.uni.lu>
Date: Fri, 21 Sep 2018 11:37:30 +0200
Subject: [PATCH] unit tests for property role of worker model. Issue #238

---
 smash/web/tests/models/test_worker.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/smash/web/tests/models/test_worker.py b/smash/web/tests/models/test_worker.py
index 1b479467..338ef668 100644
--- a/smash/web/tests/models/test_worker.py
+++ b/smash/web/tests/models/test_worker.py
@@ -91,3 +91,29 @@ class WorkerModelTests(TestCase):
         worker.roles.update(role="unk")
         with self.assertRaises(Exception):
             worker_type_by_worker(worker)
+
+    #worker role tests
+
+    def test_worker_role_by_worker_without_role(self):
+        self.assertEqual(WORKER_STAFF, Worker().role)
+
+    def test_worker_role_by_worker_with_secretary_role(self):
+        worker = create_worker()
+        worker.roles.update(role=ROLE_CHOICES_SECRETARY)
+        self.assertEqual(ROLE_CHOICES_SECRETARY, worker.role)
+
+    def test_worker_role_by_worker_with_health_partner_role(self):
+        worker = create_worker()
+        worker.roles.update(role=ROLE_CHOICES_HEALTH_PARTNER)
+        self.assertEqual(ROLE_CHOICES_HEALTH_PARTNER, worker.role)
+
+    def test_worker_role_by_worker_with_voucher_partner_role(self):
+        worker = create_worker()
+        worker.roles.update(role=ROLE_CHOICES_VOUCHER_PARTNER)
+        self.assertEqual(ROLE_CHOICES_VOUCHER_PARTNER, worker.role)
+
+    def test_worker_role_by_worker_with_invalid_role(self):
+        worker = create_worker()
+        worker.roles.update(role="unk")
+        with self.assertRaises(Exception):
+            worker.role
-- 
GitLab