Refs #24121 -- Added __repr__() to PermWrapper.

This commit is contained in:
abhiabhi94 2021-05-27 16:48:50 +05:30 committed by Mariusz Felisiak
parent 214b36f50a
commit 22da686ca9
2 changed files with 10 additions and 0 deletions

View File

@ -25,6 +25,9 @@ class PermWrapper:
def __init__(self, user):
self.user = user
def __repr__(self):
return f'{self.__class__.__qualname__}({self.user!r})'
def __getitem__(self, app_label):
return PermLookupDict(self.user, app_label)

View File

@ -9,6 +9,9 @@ from .settings import AUTH_MIDDLEWARE, AUTH_TEMPLATES
class MockUser:
def __repr__(self):
return 'MockUser()'
def has_module_perms(self, perm):
return perm == 'mockapp'
@ -33,6 +36,10 @@ class PermWrapperTests(SimpleTestCase):
self.eq_calls += 1
return False
def test_repr(self):
perms = PermWrapper(MockUser())
self.assertEqual(repr(perms), 'PermWrapper(MockUser())')
def test_permwrapper_in(self):
"""
'something' in PermWrapper works as expected.