1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
"""
Integration tests that actually try and use the tools setup in admin.py
"""
from django.contrib.admin.utils import quote
from django.http import HttpResponse
from django.urls import reverse
from unittest.mock import patch
from .tests import LoggedInTestCase
from example_project.polls.factories import (
CommentFactory,
PollFactory,
RelatedDataFactory,
)
class CommentTests(LoggedInTestCase):
def test_action_on_a_model_with_uuid_pk_works(self):
comment = CommentFactory()
comment_url = reverse("admin:polls_comment_change", args=(comment.pk,))
action_url = "/admin/polls/comment/{0}/actions/hodor/".format(comment.pk)
# sanity check that url has a uuid
self.assertIn("-", action_url)
response = self.client.get(action_url)
self.assertRedirects(response, comment_url)
@patch("django_object_actions.utils.ChangeActionView.dispatch")
def test_action_on_a_model_with_arbitrary_pk_works(self, mock_view):
mock_view.__annotations__ = {}
mock_view.return_value = HttpResponse()
action_url = "/admin/polls/comment/{0}/actions/hodor/".format(" i am a pk ")
self.client.get(action_url)
self.assertTrue(mock_view.called)
self.assertEqual(mock_view.call_args[1]["pk"], " i am a pk ")
@patch("django_object_actions.utils.ChangeActionView.dispatch")
def test_action_on_a_model_with_slash_in_pk_works(self, mock_view):
mock_view.__annotations__ = {}
mock_view.return_value = HttpResponse()
action_url = "/admin/polls/comment/{0}/actions/hodor/".format("pk/slash")
self.client.get(action_url)
self.assertTrue(mock_view.called)
self.assertEqual(mock_view.call_args[1]["pk"], "pk/slash")
class ExtraTests(LoggedInTestCase):
def test_action_on_a_model_with_complex_id(self):
related_data = RelatedDataFactory()
related_data_url = reverse(
"admin:polls_relateddata_change", args=(related_data.pk,)
)
action_url = "/admin/polls/relateddata/{}/actions/fill_up/".format(
quote(related_data.pk)
)
response = self.client.get(action_url)
self.assertNotEqual(response.status_code, 404)
self.assertRedirects(response, related_data_url)
class ChangeTests(LoggedInTestCase):
def test_buttons_load(self):
url = "/admin/polls/choice/"
response = self.client.get(url)
self.assertIn("objectactions", response.context_data)
self.assertIn("Delete all", response.rendered_content)
def test_changelist_template_context(self):
url = reverse("admin:polls_poll_changelist")
response = self.client.get(url)
self.assertIn("objectactions", response.context_data)
self.assertIn("tools_view_name", response.context_data)
self.assertIn("foo", response.context_data)
def test_changelist_action_view(self):
url = reverse("admin:polls_choice_actions", args=("delete_all",))
response = self.client.get(url)
self.assertRedirects(response, "/admin/polls/choice/")
def test_changelist_action_post_only_tool_rejects_get(self):
poll = PollFactory.create()
url = reverse("admin:polls_choice_actions", args=(poll.pk, "reset_vote"))
response = self.client.get(url)
self.assertEqual(response.status_code, 405)
def test_changelist_nonexistent_action(self):
url = "/admin/polls/choice/actions/xyzzy/"
response = self.client.get(url)
self.assertEqual(response.status_code, 404)
def test_get_changelist_can_remove_action(self):
poll = PollFactory.create()
self.assertFalse(poll.question.endswith("?"))
admin_change_url = reverse("admin:polls_poll_change", args=(poll.pk,))
action_url = "/admin/polls/poll/1/actions/question_mark/"
# button is in the admin
response = self.client.get(admin_change_url)
self.assertIn(action_url, response.rendered_content)
response = self.client.get(action_url) # Click on the button
self.assertRedirects(response, admin_change_url)
# button is not in the admin anymore
response = self.client.get(admin_change_url)
self.assertNotIn(action_url, response.rendered_content)
class ChangeListTests(LoggedInTestCase):
def test_changelist_template_context(self):
poll = PollFactory()
url = reverse("admin:polls_poll_change", args=(poll.pk,))
response = self.client.get(url)
self.assertIn("objectactions", response.context_data)
self.assertIn("tools_view_name", response.context_data)
self.assertIn("foo", response.context_data)
class MultipleAdminsTests(LoggedInTestCase):
def test_redirect_back_from_secondary_admin(self):
poll = PollFactory()
admin_change_url = reverse(
"admin:polls_poll_change", args=(poll.pk,), current_app="support"
)
action_url = "/support/polls/poll/1/actions/question_mark/"
self.assertTrue(admin_change_url.startswith("/support/"))
response = self.client.get(action_url)
self.assertRedirects(response, admin_change_url)
|
