rename TrakOrg* serializers and test classes to Org*

USEPA · May 28, 2024 · 6b5ceeb · 6b5ceeb
1 parent 39e4529
commit 6b5ceeb
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 12 deletions.
diff --git a/server/apps/org/serializers.py b/server/apps/org/serializers.py
@@ -4,7 +4,7 @@
 from apps.org.models import Org
 
 
-class TrakOrgSerializer(ModelSerializer):
+class OrgSerializer(ModelSerializer):
     """Haztrak Organization Model Serializer"""
 
     id = serializers.CharField(

diff --git a/server/apps/org/tests/test_models.py b/server/apps/org/tests/test_models.py
@@ -1,7 +1,7 @@
 from apps.org.models import Org
 
 
-class TestTrakOrgModel:
+class TestOrgModel:
     def test_haztrak_org_factory(self, org_factory):
         org = org_factory()
         assert isinstance(org, Org)

diff --git a/server/apps/org/tests/test_serializers.py b/server/apps/org/tests/test_serializers.py
@@ -1,7 +1,7 @@
-from apps.org.serializers import TrakOrgSerializer
+from apps.org.serializers import OrgSerializer
 
 
-class TestTrakOrgSerializer:
+class TestOrgSerializer:
     def test_rcrainfo_integrated_is_true(
         self, org_factory, profile_factory, rcrainfo_profile_factory, user_factory
     ):
@@ -12,7 +12,7 @@ def test_rcrainfo_integrated_is_true(
         )
         profile_factory(user=admin, rcrainfo_profile=rcra_profile)
         org = org_factory(admin=admin)
-        serializer = TrakOrgSerializer(org)
+        serializer = OrgSerializer(org)
         assert serializer.data["rcrainfoIntegrated"] is True
 
     def test_rcrainfo_integrated_is_false(
@@ -25,5 +25,5 @@ def test_rcrainfo_integrated_is_false(
         )
         profile_factory(user=admin, rcrainfo_profile=rcra_profile)
         org = org_factory(admin=admin)
-        serializer = TrakOrgSerializer(org)
+        serializer = OrgSerializer(org)
         assert serializer.data["rcrainfoIntegrated"] is False
diff --git a/server/apps/org/tests/test_views.py b/server/apps/org/tests/test_views.py
@@ -4,7 +4,7 @@
 from apps.org.views import OrgDetailsView
 
 
-class TestTrakOrgDetailsView:
+class TestOrgDetailsView:
     base_url = "/api/org"
     factory = APIRequestFactory()
 

diff --git a/server/apps/org/views.py b/server/apps/org/views.py
@@ -7,7 +7,7 @@
 from rest_framework.response import Response
 
 from apps.org.models import Org
-from apps.org.serializers import TrakOrgSerializer
+from apps.org.serializers import OrgSerializer
 from apps.org.services import get_org_by_id
 
 logger = logging.getLogger(__name__)
@@ -16,12 +16,12 @@
 class OrgDetailsView(RetrieveAPIView):
     """Retrieve details for a given Org"""
 
-    serializer_class = TrakOrgSerializer
+    serializer_class = OrgSerializer
     queryset = Org.objects.all()
     lookup_url_kwarg = "org_id"
 
     @method_decorator(cache_page(60 * 15))
     def get(self, request, *args, **kwargs):
         org = get_org_by_id(self.kwargs["org_id"])
-        serializer = TrakOrgSerializer(org)
+        serializer = OrgSerializer(org)
         return Response(data=serializer.data, status=status.HTTP_200_OK)
diff --git a/server/apps/profile/serializers/trak_profile.py b/server/apps/profile/serializers/trak_profile.py
@@ -1,7 +1,7 @@
 from rest_framework import serializers
 from rest_framework.serializers import ModelSerializer
 
-from apps.org.serializers import TrakOrgSerializer
+from apps.org.serializers import OrgSerializer
 from apps.profile.models import TrakProfile
 from apps.site.serializers import SiteAccessSerializer
 
@@ -16,7 +16,7 @@ class TrakProfileSerializer(ModelSerializer):
         source="user.site_permissions",
         many=True,
     )
-    org = TrakOrgSerializer(
+    org = OrgSerializer(
         source="user.org_permissions.org",
         required=False,
     )