def test_single_story(self):
url = "/api/v1/stories/%d/" % self.uganda_story.pk
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
story = self.uganda_story
category = response.data.pop("category")
self.assertDictEqual(
response.data,
dict(
id=story.pk,
title=story.title,
video_id=story.video_id,
audio_link=story.audio_link,
summary=story.summary,
featured=story.featured,
tags=story.tags,
images=StoryReadSerializer().get_images(story),
org=story.org_id,
created_on=story.created_on.strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
),
)
self.assertDictEqual(
dict(category),
dict(name=story.category.name, image_url=CategoryReadSerializer().get_image_url(story.category)),
)
def test_single_poll(self):
url = '/api/v1/polls/%d/' % self.reg_poll.pk
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
poll = self.reg_poll
category = response.data.pop('category')
self.assertDictEqual(response.data, dict(id=poll.pk,
flow_uuid=poll.flow_uuid,
title=poll.title,
org=poll.org_id,
questions=[],
created_on=poll.created_on.strftime('%Y-%m-%dT%H:%M:%S.%fZ')))
self.assertDictEqual(dict(category), dict(OrderedDict(name=poll.category.name,
image_url=CategoryReadSerializer().
get_image_url(poll.category))))
with patch('ureport.polls.models.PollQuestion.get_results') as mock_get_results:
mock_get_results.return_value = [dict(set=20, unset=10, open_ended=False, categories="CATEGORIES-DICT")]
poll_question = PollQuestion.objects.create(poll=self.reg_poll, title="What's on mind? :)",
ruleset_uuid='uuid1', created_by=self.superuser,
modified_by=self.superuser)
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
category = response.data.pop('category')
self.assertDictEqual(response.data, dict(id=poll.pk,
flow_uuid=poll.flow_uuid,
title=poll.title,
org=poll.org_id,
created_on=poll.created_on.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
questions=[dict(id=poll_question.pk,
ruleset_uuid='uuid1',
title="What's on mind? :)",
results=dict(set=20,
unset=10,
open_ended=False,
categories="CATEGORIES-DICT"))]))
poll_question.is_active = False
poll_question.save()
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
category = response.data.pop('category')
self.assertDictEqual(response.data, dict(id=poll.pk,
flow_uuid=poll.flow_uuid,
title=poll.title,
org=poll.org_id,
created_on=poll.created_on.strftime('%Y-%m-%dT%H:%M:%S.%fZ'),
questions=[]))
def test_single_video(self):
url = '/api/v1/videos/%d/' % self.uganda_video.pk
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
video = self.uganda_video
category = response.data.pop('category')
self.assertDictEqual(response.data, dict(id=video.pk,
title=video.title,
video_id=video.video_id,
description=video.description,
org=video.org_id,
created_on=video.created_on.strftime('%Y-%m-%dT%H:%M:%S.%fZ')))
self.assertDictEqual(dict(category), dict(name=video.category.name,
image_url=CategoryReadSerializer().get_image_url(video.category)))
def test_single_news_item(self):
url = '/api/v1/news/%d/' % self.news_item.pk
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
news = self.news_item
category = response.data.pop('category')
self.assertDictEqual(response.data, dict(id=news.pk,
short_description=news.short_description(),
title=news.title,
description=news.description,
link=news.link,
org=news.org_id,
created_on=news.created_on.strftime('%Y-%m-%dT%H:%M:%S.%fZ')))
self.assertDictEqual(dict(category), dict(name=news.category.name,
image_url=CategoryReadSerializer().get_image_url(news.category)))
def test_single_poll(self):
url = '/api/v1/polls/%d/' % self.reg_poll.pk
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
poll = self.reg_poll
category = response.data.pop('category')
self.assertDictEqual(
response.data,
dict(
id=poll.pk,
flow_uuid=poll.flow_uuid,
title=poll.title,
org=poll.org_id,
))
self.assertDictEqual(
dict(category),
dict(
OrderedDict(name=poll.category.name,
image_url=CategoryReadSerializer().get_image_url(
poll.category))))
def test_single_story(self):
url = '/api/v1/stories/%d/' % self.uganda_story.pk
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
story = self.uganda_story
category = response.data.pop('category')
self.assertDictEqual(
response.data,
dict(id=story.pk,
title=story.title,
video_id=story.video_id,
audio_link=story.audio_link,
summary=story.summary,
featured=story.featured,
tags=story.tags,
images=StoryReadSerializer().get_images(story),
org=story.org_id))
self.assertDictEqual(
dict(category),
dict(name=story.category.name,
image_url=CategoryReadSerializer().get_image_url(
story.category)))
