26 lines
941 B
Python
26 lines
941 B
Python
|
from django.contrib.auth import get_user_model
|
||
|
|
from django.core.urlresolvers import reverse
|
||
|
|
from rest_framework import status
|
||
|
|
from rest_framework.test import APIClient
|
||
|
|
|
||
|
|
from openslides.assignments.models import Assignment
|
||
|
|
from openslides.utils.test import TestCase
|
||
|
|
|
||
|
|
|
||
|
|
class PDF(TestCase):
|
||
|
|
"""
|
||
|
|
Tests assignment PDF.
|
||
|
|
"""
|
||
|
|
def setUp(self):
|
||
|
|
self.client = APIClient()
|
||
|
|
self.client.login(username='admin', password='admin')
|
||
|
|
self.admin = get_user_model().objects.get(username='admin')
|
||
|
|
self.assignment = Assignment.objects.create(title='test_assignment_OxieG7BioChahteY4aeM', open_posts=1)
|
||
|
|
|
||
|
|
def test_pdf_with_ballot(self):
|
||
|
|
self.assignment.set_candidate(self.admin)
|
||
|
|
self.assignment.create_poll()
|
||
|
|
self.assignment.polls.all()[0].set_published(True)
|
||
|
|
response = self.client.get(reverse('assignments_pdf'))
|
||
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|