Move self.extra assignment to setUp function

onaio · Aug 14, 2024 · d3f7bd7 · d3f7bd7
1 parent 8c89a96
commit d3f7bd7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onadata/apps/api/tests/test_tasks.py b/onadata/apps/api/tests/test_tasks.py
@@ -140,10 +140,10 @@ def setUp(self):
         self.org = OrganizationProfile.objects.create(
             user=self.org_user, name="Ona Org", creator=alice
         )
+        self.extra = {"HTTP_AUTHORIZATION": "Token %s" % self.user.auth_token}
 
     def test_user_added_to_org(self, mock_add):
         """User is added to organization"""
-        self.extra = {"HTTP_AUTHORIZATION": "Token %s" % self.user.auth_token}
         request = self.factory.get("/", **self.extra)
         request.user = self.user
         set_cache_for_org(self.org, request)
@@ -241,10 +241,10 @@ def setUp(self):
         self.org = OrganizationProfile.objects.create(
             user=self.org_user, name="Ona Org", creator=alice
         )
+        self.extra = {"HTTP_AUTHORIZATION": "Token %s" % self.user.auth_token}
 
     def test_user_removed_from_org(self, mock_remove):
         """User is removed from organization"""
-        self.extra = {"HTTP_AUTHORIZATION": "Token %s" % self.user.auth_token}
         request = self.factory.get("/", **self.extra)
         request.user = self.user
         set_cache_for_org(self.org, request)