from django.contrib.auth import get_user_model
from django.shortcuts import reverse
from django.test import TransactionTestCase

User = get_user_model()


class TestRegister(TransactionTestCase):
    def setUp(self) -> None:
        self.password = 'theonering'
        self.email = 'sauron@mordor.middleearth'
        self.first_name = 'Sauron'
        self.last_name = 'The Lord of the Rings'

    def test_get_register_page(self):
        response = self.client.get(reverse('web:register'))
        assert response.status_code == 200

    def test_create_user_account(self):
        response = self.client.post(
            reverse('web:register'),
            {
                'email': self.email,
                'password1': self.password,
                'password2': self.password,
                'first_name': self.first_name,
                'last_name': self.last_name,
            },
        )

        user = User.objects.filter(
            username=self.email, email=self.email, first_name=self.first_name, last_name=self.last_name
        ).first()

        assert user is not None

        assert response.status_code == 302
        assert response.url == reverse('web:login')

    def test_create_user_account_distinct_password(self):
        response = self.client.post(
            reverse('web:register'),
            {
                'email': self.email,
                'password1': self.password,
                'password2': 'otherpassword',
                'first_name': self.first_name,
                'last_name': self.last_name,
            },
        )

        assert not User.objects.filter(
            username=self.email, email=self.email, first_name=self.first_name, last_name=self.last_name
        ).exists()

        assert response.status_code == 400

    def test_create_user_account_no_first_name(self):
        response = self.client.post(
            reverse('web:register'),
            {
                'email': self.email,
                'password1': self.password,
                'password2': 'otherpassword',
                'first_name': '',
                'last_name': self.last_name,
            },
        )

        assert not User.objects.filter(
            username=self.email, email=self.email, first_name='', last_name=self.last_name
        ).exists()

        assert response.status_code == 400

    def test_create_user_account_no_last_name(self):
        response = self.client.post(
            reverse('web:register'),
            {
                'email': self.email,
                'password1': self.password,
                'password2': self.password,
                'first_name': self.first_name,
                'last_name': '',
            },
        )

        assert not User.objects.filter(
            username=self.email, email=self.email, first_name=self.first_name, last_name=''
        ).exists()

        assert response.status_code == 400

    def test_create_user_account_already_existing(self):
        assert User.objects.all().count() == 0
        User.objects.create_user(email=self.email, username=self.email)
        response = self.client.post(
            reverse('web:register'),
            {
                'email': self.email,
                'password1': self.password,
                'password2': self.password,
                'first_name': self.first_name,
                'last_name': self.last_name,
            },
        )

        assert not User.objects.filter(
            username=self.email, email=self.email, first_name=self.first_name, last_name=''
        ).exists()

        assert response.status_code == 400