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