68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
from django.test import TestCase
|
|
from django.urls import reverse
|
|
from rest_framework.test import APITestCase, APIClient
|
|
from rest_framework import status
|
|
|
|
from .routers import router
|
|
|
|
|
|
class RouterTestCase(TestCase):
|
|
def test_router_urls(self):
|
|
"""Test that the router generates the expected URL patterns"""
|
|
url_patterns = router.urls
|
|
|
|
# Check that all expected viewsets are registered
|
|
expected_basenames = ['user', 'torrent', 'file', 'friend-request']
|
|
registered_basenames = [url.name.split('-')[0] for url in url_patterns if '-list' in url.name]
|
|
|
|
for basename in expected_basenames:
|
|
self.assertIn(basename, registered_basenames)
|
|
|
|
# Check that list and detail URLs are generated for each viewset
|
|
for basename in expected_basenames:
|
|
list_url_name = f"{basename}-list"
|
|
detail_url_name = f"{basename}-detail"
|
|
|
|
self.assertTrue(any(url.name == list_url_name for url in url_patterns))
|
|
self.assertTrue(any(url.name == detail_url_name for url in url_patterns))
|
|
|
|
|
|
class APIEndpointsTestCase(APITestCase):
|
|
def setUp(self):
|
|
from user.models import User
|
|
|
|
# Create a test user
|
|
self.user = User.objects.create_user(
|
|
username='testuser',
|
|
email='test@example.com',
|
|
password='testpassword'
|
|
)
|
|
|
|
# Authenticate the client
|
|
self.client = APIClient()
|
|
self.client.force_authenticate(user=self.user)
|
|
|
|
def test_user_endpoint(self):
|
|
"""Test that the users endpoint is accessible"""
|
|
url = reverse('user-list')
|
|
response = self.client.get(url)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
def test_torrent_endpoint(self):
|
|
"""Test that the torrents endpoint is accessible"""
|
|
url = reverse('torrent-list')
|
|
response = self.client.get(url)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
def test_file_endpoint(self):
|
|
"""Test that the files endpoint is accessible"""
|
|
url = reverse('file-list')
|
|
response = self.client.get(url)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
def test_friend_request_endpoint(self):
|
|
"""Test that the friend requests endpoint is accessible"""
|
|
url = reverse('friendrequest-list')
|
|
response = self.client.get(url)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|