fix news_type

apps/establishment/tests.py

@@ -13,11 +13,13 @@ class BaseTestCase(APITestCase):
         self.password = 'sedragurdaredips19'
         self.email = 'sedragurda@desoz.com'
         self.newsletter = True
-        self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
+        self.user = User.objects.create_user(
+            username=self.username, email=self.email, password=self.password)
         #get tokkens
         tokkens = User.create_jwt_tokens(self.user)
-        self.client.cookies = SimpleCookie({'access_token': tokkens.get('access_token'),
-                                            'refresh_token': tokkens.get('refresh_token')})
+        self.client.cookies = SimpleCookie(
+            {'access_token': tokkens.get('access_token'),
+             'refresh_token': tokkens.get('refresh_token')})
 
 
 class EmployeeTests(BaseTestCase):

apps/search_indexes/documents/news.py

@@ -46,3 +46,11 @@ class NewsDocument(Document):
 
     def prepare_description(self, instance):
         return instance.description
+
+    def get_instances_from_related(self, related_instance):
+        """If related_models is set, define how to retrieve the Car instance(s) from the related model.
+        The related_models option should be used with caution because it can lead in the index
+        to the updating of a lot of items.
+        """
+        if isinstance(related_instance, models.NewsType):
+            return related_instance.news_set.all()