Source code for intranet.apps.eighth.tests.test_activities

import csv
import datetime

from django.contrib.auth import get_user_model
from django.urls import reverse
from django.utils import timezone

from intranet.utils.date import get_senior_graduation_year

from ..models import EighthScheduledActivity, EighthSignup
from ..views.activities import calculate_statistics, generate_statistics_pdf
from .eighth_test import EighthAbstractTest


class EighthActivitiesTestCase(EighthAbstractTest):
    """Test cases for ``views.activities``."""

    def test_past_activities_listed_properly(self):
        self.make_admin()
        activity = self.add_activity(name="Test Activity 1")

        cur_date = timezone.localtime(timezone.now()).date()
        one_day = datetime.timedelta(days=1)

        past_date_str = (cur_date - one_day).strftime("%Y-%m-%d")
        today_date_str = cur_date.strftime("%Y-%m-%d")
        future_date_str = (cur_date + one_day).strftime("%Y-%m-%d")

        block_past = self.add_block(date=past_date_str, block_letter="A")

        schact_past = self.schedule_activity(block_past.id, activity.id)
        response = self.client.get(reverse("eighth_activity", args=[activity.id]))
        self.assertQuerysetEqual(response.context["scheduled_activities"], [])
        response = self.client.get(reverse("eighth_activity", args=[activity.id]), {"show_all": 1})
        self.assertQuerysetEqual(response.context["scheduled_activities"], [repr(schact_past)], transform=repr)

        block_today = self.add_block(date=today_date_str, block_letter="A")
        block_future = self.add_block(date=future_date_str, block_letter="A")

        response = self.client.get(reverse("eighth_activity", args=[activity.id]))
        self.assertQuerysetEqual(response.context["scheduled_activities"], [])
        response = self.client.get(reverse("eighth_activity", args=[activity.id]), {"show_all": 1})
        self.assertQuerysetEqual(response.context["scheduled_activities"], [repr(schact_past)], transform=repr)

        schact_today = self.schedule_activity(block_today.id, activity.id)
        response = self.client.get(reverse("eighth_activity", args=[activity.id]))
        self.assertQuerysetEqual(response.context["scheduled_activities"], [repr(schact_today)], transform=repr)
        response = self.client.get(reverse("eighth_activity", args=[activity.id]), {"show_all": 1})
        self.assertQuerysetEqual(response.context["scheduled_activities"], [repr(schact_past), repr(schact_today)], transform=repr)

        schact_future = self.schedule_activity(block_future.id, activity.id)
        response = self.client.get(reverse("eighth_activity", args=[activity.id]))
        self.assertQuerysetEqual(response.context["scheduled_activities"], [repr(schact_today), repr(schact_future)], transform=repr)
        response = self.client.get(reverse("eighth_activity", args=[activity.id]), {"show_all": 1})
        self.assertQuerysetEqual(
            response.context["scheduled_activities"], [repr(schact_past), repr(schact_today), repr(schact_future)], transform=repr
        )

    def test_stats_global_view(self):
        # I am unauthorized; this should 403
        self.login("awilliam")
        response = self.client.get(reverse("eighth_statistics_global"))
        self.assertEqual(403, response.status_code)

        self.make_admin()

        response = self.client.get(reverse("eighth_statistics_global"))
        self.assertEqual(200, response.status_code)

        # Generate PDF
        response = self.client.post(reverse("eighth_statistics_global"), data={"year": get_senior_graduation_year(), "generate": "pdf"})
        self.assertEqual(200, response.status_code)

        # Generate CSV
        response = self.client.post(reverse("eighth_statistics_global"), data={"year": get_senior_graduation_year(), "generate": "csv"})
        self.assertEqual(200, response.status_code)

        # Add an activity then do it again
        act = self.add_activity(name="Test Activity 1")

        # Generate PDF
        response = self.client.post(reverse("eighth_statistics_global"), data={"year": get_senior_graduation_year(), "generate": "pdf"})
        self.assertEqual(200, response.status_code)

        # Generate CSV
        response = self.client.post(reverse("eighth_statistics_global"), data={"year": get_senior_graduation_year(), "generate": "csv"})
        self.assertEqual(200, response.status_code)

        # Attempt to parse the CSV
        reader = csv.DictReader(response.content.decode(encoding="UTF-8").split("\n"))

        # Loop over all of them, but there should only be one
        for row in reader:
            self.assertEqual(act.name, row["Activity"])

    def test_stats_view(self):
        self.make_admin()
        act = self.add_activity(name="Test Activity 2")

        response = self.client.get(reverse("eighth_statistics", kwargs={"activity_id": act.id}))
        self.assertEqual(200, response.status_code)

        # Add a block and scheduled activity
        today = timezone.localtime().date()
        block = self.add_block(date=today, block_letter="A")

        EighthScheduledActivity.objects.create(activity=act, block=block)

        response = self.client.get(reverse("eighth_statistics", kwargs={"activity_id": act.id}), data={"year": get_senior_graduation_year()})
        self.assertEqual(200, response.status_code)

        response = self.client.get(reverse("eighth_statistics", kwargs={"activity_id": act.id}), data={"print": True})
        self.assertEqual(200, response.status_code)

    def test_generate_statistics_pdf(self):
        self.make_admin()
        act = self.add_activity(name="Test Activity 1")

        today = timezone.localtime().date()
        block = self.add_block(date=today, block_letter="A")

        EighthScheduledActivity.objects.create(activity=act, block=block)

        generate_statistics_pdf(activities=[act])
        # There is no way AFAIK to interpret a PDF file without installing other dependencies.

    def test_calculate_statistics(self):
        self.make_admin()
        act = self.add_activity(name="Test Activity 1")

        stats = calculate_statistics(act)

        expected = {
            "members": [],
            "students": 0,
            "total_blocks": 0,
            "total_signups": 0,
            "average_signups": 0,
            "average_user_signups": 0,
            "old_blocks": 0,
            "cancelled_blocks": 0,
            "scheduled_blocks": 0,
            "empty_blocks": 0,
        }
        subset = {key: value for key, value in stats.items() if key in expected}
        self.assertDictEqual(subset, expected)

    def test_activity_stats(self):
        self.make_admin()
        user = get_user_model().objects.get_or_create(username="user1", graduation_year=get_senior_graduation_year())[0]
        block_a = self.add_block(date="2013-4-20", block_letter="A")
        block_b = self.add_block(date="2013-4-20", block_letter="B")
        act1 = self.add_activity(name="Test1")
        act2 = self.add_activity(name="Test2")
        schact_a = EighthScheduledActivity.objects.create(block=block_a, activity=act1)
        schact_b = EighthScheduledActivity.objects.create(block=block_b, activity=act1)
        EighthSignup.objects.create(scheduled_activity=schact_a, user=user)
        EighthSignup.objects.create(scheduled_activity=schact_b, user=user)

        response = self.client.get(reverse("eighth_statistics_multiple"))
        self.assertEqual(200, response.status_code)

        response = self.client.post(
            reverse("eighth_statistics_multiple"),
            {
                "activities": [act1.id, act2.id],
                "lower": "",
                "upper": "",
                "start": "2020-10-01",
                "end": "2020-10-24",
                "freshmen": "on",
                "sophomores": "on",
                "juniors": "on",
                "seniors": "on",
            },
        )
        self.assertEqual(len(response.context["signed_up"]), 0)
        response = self.client.post(
            reverse("eighth_statistics_multiple"),
            {
                "activities": [act1.id, act2.id],
                "lower": "",
                "upper": "",
                "start": "2013-01-01",
                "end": "2020-10-24",
                "freshmen": "on",
                "sophomores": "on",
                "juniors": "on",
                "seniors": "on",
            },
        )
        self.assertEqual(len(response.context["signed_up"]), 1)
        self.assertEqual(response.context["signed_up"][0]["signups"], 2)