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
|
# -*- coding: utf-8 -*-
import sys
from io import StringIO
from django.apps import apps
from django.core.management import call_command
from django.test import TestCase
class ShowPermissionsTests(TestCase):
def _run_command(self, *args, **kwargs):
"""
Utility to run the command and return captured output.
"""
out = StringIO()
sys_stdout = sys.stdout
sys.stdout = out
try:
call_command("show_permissions", *args, **kwargs)
finally:
sys.stdout = sys_stdout
return out.getvalue()
def _check_header_in_output(self, app_labels, model_verbose, output):
"""
Accepts a list of app label variants and checks if any of them exists with the model_verbose in the output.
"""
for app_label in app_labels:
header = f"Permissions for {app_label} | {model_verbose}"
if header in output:
return
raise AssertionError(
f"None of the expected headers found in output. Tried: {[f'Permissions for {label} | {model_verbose}' for label in app_labels]}"
)
def test_should_list_permissions_for_all_apps_excluding_defaults(self):
output = self._run_command(verbosity=3)
auth_verbose = apps.get_app_config("auth").verbose_name
user_verbose = apps.get_model("auth", "user")._meta.verbose_name
self.assertNotIn(
f"Permissions for {auth_verbose} | {user_verbose}",
output,
"Should not list auth permissions without --all flag",
)
self.assertNotIn("auth.add_user", output)
def test_should_include_all_apps_with_flag(self):
output = self._run_command("--all", verbosity=3)
auth_config = apps.get_app_config("auth")
user_verbose = apps.get_model("auth", "user")._meta.verbose_name
self._check_header_in_output(
[auth_config.verbose_name, auth_config.label], user_verbose, output
)
self.assertIn("auth.add_user", output)
admin_config = apps.get_app_config("admin")
for model in admin_config.get_models():
model_verbose = model._meta.verbose_name
self._check_header_in_output(
[admin_config.verbose_name, admin_config.label], model_verbose, output
)
def test_should_filter_by_app_label(self):
output = self._run_command("--app-label", "auth", verbosity=3)
auth_config = apps.get_app_config("auth")
for model in auth_config.get_models():
model_verbose = model._meta.verbose_name
self._check_header_in_output(
[auth_config.verbose_name, auth_config.label], model_verbose, output
)
self.assertIn("auth.change_user", output)
def test_should_filter_by_app_and_model(self):
output = self._run_command("auth.user", verbosity=3)
auth_config = apps.get_app_config("auth")
user_verbose = apps.get_model("auth", "user")._meta.verbose_name
self._check_header_in_output(
[auth_config.verbose_name, auth_config.label], user_verbose, output
)
self.assertIn("auth.change_user", output)
def test_should_raise_error_for_invalid_model(self):
with self.assertRaisesMessage(
Exception, "Content type not found for 'fakeapp.nosuchmodel'"
):
self._run_command("fakeapp.nosuchmodel", verbosity=3)
def test_should_return_permissions_for_test_model(self):
if apps.is_installed("tests"):
output = self._run_command("tests.samplemodel", verbosity=3)
self.assertIn("tests.samplemodel", output.lower())
self.assertIn("can add", output.lower())
def test_should_raise_error_for_invalid_app_label(self):
with self.assertRaisesMessage(
Exception, 'No content types found for app label "noapp".'
):
self._run_command("--app-label", "noapp", verbosity=3)
|
