From 74cb029214e61bff1a3c6fda664b810b800b4d8a Mon Sep 17 00:00:00 2001 From: Chneemann Date: Tue, 27 Aug 2024 19:18:47 +0200 Subject: [PATCH] added CustomUserModelTests --- backend/videoflix/users/tests.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/backend/videoflix/users/tests.py b/backend/videoflix/users/tests.py index c7bb4fa..054c30b 100644 --- a/backend/videoflix/users/tests.py +++ b/backend/videoflix/users/tests.py @@ -2,11 +2,32 @@ from django.test import TestCase from django.urls import reverse from rest_framework import status from rest_framework.test import APIClient +from .serializer import LikedVideosSerializer from .models import CustomUser +from content.models import Video from django.contrib.auth import get_user_model +from django.core.files.uploadedfile import SimpleUploadedFile User = get_user_model() +class CustomUserModelTests(TestCase): + def test_generate_verification_token(self): + """ + Ensure the token generated is 20 characters long and alphanumeric. + """ + user = CustomUser(username='testuser') + token = user.generate_verification_token() + self.assertEqual(len(token), 20) + self.assertTrue(token.isalnum()) + + def test_save_without_token_generates_token(self): + """ + Verify a token is assigned if not present when saving a user. + """ + user = CustomUser(username='testuser') + user.save() + self.assertIsNotNone(user.verify_email_token) + class UserTests(TestCase): def setUp(self): """ @@ -73,3 +94,4 @@ class UserTests(TestCase): url = reverse('user_detail', args=[9999]) response = self.client.get(url) self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND) +