Property role of worker model. Check if the role is valid role. #238

smash/web/models/worker.py

@@ -7,7 +7,7 @@ from django.db import models
 
 from web.models.constants import GLOBAL_STUDY_ID, COUNTRY_OTHER_ID
 from web.models.worker_study_role import STUDY_ROLE_CHOICES, HEALTH_PARTNER_ROLE_CHOICES, \
-    VOUCHER_PARTNER_ROLE_CHOICES, WORKER_STAFF, WORKER_HEALTH_PARTNER, WORKER_VOUCHER_PARTNER
+    VOUCHER_PARTNER_ROLE_CHOICES, WORKER_STAFF, WORKER_HEALTH_PARTNER, WORKER_VOUCHER_PARTNER, ROLE_CHOICES
 
 logger = logging.getLogger(__name__)
 
@@ -167,6 +167,8 @@ class Worker(models.Model):
         if roles.count() == 0:
             return WORKER_STAFF
         role = roles[0].role
+        if role not in [role_type for role_type, _ in ROLE_CHOICES]:
+            raise TypeError("Unknown worker role")
         return role
 
     @staticmethod