Creating crud endpoint for user and facility flag

care/facility/api/serializers/facility_flag.py

@@ -0,0 +1,13 @@
+from rest_framework import serializers
+
+from care.facility.models import Facility, FacilityFlag
+from care.utils.serializers.fields import ExternalIdSerializerField
+
+
+class FacilityFlagSerializer(serializers.ModelSerializer):
+    id = serializers.UUIDField(source="external_id", read_only=True)
+    facility = ExternalIdSerializerField(queryset=Facility.objects.all(), required=True)
+
+    class Meta:
+        model = FacilityFlag
+        exclude = ["external_id", "deleted", "modified_date", "created_date"]

care/facility/api/viewsets/facility_flag.py

@@ -0,0 +1,37 @@
+from django_filters import rest_framework as filters
+from rest_framework import status, viewsets
+from rest_framework.decorators import action
+from rest_framework.response import Response
+
+from care.facility.api.serializers.facility_flag import FacilityFlagSerializer
+from care.facility.models import FacilityFlag
+from care.utils.custom_permissions import IsSuperUser
+
+
+class FacilityFlagFilter(filters.FilterSet):
+    flag = filters.CharFilter(field_name="flag", lookup_expr="icontains")
+    facility = filters.UUIDFilter(field_name="facility__external_id")
+
+
+class FacilityFlagViewSet(viewsets.ModelViewSet):
+    """
+    CRUD operations for FacilityFlag model.
+
+    This viewset is restricted to superusers only and provides endpoints to manage facility flags.
+    """
+
+    queryset = FacilityFlag.objects.all()
+    serializer_class = FacilityFlagSerializer
+    permission_classes = [IsSuperUser]
+    lookup_field = "external_id"
+
+    filter_backends = [filters.DjangoFilterBackend]
+    filterset_class = FacilityFlagFilter
+
+    @action(detail=False, methods=["get"], url_path="available-flags")
+    def list_available_flags(self, request):
+        """
+        List all available flags for FacilityFlag.
+        """
+        flags = FacilityFlag.objects.values_list("flag", flat=True).distinct()
+        return Response({"available_flags": list(flags)}, status=status.HTTP_200_OK)

care/facility/tests/test_facility_flags_api.py

