python-tdd/src/apps/lyric/tests/test_authentication.py

import uuid
from django.http import HttpRequest
from django.test import TestCase

from ..authentication import PasswordlessAuthenticationBackend
from ..models import Token, User


class AuthenticateTest(TestCase):
    def test_returns_None_if_token_is_invalid_uuid(self):
        result = PasswordlessAuthenticationBackend().authenticate(
            HttpRequest(), "no-such-token"
        )
        self.assertIsNone(result)

    def test_returns_None_if_token_uuid_not_found(self):
        uid = uuid.uuid4()
        result = PasswordlessAuthenticationBackend().authenticate(
            HttpRequest(), uid
        )
        self.assertIsNone(result)

    def test_returns_new_user_with_correct_email_if_token_exists(self):
        email = "discoman@example.com"
        token = Token.objects.create(email=email)
        user = PasswordlessAuthenticationBackend().authenticate(
            HttpRequest(), token.uid
        )
        new_user = User.objects.get(email=email)
        self.assertEqual(user, new_user)

    def test_returns_existing_user_with_correct_email_if_token_exists(self):
        email = "discoman@example.com"
        existing_user = User.objects.create(email=email)
        token = Token.objects.create(email=email)
        user = PasswordlessAuthenticationBackend().authenticate(
            HttpRequest(), token.uid
        )
        self.assertEqual(user, existing_user)

    def test_can_retrieve_token_by_uuid(self):
        token = Token.objects.create(email="a@b.cde")
        fetched = Token.objects.get(pk=token.uid)
        self.assertEqual(fetched, token)

class GetUserTest(TestCase):
    def test_gets_user_by_uuid(self):
        User.objects.create(email="fantaman@example.com")
        desired_user = User.objects.create(email="discoman@example.com")
        found_user = PasswordlessAuthenticationBackend().get_user("discoman@example.com")
        self.assertEqual(found_user, desired_user)

    def test_returns_None_if_no_user_with_that_email(self):
        self.assertIsNone(
            PasswordlessAuthenticationBackend().get_user("discoman@example.com")
        )