84 lines
3.3 KiB
Python
84 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from django.test.client import Client
|
|
|
|
from openslides.assignment.models import Assignment
|
|
from openslides.participant.models import Group, User
|
|
from openslides.utils.test import TestCase
|
|
|
|
|
|
class AssignmentViewTestCase(TestCase):
|
|
def setUp(self):
|
|
# Admin
|
|
self.admin = User.objects.get(pk=1)
|
|
self.admin_client = Client()
|
|
self.admin_client.login(username='admin', password='admin')
|
|
|
|
# Staff
|
|
self.staff = User.objects.create_user('staff', 'staff@user.user', 'staff')
|
|
staff_group = Group.objects.get(name='Staff')
|
|
self.staff.groups.add(staff_group)
|
|
self.staff.save()
|
|
self.staff_client = Client()
|
|
self.staff_client.login(username='staff', password='staff')
|
|
|
|
# Delegate
|
|
self.delegate = User.objects.create_user('delegate', 'delegate@user.user', 'delegate')
|
|
delegate_group = Group.objects.get(name='Delegates')
|
|
self.delegate.groups.add(delegate_group)
|
|
self.delegate.save()
|
|
self.delegate_client = Client()
|
|
self.delegate_client.login(username='delegate', password='delegate')
|
|
|
|
# Registered
|
|
self.registered = User.objects.create_user('registered', 'registered@user.user', 'registered')
|
|
self.registered_client = Client()
|
|
self.registered_client.login(username='registered', password='registered')
|
|
|
|
self.assignment1 = Assignment.objects.create(name='test', posts=2)
|
|
|
|
def check_url(self, url, test_client, response_cose):
|
|
response = test_client.get(url)
|
|
self.assertEqual(response.status_code, response_cose)
|
|
return response
|
|
|
|
|
|
class TestAssignmentPollDelete(AssignmentViewTestCase):
|
|
def setUp(self):
|
|
super(TestAssignmentPollDelete, self).setUp()
|
|
self.assignment1.gen_poll()
|
|
|
|
def test_get(self):
|
|
response = self.check_url('/assignment/poll/1/del/', self.admin_client, 302)
|
|
self.assertRedirects(response, 'assignment/1/')
|
|
|
|
def test_post(self):
|
|
response = self.admin_client.post('/assignment/poll/1/del/', {'yes': 1})
|
|
self.assertRedirects(response, '/assignment/1/')
|
|
|
|
|
|
class TestAssignmentDetailView(AssignmentViewTestCase):
|
|
def test_blocked_candidates_view(self):
|
|
"""
|
|
Tests that a delegate runs for a vote and then withdraws himself.
|
|
"""
|
|
response = self.staff_client.get('/assignment/1/')
|
|
self.assertContains(response, 'No candidates available.')
|
|
self.assertNotContains(response, 'Blocked Candidates')
|
|
|
|
response = self.delegate_client.get('/assignment/1/run/')
|
|
self.assertTrue(self.assignment1.is_candidate(self.delegate))
|
|
self.assertFalse(self.assignment1.is_blocked(self.delegate))
|
|
|
|
response = self.staff_client.get('/assignment/1/')
|
|
self.assertNotContains(response, 'No candidates available.')
|
|
self.assertNotContains(response, 'Blocked Candidates')
|
|
|
|
response = self.delegate_client.get('/assignment/1/delrun/')
|
|
self.assertFalse(self.assignment1.is_candidate(self.delegate))
|
|
self.assertTrue(self.assignment1.is_blocked(self.delegate))
|
|
|
|
response = self.staff_client.get('/assignment/1/')
|
|
self.assertContains(response, 'No candidates available.')
|
|
self.assertContains(response, 'Blocked Candidates')
|