61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
|
from django.test import TestCase
|
||
|
|
from django.urls import reverse
|
||
|
|
from tasks.models import Client, Task
|
||
|
|
from tasks.tests.test_factories import ClientFactory, TaskFactory
|
||
|
|
|
||
|
|
class TaskViewTest(TestCase):
|
||
|
|
def test_task_list_view(self):
|
||
|
|
"""Test that the task list view renders correctly"""
|
||
|
|
# Create some test tasks
|
||
|
|
TaskFactory.create_batch(3)
|
||
|
|
|
||
|
|
url = reverse('task_list')
|
||
|
|
response = self.client.get(url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertTemplateUsed(response, 'tasks/task_list.html')
|
||
|
|
self.assertContains(response, '任务列表')
|
||
|
|
|
||
|
|
def test_task_create_view(self):
|
||
|
|
"""Test that the task create view renders correctly"""
|
||
|
|
url = reverse('task_create')
|
||
|
|
response = self.client.get(url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertTemplateUsed(response, 'tasks/task_create.html')
|
||
|
|
self.assertContains(response, '创建任务')
|
||
|
|
|
||
|
|
def test_task_detail_view(self):
|
||
|
|
"""Test that the task detail view renders correctly"""
|
||
|
|
# Create a test task
|
||
|
|
task = TaskFactory()
|
||
|
|
|
||
|
|
url = reverse('task_detail', args=[task.id])
|
||
|
|
response = self.client.get(url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertTemplateUsed(response, 'tasks/task_detail.html')
|
||
|
|
self.assertContains(response, task.name)
|
||
|
|
|
||
|
|
class ClientViewTest(TestCase):
|
||
|
|
def test_client_list_view(self):
|
||
|
|
"""Test that the client list view renders correctly"""
|
||
|
|
# Create some test clients
|
||
|
|
ClientFactory.create_batch(3)
|
||
|
|
|
||
|
|
url = reverse('client_list')
|
||
|
|
response = self.client.get(url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertTemplateUsed(response, 'tasks/client_list.html')
|
||
|
|
self.assertContains(response, '客户端列表')
|
||
|
|
|
||
|
|
def test_client_create_view(self):
|
||
|
|
"""Test that the client create view renders correctly"""
|
||
|
|
url = reverse('client_create')
|
||
|
|
response = self.client.get(url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertTemplateUsed(response, 'tasks/client_create.html')
|
||
|
|
self.assertContains(response, '创建客户端')
|