@@ -0,0 +1,165 @@
+from rest_framework import status
+from rest_framework.test import APITestCase
+
+from care.facility.models import FacilityFlag
+from care.utils.registries.feature_flag import FlagRegistry, FlagType
+from care.utils.tests.test_utils import TestUtils
+
+
+class FacilityFlagsViewSetTestCase(TestUtils, APITestCase):
+    @classmethod
+    def setUpTestData(cls):
+        FlagRegistry.register(FlagType.FACILITY, "TEST_FLAG")
+        FlagRegistry.register(FlagType.FACILITY, "TEST_FLAG_2")
+
+        cls.state = cls.create_state()
+        cls.district = cls.create_district(cls.state)
+        cls.local_body = cls.create_local_body(cls.district)
+        cls.super_user = cls.create_super_user("su", cls.district)
+        cls.facility = cls.create_facility(cls.super_user, cls.district, cls.local_body)
+        cls.facility2 = cls.create_facility(
+            cls.super_user, cls.district, cls.local_body
+        )
+        cls.user = cls.create_user("staff", cls.district, home_facility=cls.facility)
+        cls.facility_flag_2 = cls.create_facility_flag("TEST_FLAG_2", cls.facility)
+
+    def setUp(self):
+        self.facility_flag_1 = self.create_facility_flag("TEST_FLAG", self.facility)
+
+    def get_url(self, facility_flag_id=None, action=None):
+        base_url = "/api/v1/facility_flags/"
+        if facility_flag_id is not None:
+            base_url += f"{facility_flag_id}/"
+        if action is not None:
+            base_url += f"{action}/"
+        return base_url
+
+    def test_access_with_non_super_user(self):
+        self.client.force_authenticate(user=self.user)
+        response = self.client.get(self.get_url())
+        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
+
+    def test_access_with_super_user(self):
+        self.client.force_authenticate(user=self.super_user)
+        response = self.client.get(self.get_url())
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+    def test_list_facility_flags(self):
+        self.client.force_authenticate(user=self.super_user)
+        response = self.client.get(self.get_url())
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        self.assertEqual(data["count"], 2)
+
+    def test_list_available_flags(self):
+        self.client.force_authenticate(user=self.super_user)
+        response = self.client.get(self.get_url(action="available-flags"))
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        self.assertEqual(sorted(data["available_flags"]), ["TEST_FLAG", "TEST_FLAG_2"])
+
+    def test_create_facility_flag(self):
+        self.client.force_authenticate(user=self.super_user)
+
+        response = self.client.post(self.get_url(), {})
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertIn("This field is required.", response.json()["flag"])
+        self.assertIn("This field is required.", response.json()["facility"])
+
+        # Attempting to create a duplicate flag
+        response = self.client.post(
+            self.get_url(), {"flag": "TEST_FLAG", "facility": self.facility.external_id}
+        )
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertIn(
+            "The fields facility, flag must make a unique set.",
+            response.json()["non_field_errors"],
+        )
+
+        # Creating a new facility flag
+        response = self.client.post(
+            self.get_url(),
+            {"flag": "TEST_FLAG", "facility": self.facility2.external_id},
+        )
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+    def test_retrieve_facility_flag(self):
+        self.client.force_authenticate(user=self.super_user)
+        response = self.client.get(self.get_url(self.facility_flag_1.external_id))
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        self.assertEqual(data["flag"], "TEST_FLAG")
+        self.assertEqual(data["facility"], str(self.facility.external_id))
+
+    def test_update_facility_flag(self):
+        self.client.force_authenticate(user=self.super_user)
+
+        # Confirming original values
+        response = self.client.get(self.get_url(self.facility_flag_1.external_id))
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        self.assertEqual(data["flag"], "TEST_FLAG")
+        self.assertEqual(data["facility"], str(self.facility.external_id))
+
+        # Update the facility flag
+        response = self.client.put(
+            self.get_url(self.facility_flag_1.external_id),
+            {"flag": "TEST_FLAG", "facility": self.facility2.external_id},
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.facility_flag_1.refresh_from_db()
+        self.assertEqual(self.facility_flag_1.flag, "TEST_FLAG")
+        self.assertEqual(
+            self.facility_flag_1.facility.external_id, self.facility2.external_id
+        )
+
+    def test_patch_facility_flag(self):
+        self.client.force_authenticate(user=self.super_user)
+
+        # Confirming original values
+        response = self.client.get(self.get_url(self.facility_flag_1.external_id))
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        self.assertEqual(data["flag"], "TEST_FLAG")
+        self.assertEqual(data["facility"], str(self.facility.external_id))
+
+        # Patch the facility flag
+        response = self.client.patch(
+            self.get_url(self.facility_flag_1.external_id),
+            {"facility": self.facility2.external_id},
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.facility_flag_1.refresh_from_db()
+        self.assertEqual(self.facility_flag_1.flag, "TEST_FLAG")
+        self.assertEqual(
+            self.facility_flag_1.facility.external_id, self.facility2.external_id
+        )
+
+    def test_delete_facility_flag(self):
+        self.client.force_authenticate(user=self.super_user)
+
+        # Confirm if the object exist
+        self.assertTrue(
+            FacilityFlag.objects.filter(
+                external_id=self.facility_flag_1.external_id
+            ).exists()
+        )
+        response = self.client.delete(self.get_url(self.facility_flag_1.external_id))
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+        self.facility_flag_1.refresh_from_db()
+        self.assertFalse(
+            FacilityFlag.objects.filter(
+                external_id=self.facility_flag_1.external_id
+            ).exists()
+        )
+
+    def test_creating_facility_flag_with_non_existing_flag(self):
+        self.client.force_authenticate(user=self.super_user)
+
+        response = self.client.post(
+            self.get_url(),
+            {"flag": "TEST_FLAG_NON_EXISTING", "facility": self.facility2.external_id},
+        )
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertEqual(response.json()["detail"], "Flag not registered")

care/users/api/serializers/user_flag.py

@@ -0,0 +1,13 @@
+from rest_framework import serializers
+
+from care.users.models import User, UserFlag
+from care.utils.serializers.fields import ExternalIdSerializerField
+
+
+class UserFlagSerializer(serializers.ModelSerializer):
+    id = serializers.UUIDField(source="external_id", read_only=True)
+    user = ExternalIdSerializerField(queryset=User.objects.all(), required=True)
+
+    class Meta:
+        model = UserFlag
+        exclude = ["external_id", "deleted", "modified_date", "created_date"]

care/users/api/viewsets/user_flag.py

@@ -0,0 +1,37 @@
+from django_filters import rest_framework as filters
+from rest_framework import status, viewsets
+from rest_framework.decorators import action
+from rest_framework.response import Response
+
+from care.users.api.serializers.user_flag import UserFlagSerializer
+from care.users.models import UserFlag
+from care.utils.custom_permissions import IsSuperUser
+
+
+class UserFlagFilter(filters.FilterSet):
+    flag = filters.CharFilter(field_name="flag", lookup_expr="icontains")
+    user = filters.UUIDFilter(field_name="user__external_id")
+
+
+class UserFlagViewSet(viewsets.ModelViewSet):
+    """
+    CRUD operations for UserFlag model.
+
+    This viewset is restricted to superusers only and provides endpoints to manage user flags.
+    """
+
+    queryset = UserFlag.objects.all()
+    serializer_class = UserFlagSerializer
+    permission_classes = [IsSuperUser]
+    lookup_field = "external_id"
+
+    filter_backends = [filters.DjangoFilterBackend]
+    filterset_class = UserFlagFilter
+
+    @action(detail=False, methods=["get"], url_path="available-flags")
+    def list_available_flags(self, request):
+        """
+        List all available flags for FacilityFlag.
+        """
+        flags = UserFlag.objects.values_list("flag", flat=True).distinct()
+        return Response({"available_flags": list(flags)}, status=status.HTTP_200_OK)