bunch of stuff

products/tests.py

@@ -952,9 +952,13 @@ class MyProductsViewTest(APITestCase):
 
     def test_auth_user_gets_data(self):
         # create instance
+        company = CompanyFactory()
+        self.user.company = company
+        self.user.save()
+
         user_instances = [
-            self.factory(creator=self.user),
-            self.factory(creator=self.user),
+            self.factory(company=company),
+            self.factory(company=company),
         ]
 
         # Authenticate
@@ -976,7 +980,11 @@ class MyProductsViewTest(APITestCase):
         self.client.credentials(HTTP_AUTHORIZATION=f"Bearer {token['access']}")
 
         # create instances
-        instances = [self.factory(creator=self.user) for n in range(12)]
+        company = CompanyFactory()
+        self.user.company = company
+        self.user.save()
+
+        instances = [self.factory(company=company) for n in range(12)]
 
         # Request list
         url = f"{self.endpoint}?limit=5&offset=10"

products/views.py

@@ -72,10 +72,7 @@ class MyProductsViewSet(viewsets.ModelViewSet):
     permission_classes = [IsAuthenticated]
 
     def get_queryset(self):
-        return self.model.objects.filter(creator=self.request.user).order_by('-created')
-
-    def perform_create(self, serializer):
-        serializer.save(creator=self.request.user)
+        return self.model.objects.filter(company=self.request.user.company).order_by('-created')
 
 
 class AdminProductsViewSet(viewsets.ModelViewSet):