from django.test import TestCase
from django.urls import reverse

from wagtail.core.models import Page
from wagtail.tests.utils import WagtailTestUtils


class TestLoginView(TestCase, WagtailTestUtils):
    fixtures = ['test.json']

    def setUp(self):
        self.user = self.create_test_user()
        self.events_index = Page.objects.get(url_path='/home/events/')

    def test_get(self):
        response = self.client.get(reverse('wagtailcore_login'))
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "<h1>Log in</h1>")
        self.assertNotContains(response, "<p>Your username and password didn't match. Please try again.</p>")

    def test_post_incorrect_password(self):
        response = self.client.post(reverse('wagtailcore_login'), {
            'username': 'test@email.com',
            'password': 'wrongpassword',
            'next': self.events_index.url,
        })
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, "<h1>Log in</h1>")
        self.assertContains(response, "<p>Your username and password didn't match. Please try again.</p>")

    def test_post_correct_password(self):
        response = self.client.post(reverse('wagtailcore_login'), {
            'username': 'test@email.com',
            'password': 'password',
            'next': self.events_index.url,
        })
        self.assertRedirects(response, self.events_index.url)