Add tests for site settings

This commit is contained in:
Christopher C. Wells 2023-05-02 17:33:40 -07:00
parent 0a0582217b
commit 1f3e1ca6db
1 changed files with 73 additions and 0 deletions

View File

@ -0,0 +1,73 @@
# -*- coding: utf-8 -*-
from django.contrib.auth import get_user_model
from django.core.management import call_command
from django.test import Client as HttpClient, TestCase
from faker import Faker
from core.models import Sleep
class SiteSettingsTestCase(TestCase):
@classmethod
def setUpClass(cls):
super(SiteSettingsTestCase, cls).setUpClass()
fake = Faker()
call_command("migrate", verbosity=0)
call_command("fake", verbosity=0)
cls.c = HttpClient()
fake_user = fake.simple_profile()
cls.credentials = {
"username": fake_user["username"],
"password": fake.password(),
}
cls.user = get_user_model().objects.create_user(
is_superuser=True, is_staff=True, **cls.credentials
)
def test_default_settings(self):
self.c.login(**self.credentials)
page = self.c.get("/settings/")
self.assertEqual(page.status_code, 200)
self.assertEqual(
page.context["form"]["core.models__Sleep__nap_start_max"].value(),
"18:00:00",
)
self.assertEqual(
page.context["form"]["core.models__Sleep__nap_start_min"].value(),
"06:00:00",
)
def test_nap_start_settings(self):
self.c.login(**self.credentials)
self.assert_naps()
params = {
"core.models__Sleep__nap_start_max": "20:00:00",
"core.models__Sleep__nap_start_min": "09:00:00",
}
page = self.c.post("/settings/", params, follow=True)
self.assertEqual(page.status_code, 200)
self.assertEqual(
Sleep.settings.nap_start_max.strftime("%H:%M:%S"),
params["core.models__Sleep__nap_start_max"],
)
self.assertEqual(
Sleep.settings.nap_start_min.strftime("%H:%M:%S"),
params["core.models__Sleep__nap_start_min"],
)
self.assert_naps()
def assert_naps(self):
"""
Asserts sleep instances filtered with nap start min and max match nap instances.
"""
instances = Sleep.objects.filter(
start__time__range=(
Sleep.settings.nap_start_min.strftime("%H:%M:%S"),
Sleep.settings.nap_start_max.strftime("%H:%M:%S"),
)
)
naps = Sleep.naps.all()
self.assertQuerySetEqual(instances, naps)