mirror of https://github.com/snachodog/mybuddy.git
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
|
# -*- 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)
|