def get(self, request, *args, **kwargs):
in_flow, flow_id = user_in_flow(request.user)
form = self.get_form()
context = self.get_context_data(form=form)
if not in_flow:
return self.render_to_response(context)
else:
poll_id = get_poll_id(flow_id)
context['poll_id'] = poll_id
has_poll = poll_id is not None
context['has_poll'] = has_poll
if has_poll:
self.template_name = 'user_in_flow.html'
messages.info(self.request, _('Please complete the poll before you can send us a message'))
return self.render_to_response(context)
def test_can_get_poll_id_from_ureport(self):
url = '%s/api/v1/polls/org/%s/' % (settings.UREPORT_ROOT, settings.UREPORT_ORG_ID)
responses.add(responses.GET, url,
body="""
{
"count": 1,
"next": null,
"previous": null,
"results": [
{
"id": "poll1"
},
{
"id": "05c66a1e-ff50-4b46-8e7efeb699b0"
}
]
}
""",
status=200,
content_type='application/json')
poll_id = get_poll_id('whoslsads')
self.assertEqual(poll_id, 'poll1')