penn-mobile/backend/tests/wrapped/test_models.py at 0a47b80bac977422096e8d04c92e17b766103051 · pennlabs/penn-mobile · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
from datetime import timedelta

from django.contrib.auth import get_user_model
from django.db import IntegrityError
from django.test import TestCase

from wrapped.models import (
    GlobalStat,
    GlobalStatKey,
    GlobalStatPageField,
    IndividualStat,
    IndividualStatKey,
    IndividualStatPageField,
    Page,
    Semester,
)


User = get_user_model()


# Will add more failing tests


class WrappedModelsTestCase(TestCase):
    def setUp(self):
        self.user = User.objects.create_user(username="test", password="test")
        self.semester = Semester.objects.create(semester="2025fa")
        self.semester2 = Semester.objects.create(semester="2025sp")

        self.ind_key = IndividualStatKey.objects.create(key="gsr_hours")
        self.glob_key = GlobalStatKey.objects.create(key="total_gsr_hours")

        self.ind_stat = IndividualStat.objects.create(
            user=self.user,
            key=self.ind_key,
            value="5",
            semester=self.semester,
        )
        self.glob_stat = GlobalStat.objects.create(
            key=self.glob_key,
            value="1000",
            semester=self.semester,
        )

        self.page = Page.objects.create(
            name="GSR_Page",
            id=1,
            template_path="wrapped/gsr_page.html",
            duration=timedelta(minutes=1),
        )
        self.page2 = Page.objects.create(
            name="GSR_Page2",
            id=2,
            template_path="wrapped/gsr_page2.html",
            duration=timedelta(minutes=1),
        )
        self.semester.pages.add(self.page)
        self.semester.pages.add(self.page2)

        # Through models for page fields
        self.ind_field = IndividualStatPageField.objects.create(
            individual_stat_key=self.ind_key,
            page=self.page,
            text_field_name="top",
        )
        self.glob_field = GlobalStatPageField.objects.create(
            global_stat_key=self.glob_key,
            page=self.page,
            text_field_name="bottom",
        )

    def test_str_methods(self):
        self.assertEqual("GSR_Page", str(self.page))
        self.assertEqual("User: test -- gsr_hours-2025fa : 5", str(self.ind_stat))
        self.assertEqual("Global -- total_gsr_hours-2025fa : 1000", str(self.glob_stat))
        self.assertEqual("2025fa", str(self.semester))
        self.assertEqual("GSR_Page -> top : gsr_hours", str(self.ind_field))
        self.assertEqual("GSR_Page -> bottom : total_gsr_hours", str(self.glob_field))

    def test_semester_pages(self):
        self.assertEqual([self.page, self.page2], list(self.semester.pages.all()))

    def test_unique_together_individualstat(self):
        with self.assertRaises(IntegrityError):
            IndividualStat.objects.create(
                user=self.user,
                key=self.ind_key,
                value="6",
                semester=self.semester,
            )

    def test_unique_together_globalstat(self):
        with self.assertRaises(IntegrityError):
            GlobalStat.objects.create(
                key=self.glob_key,
                value="200",
                semester=self.semester,
            )

    def test_page_fields(self):
        self.assertEqual([self.ind_key], list(self.page.individual_stats.all()))
        self.assertEqual([self.glob_key], list(self.page.global_stats.all()))

    def test_stat_page_fields(self):
        self.assertEqual(self.ind_key, self.ind_field.individual_stat_key)
        self.assertEqual(self.glob_key, self.glob_field.global_stat_key)
        self.assertEqual(self.page, self.ind_field.page)
        self.assertEqual(self.page, self.glob_field.page)
        self.assertEqual("top", self.ind_field.text_field_name)
        self.assertEqual("bottom", self.glob_field.text_field_name)

    def test_stat_page_get_value(self):
        self.assertEqual("5", self.ind_field.get_value(self.user, self.semester))
        self.assertEqual("1000", self.glob_field.get_value(self.user, self.semester))

    def updating_semester_non_duplicate(self):
        self.semester.semester = "2025T"
        self.semester.save()
        self.assertEqual(Semester.objects.get(semester="2025T"), self.semester)
        self.assertEqual(Semester.objects.get(semester="2025fa"), None)
        self.assertEqual(len(Semester.objects.all()), 2)
        print(Semester.objects.all())