# -*- 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)