From 019c22a0c886461878ee8debfcf942ebbad68778 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=92=D0=B8=D0=BA=D1=82=D0=BE=D1=80=20=D0=93=D0=BB=D0=B0?=
 =?UTF-8?q?=D0=B4=D0=BA=D0=B8=D1=85?= <v.gladkikh@spider.ru>
Date: Mon, 21 Oct 2019 14:39:04 +0300
Subject: [PATCH] Fix test

---
 apps/news/serializers.py             | 2 +-
 apps/news/views.py                   | 1 +
 apps/utils/tests/tests_translated.py | 7 +++++--
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/apps/news/serializers.py b/apps/news/serializers.py
index c473be1d..f2e307e4 100644
--- a/apps/news/serializers.py
+++ b/apps/news/serializers.py
@@ -22,7 +22,7 @@ class NewsBaseSerializer(ProjectModelSerializer):
     """Base serializer for News model."""
 
     # read only fields
-    title_translated = TranslatedField()
+    title_translated = TranslatedField(source='title')
     subtitle_translated = TranslatedField()
 
     # related fields
diff --git a/apps/news/views.py b/apps/news/views.py
index 105b9064..7e272657 100644
--- a/apps/news/views.py
+++ b/apps/news/views.py
@@ -35,6 +35,7 @@ class NewsDetailView(NewsMixinView, generics.RetrieveAPIView):
         """Override get_queryset method."""
         return super().get_queryset().with_extended_related()
 
+
 class NewsTypeListView(generics.ListAPIView):
     """NewsType list view."""
 
diff --git a/apps/utils/tests/tests_translated.py b/apps/utils/tests/tests_translated.py
index 8f5caaaf..c6a990c0 100644
--- a/apps/utils/tests/tests_translated.py
+++ b/apps/utils/tests/tests_translated.py
@@ -37,6 +37,7 @@ class TranslateFieldTests(BaseTestCase):
         super().setUp()
 
         self.news_type = NewsType.objects.create(name="Test news type")
+        self.news_type.save()
 
         self.news_item = News.objects.create(
             created_by=self.user,
@@ -53,9 +54,11 @@ class TranslateFieldTests(BaseTestCase):
             slug='test',
             state=News.PUBLISHED,
         )
+        self.news_item.save()
 
     def test_model_field(self):
-        self.assertIsNotNone(getattr(self.news_item, "title_translated", None))
+        self.assertTrue(hasattr(self.news_item, "title_translated"))
+
 
     def test_read_locale(self):
         response = self.client.get(f"/api/web/news/slug/{self.news_item.slug}/", format='json')
@@ -64,7 +67,7 @@ class TranslateFieldTests(BaseTestCase):
 
         self.assertIn("title_translated", news_data)
 
-        self.assertEqual(news_data['title_translated'], "Test news item")
+        self.assertIn("Test news item", news_data['title_translated'])
 
 
 class BaseAttributeTests(BaseTestCase):