This commit is contained in:
Виктор Гладких 2019-10-21 14:39:04 +03:00
parent 6dd66275ef
commit 019c22a0c8
3 changed files with 7 additions and 3 deletions

View File

@ -22,7 +22,7 @@ class NewsBaseSerializer(ProjectModelSerializer):
"""Base serializer for News model.""" """Base serializer for News model."""
# read only fields # read only fields
title_translated = TranslatedField() title_translated = TranslatedField(source='title')
subtitle_translated = TranslatedField() subtitle_translated = TranslatedField()
# related fields # related fields

View File

@ -35,6 +35,7 @@ class NewsDetailView(NewsMixinView, generics.RetrieveAPIView):
"""Override get_queryset method.""" """Override get_queryset method."""
return super().get_queryset().with_extended_related() return super().get_queryset().with_extended_related()
class NewsTypeListView(generics.ListAPIView): class NewsTypeListView(generics.ListAPIView):
"""NewsType list view.""" """NewsType list view."""

View File

@ -37,6 +37,7 @@ class TranslateFieldTests(BaseTestCase):
super().setUp() super().setUp()
self.news_type = NewsType.objects.create(name="Test news type") self.news_type = NewsType.objects.create(name="Test news type")
self.news_type.save()
self.news_item = News.objects.create( self.news_item = News.objects.create(
created_by=self.user, created_by=self.user,
@ -53,9 +54,11 @@ class TranslateFieldTests(BaseTestCase):
slug='test', slug='test',
state=News.PUBLISHED, state=News.PUBLISHED,
) )
self.news_item.save()
def test_model_field(self): 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): def test_read_locale(self):
response = self.client.get(f"/api/web/news/slug/{self.news_item.slug}/", format='json') 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.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): class BaseAttributeTests(BaseTestCase):