Chneemann/videoflix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

clean code

Browse source
This commit is contained in:
Chneemann 2024-08-27 19:19:50 +02:00
parent 74cb029214
commit 142bed640e
1 changed files with 5 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
backend/videoflix/users/tests.py
View file

@ -2,13 +2,7 @@ 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):
@ -34,9 +28,9 @@ class UserTests(TestCase):
Initializes the APIClient and creates test users. Authenticates a user for the tests.
"""
self.client = APIClient()
self.user = User.objects.create_user(username='testuser', password='testpass')
self.user = CustomUser.objects.create_user(username='testuser', password='testpass')
self.client.force_authenticate(user=self.user)
self.user2 = User.objects.create_user(username='testuser2', password='testpass2')
self.user2 = CustomUser.objects.create_user(username='testuser2', password='testpass2')
def test_get_users(self):
"""
@ -55,8 +49,8 @@ class UserTests(TestCase):
data = {'username': 'testuser3', 'password': 'testpass3'}
response = self.client.post(url, data, format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(User.objects.count(), 3)
self.assertEqual(User.objects.get(username='testuser3').username, 'testuser3')
self.assertEqual(CustomUser.objects.count(), 3)
self.assertEqual(CustomUser.objects.get(username='testuser3').username, 'testuser3')
def test_get_user_detail(self):
"""
@ -85,7 +79,7 @@ class UserTests(TestCase):
url = reverse('user_detail', args=[self.user.pk])
response = self.client.delete(url)
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
self.assertFalse(User.objects.filter(pk=self.user.pk).exists())
self.assertFalse(CustomUser.objects.filter(pk=self.user.pk).exists())
def test_get_user_detail_not_found(self):
"""