feat: add staff role view

Signed-off-by: maxwellgithinji <maxwellgithinji@gmail.com>

mycarehub/authority/forms.py

@@ -0,0 +1,9 @@
+from mycarehub.authority.models import AuthorityRole
+
+
+def get_role_choices():
+    choices = []
+    for role in AuthorityRole.objects.all():
+        choices.append((role.name, role.name))
+
+    return choices

mycarehub/staff/forms.py

@@ -1,6 +1,7 @@
 from django import forms
 from phonenumber_field.formfields import PhoneNumberField
 
+from mycarehub.authority.forms import get_role_choices
 from mycarehub.clients.forms import get_facility_choices, validate_date_past
 from mycarehub.users.models import GenderChoices
 
@@ -58,3 +59,10 @@ class StaffRegistrationForm(forms.Form):
         label="Staff Number",
         help_text="The Staff's currently assigned staff number",
     )
+
+    role = forms.ChoiceField(
+        required=True,
+        choices=get_role_choices,
+        label="Staff's Roles",
+        help_text="The Staff's currently assigned roles",
+    )

mycarehub/staff/tests/test_views.py

@@ -4,6 +4,7 @@
 from model_bakery import baker
 from rest_framework import status
 
+from mycarehub.authority.models import AuthorityRole
 from mycarehub.common.models.common_models import Facility
 
 fake = Faker()
@@ -15,6 +16,7 @@ def test_staff_registration_view_valid(user_with_all_permissions, client):
     url = reverse("staff_registration")
     org = user_with_all_permissions.organisation
     facility = baker.make(Facility, organisation=org)
+    role = baker.make(AuthorityRole, organisation=org)
     response = client.post(
         url,
         data={
@@ -25,6 +27,7 @@ def test_staff_registration_view_valid(user_with_all_permissions, client):
             "phone_number": "+254722000000",
             "id_number": fake.random_int(),
             "staff_number": fake.random_int(),
+            "role": role.name,
         },
         content_type="application/json",
         accept="application/json",

mycarehub/staff/views.py

@@ -3,6 +3,7 @@
 from rest_framework.response import Response
 from rest_framework.views import APIView
 
+from mycarehub.authority.models import AuthorityRole
 from mycarehub.clients.models import Identifier
 from mycarehub.common.models.common_models import Contact, Facility
 from mycarehub.users.models import User
@@ -94,6 +95,9 @@ def post(self, request, format=None):
                 # add the identifier to the staff
                 staff.identifiers.add(identifier)
 
+                # add user to roles
+                for role in AuthorityRole.objects.filter(name=data["role"]):
+                    role.users.add(new_user)
                 # return the newly created staff
                 serialized_staff = StaffSerializer(staff)
                 return Response(serialized_staff.data, status=status.HTTP_201_CREATED)