kpsul/events/tests/test_views.py

from unittest import mock

from django.contrib.auth import get_user_model
from django.contrib.auth.models import Permission
from django.test import Client, TestCase
from django.urls import reverse

from events.models import Event

User = get_user_model()


def make_user(name):
    return User.objects.create_user(username=name, password=name)


def make_staff_user(name):
    view_event_perm = Permission.objects.get_by_natural_key(
        codename="view_event", app_label="events", model="event"
    )
    user = make_user(name)
    user.user_permissions.add(view_event_perm)
    return user


class CSVExportTest(TestCase):
    def setUp(self):
        # Signals handlers on login/logout send messages.
        # Due to the way the Django' test Client performs login, this raise an
        # error. As workaround, we mock the Django' messages module.
        patcher_messages = mock.patch("gestioncof.signals.messages")
        patcher_messages.start()
        self.addCleanup(patcher_messages.stop)

        self.staff = make_staff_user("staff")
        self.u1 = make_user("toto")
        self.u2 = make_user("titi")
        self.event = Event.objects.create(title="test_event", location="somewhere")
        self.event.subscribers.set([self.u1, self.u2])
        self.url = reverse("events:csv-participants", args=[self.event.id])

    def test_get(self):
        client = Client()
        client.force_login(self.staff)
        r = client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_anonymous(self):
        client = Client()
        r = client.get(self.url)
        self.assertRedirects(
            r, "/login?next={}".format(self.url), fetch_redirect_response=False
        )

    def test_unauthorised(self):
        client = Client()
        client.force_login(self.u1)
        r = client.get(self.url)
        self.assertEqual(r.status_code, 403)