Update unit test

This commit is contained in:
Ramon Wenger 2022-01-28 20:22:06 +01:00
parent 3403c2db3d
commit e843f6be6a
1 changed files with 14 additions and 17 deletions

View File

@ -1,14 +1,15 @@
from unittest import TestCase
from django.test import TestCase
from graphene.test import Client
from api.schema import schema
from news.factories import NewsTeaserFactory
from news.models import NewsTeaser
class NewsTeaserTests(TestCase):
def setUp(self):
self.news_teaser1 = NewsTeaserFactory(order_id=1)
self.news_teaser2 = NewsTeaserFactory(order_id=2)
self.news_teaser1 = NewsTeaserFactory(order_id=1, title="should come first")
self.news_teaser2 = NewsTeaserFactory(order_id=2, title="should be second")
self.client = Client(schema=schema)
@ -16,17 +17,13 @@ class NewsTeaserTests(TestCase):
query = '''
query NewsTeasers {
newsTeasers {
edges {
node {
id
description
title
imageUrl
newsArticleUrl
displayDate
imageSource
}
}
id
description
title
imageUrl
newsArticleUrl
displayDate
imageSource
}
}
'''
@ -36,7 +33,7 @@ class NewsTeaserTests(TestCase):
def test_can_get_news_teaser(self):
result = self.make_query()
self.assertIsNone(result.get('errors'))
news_teasers = result.get('data').get('newsTeasers').get('edges')
self.assertEqual(news_teasers[0].get('node').get('title'), self.news_teaser1.title)
self.assertEqual(news_teasers[1].get('node').get('title'), self.news_teaser2.title)
news_teasers = result.get('data').get('newsTeasers')
self.assertEqual(news_teasers[0].get('title'), self.news_teaser1.title)
self.assertEqual(news_teasers[1].get('title'), self.news_teaser2.title)