Django authenticated API View的TestCase编写

from django.urls import reverse
from rest_framework.test import APITestCase
from rest_framework import status
from django.contrib.auth.models import User

class AuthenticatedAPIViewTestCase(APITestCase):
    def setUp(self):
        self.user = User.objects.create_user(username='testuser', password='testpassword')
        self.client.login(username='testuser', password='testpassword')
    
    def test_authenticated_view(self):
        url = reverse('your_authenticated_api_view')  # 替换为你的身份验证API视图的URL
        response = self.client.get(url)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        # 进一步可以检查返回的数据是否符合预期
    
    def test_unauthenticated_view(self):
        url = reverse('your_authenticated_api_view')  # 替换为你的身份验证API视图的URL
        self.client.logout()
        response = self.client.get(url)
        self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)