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


[docs]class EighthActivitiesTestCase(EighthAbstractTest): """Test cases for ``views.activities``."""
[docs] 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 )
[docs] 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"])
[docs] 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)
[docs] 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.
[docs] 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)
[docs] 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", "sophmores": "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", "sophmores": "on", "juniors": "on", "seniors": "on", }, ) self.assertEqual(len(response.context["signed_up"]), 1) self.assertEqual(response.context["signed_up"][0]["signups"], 2)