OpenSlides/tests/integration/assignments/test_viewset.py
2020-03-17 07:24:36 +01:00

441 lines
15 KiB
Python

import pytest
from django.contrib.auth import get_user_model
from django.core.files.uploadedfile import SimpleUploadedFile
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APIClient
from openslides.assignments.models import Assignment, AssignmentPoll
from openslides.core.models import Tag
from openslides.mediafiles.models import Mediafile
from openslides.utils.autoupdate import inform_changed_data
from tests.count_queries import count_queries
from tests.test_case import TestCase
@pytest.mark.django_db(transaction=False)
def test_assignment_db_queries():
"""
Tests that only the following db queries are done:
* 1 requests to get the list of all assignments,
* 1 request to get all related users,
* 1 request to get the agenda item,
* 1 request to get the list of speakers,
* 1 request to get the tags,
* 1 request to get the attachments and
"""
for index in range(10):
assignment = Assignment.objects.create(title=f"assignment{index}", open_posts=1)
for i in range(2):
AssignmentPoll.objects.create(
assignment=assignment,
title="test_title_nah5Ahh6IkeeM8rah3ai",
pollmethod=AssignmentPoll.POLLMETHOD_YN,
type=AssignmentPoll.TYPE_NAMED,
)
assert count_queries(Assignment.get_elements)() == 6
class CreateAssignment(TestCase):
"""
Tests basic creation of assignments.
"""
def setUp(self):
self.client = APIClient()
self.client.login(username="admin", password="admin")
def test_simple(self):
response = self.client.post(
reverse("assignment-list"),
{"title": "test_title_ef3jpF)M329f30m)f82", "open_posts": 1},
)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
assignment = Assignment.objects.get()
self.assertEqual(assignment.title, "test_title_ef3jpF)M329f30m)f82")
def test_with_tags_and_mediafiles(self):
Tag.objects.create(name="test_tag")
Mediafile.objects.create(
title="test_file", mediafile=SimpleUploadedFile("title.txt", b"content")
)
response = self.client.post(
reverse("assignment-list"),
{
"title": "test_title_ef3jpF)M329f30m)f82",
"open_posts": 1,
"tags_id": [1],
"attachments_id": [1],
},
)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
assignment = Assignment.objects.get()
self.assertEqual(assignment.title, "test_title_ef3jpF)M329f30m)f82")
self.assertTrue(assignment.tags.exists())
self.assertTrue(assignment.attachments.exists())
class CandidatureSelf(TestCase):
"""
Tests self candidation view.
"""
def setUp(self):
self.client.login(username="admin", password="admin")
self.assignment = Assignment.objects.create(
title="test_assignment_oikaengeijieh3ughiX7", open_posts=1
)
def test_nominate_self(self):
response = self.client.post(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 200)
self.assertTrue(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="admin")
.exists()
)
def test_nominate_self_twice(self):
self.assignment.add_candidate(get_user_model().objects.get(username="admin"))
response = self.client.post(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 200)
self.assertTrue(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="admin")
.exists()
)
def test_nominate_self_when_finished(self):
self.assignment.set_phase(Assignment.PHASE_FINISHED)
self.assignment.save()
response = self.client.post(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 400)
def test_nominate_self_during_voting(self):
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
response = self.client.post(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 200)
self.assertTrue(
Assignment.objects.get(pk=self.assignment.pk).candidates.exists()
)
def test_nominate_self_during_voting_non_admin(self):
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
admin = get_user_model().objects.get(username="admin")
group_admin = admin.groups.get(name="Admin")
group_delegates = type(group_admin).objects.get(name="Delegates")
admin.groups.add(group_delegates)
admin.groups.remove(group_admin)
inform_changed_data(admin)
response = self.client.post(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 403)
def test_withdraw_self(self):
self.assignment.add_candidate(get_user_model().objects.get(username="admin"))
response = self.client.delete(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 200)
self.assertFalse(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="admin")
.exists()
)
def test_withdraw_self_twice(self):
response = self.client.delete(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 400)
def test_withdraw_self_when_finished(self):
self.assignment.add_candidate(get_user_model().objects.get(username="admin"))
self.assignment.set_phase(Assignment.PHASE_FINISHED)
self.assignment.save()
response = self.client.delete(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 400)
def test_withdraw_self_during_voting(self):
self.assignment.add_candidate(get_user_model().objects.get(username="admin"))
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
response = self.client.delete(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 200)
self.assertFalse(
Assignment.objects.get(pk=self.assignment.pk).candidates.exists()
)
def test_withdraw_self_during_voting_non_admin(self):
self.assignment.add_candidate(get_user_model().objects.get(username="admin"))
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
admin = get_user_model().objects.get(username="admin")
group_admin = admin.groups.get(name="Admin")
group_delegates = type(group_admin).objects.get(name="Delegates")
admin.groups.add(group_delegates)
admin.groups.remove(group_admin)
inform_changed_data(admin)
response = self.client.delete(
reverse("assignment-candidature-self", args=[self.assignment.pk])
)
self.assertEqual(response.status_code, 403)
class CandidatureOther(TestCase):
def setUp(self):
self.client = APIClient()
self.client.login(username="admin", password="admin")
self.assignment = Assignment.objects.create(
title="test_assignment_leiD6tiojigh1vei1ait", open_posts=1
)
self.user = get_user_model().objects.create_user(
username="test_user_eeheekai4Phue6cahtho",
password="test_password_ThahXazeiV8veipeePh6",
)
def test_invalid_data_empty_dict(self):
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]), {}
)
self.assertEqual(response.status_code, 400)
def test_invalid_data_string_instead_of_integer(self):
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": "string_instead_of_integer"},
)
self.assertEqual(response.status_code, 400)
def test_invalid_data_user_does_not_exist(self):
# ID of a user that does not exist.
# Be careful: Here we do not test that the user does not exist.
inexistent_user_pk = self.user.pk + 1000
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": inexistent_user_pk},
)
self.assertEqual(response.status_code, 400)
def test_nominate_other(self):
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 200)
self.assertTrue(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="test_user_eeheekai4Phue6cahtho")
.exists()
)
def test_nominate_other_twice(self):
self.assignment.add_candidate(
get_user_model().objects.get(username="test_user_eeheekai4Phue6cahtho")
)
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 400)
def test_nominate_other_when_finished(self):
self.assignment.set_phase(Assignment.PHASE_FINISHED)
self.assignment.save()
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 400)
def test_nominate_other_during_voting(self):
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 200)
self.assertTrue(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="test_user_eeheekai4Phue6cahtho")
.exists()
)
def test_nominate_other_during_voting_non_admin(self):
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
admin = get_user_model().objects.get(username="admin")
group_admin = admin.groups.get(name="Admin")
group_delegates = type(group_admin).objects.get(name="Delegates")
admin.groups.add(group_delegates)
admin.groups.remove(group_admin)
inform_changed_data(admin)
response = self.client.post(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 403)
def test_delete_other(self):
self.assignment.add_candidate(self.user)
response = self.client.delete(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 200)
self.assertFalse(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="test_user_eeheekai4Phue6cahtho")
.exists()
)
def test_delete_other_twice(self):
response = self.client.delete(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 400)
def test_delete_other_when_finished(self):
self.assignment.add_candidate(self.user)
self.assignment.set_phase(Assignment.PHASE_FINISHED)
self.assignment.save()
response = self.client.delete(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 400)
def test_delete_other_during_voting(self):
self.assignment.add_candidate(self.user)
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
response = self.client.delete(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 200)
self.assertFalse(
Assignment.objects.get(pk=self.assignment.pk)
.candidates.filter(username="test_user_eeheekai4Phue6cahtho")
.exists()
)
def test_delete_other_during_voting_non_admin(self):
self.assignment.add_candidate(self.user)
self.assignment.set_phase(Assignment.PHASE_VOTING)
self.assignment.save()
admin = get_user_model().objects.get(username="admin")
group_admin = admin.groups.get(name="Admin")
group_delegates = type(group_admin).objects.get(name="Delegates")
admin.groups.add(group_delegates)
admin.groups.remove(group_admin)
inform_changed_data(admin)
response = self.client.delete(
reverse("assignment-candidature-other", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 403)
class MarkElectedOtherUser(TestCase):
"""
Tests marking an elected user. We use an extra user here to show that
admin can not only mark himself but also other users.
"""
def setUp(self):
self.client = APIClient()
self.client.login(username="admin", password="admin")
self.assignment = Assignment.objects.create(
title="test_assignment_Ierohsh8rahshofiejai", open_posts=1
)
self.user = get_user_model().objects.create_user(
username="test_user_Oonei3rahji5jugh1eev",
password="test_password_aiphahb5Nah0cie4iP7o",
)
def test_mark_elected(self):
self.assignment.add_candidate(
get_user_model().objects.get(username="test_user_Oonei3rahji5jugh1eev")
)
response = self.client.post(
reverse("assignment-mark-elected", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 200)
self.assertTrue(
Assignment.objects.get(pk=self.assignment.pk)
.elected.filter(username="test_user_Oonei3rahji5jugh1eev")
.exists()
)
def test_mark_unelected(self):
user = get_user_model().objects.get(username="test_user_Oonei3rahji5jugh1eev")
self.assignment.set_elected(user)
response = self.client.delete(
reverse("assignment-mark-elected", args=[self.assignment.pk]),
{"user": self.user.pk},
)
self.assertEqual(response.status_code, 200)
self.assertFalse(
Assignment.objects.get(pk=self.assignment.pk)
.elected.filter(username="test_user_Oonei3rahji5jugh1eev")
.exists()
)