Skip to content

Commit 8b81c3f

Browse files
committed
test: add test for registering TaskResultAdmin on proxy model
1 parent 31c0f10 commit 8b81c3f

File tree

1 file changed

+52
-0
lines changed

1 file changed

+52
-0
lines changed

t/unit/test_admin.py

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
import pytest
2+
from celery import uuid
3+
from django.contrib import admin
4+
from django.contrib.auth import get_user_model
5+
from django.test import TestCase
6+
from django.urls import reverse
7+
8+
from django_celery_results.admin import TaskResultAdmin
9+
from django_celery_results.models import TaskResult
10+
11+
User = get_user_model()
12+
13+
14+
@pytest.mark.usefixtures("depends_on_current_app")
15+
class TaskResultAdminTests(TestCase):
16+
app_name = "django_celery_results"
17+
model_name = "taskresult"
18+
19+
def setUp(self):
20+
self.admin_user = User.objects.create_superuser(
21+
username="admin", email="[email protected]", password="password"
22+
)
23+
self.client.login(username="admin", password="password")
24+
self.task_result = TaskResult.objects.create(
25+
task_id=uuid(), task_name="test_task"
26+
)
27+
28+
def test_add_view(self):
29+
url = reverse(f"admin:{self.app_name}_{self.model_name}_add")
30+
response = self.client.get(url)
31+
self.assertEqual(response.status_code, 200)
32+
33+
def test_change_view(self):
34+
url = reverse(
35+
f"admin:{self.app_name}_{self.model_name}_change",
36+
args=[self.task_result.id],
37+
)
38+
response = self.client.get(url)
39+
self.assertEqual(response.status_code, 200)
40+
41+
42+
class TaskResultProxy(TaskResult):
43+
class Meta:
44+
proxy = True
45+
app_label = "django_celery_results"
46+
47+
48+
admin.site.register(TaskResultProxy, TaskResultAdmin)
49+
50+
51+
class TaskResultProxyAdminTests(TaskResultAdminTests):
52+
model_name = "taskresultproxy"

0 commit comments

Comments
 (0)