|
1 | 1 | from datetime import date, timedelta
|
2 | 2 |
|
3 | 3 | from django.contrib import admin
|
| 4 | +from django.contrib.auth.models import User |
| 5 | +from django.http import HttpRequest |
4 | 6 | from django.test import TestCase
|
5 | 7 |
|
6 | 8 | from .admin import CorporateMemberAdmin, StatusFilter
|
7 |
| -from .models import CorporateMember |
| 9 | +from .models import CorporateMember, IndividualMember, Team |
8 | 10 |
|
9 | 11 |
|
10 | 12 | class CorporateMemberAdminTests(TestCase):
|
@@ -85,3 +87,37 @@ def get_query_string(self, *args):
|
85 | 87 | {'display': 'All', 'query_string': '', 'selected': False},
|
86 | 88 | ]
|
87 | 89 | )
|
| 90 | + |
| 91 | + |
| 92 | +class IndividualMemberTests(TestCase): |
| 93 | + |
| 94 | + def test_has_change_permission(self): |
| 95 | + user = User.objects.create() |
| 96 | + request = HttpRequest() |
| 97 | + request.user = user |
| 98 | + self.assertIs(admin.site._registry[IndividualMember].has_change_permission(request), False) |
| 99 | + user.is_staff = True |
| 100 | + self.assertIs(admin.site._registry[IndividualMember].has_change_permission(request), True) |
| 101 | + |
| 102 | + def test_has_module_permission(self): |
| 103 | + self.assertEqual( |
| 104 | + admin.site._registry[IndividualMember].has_change_permission, |
| 105 | + admin.site._registry[IndividualMember].has_module_permission |
| 106 | + ) |
| 107 | + |
| 108 | + |
| 109 | +class TeamTests(TestCase): |
| 110 | + |
| 111 | + def test_has_change_permission(self): |
| 112 | + user = User.objects.create() |
| 113 | + request = HttpRequest() |
| 114 | + request.user = user |
| 115 | + self.assertIs(admin.site._registry[Team].has_change_permission(request), False) |
| 116 | + user.is_staff = True |
| 117 | + self.assertIs(admin.site._registry[Team].has_change_permission(request), True) |
| 118 | + |
| 119 | + def test_has_module_permission(self): |
| 120 | + self.assertEqual( |
| 121 | + admin.site._registry[Team].has_change_permission, |
| 122 | + admin.site._registry[Team].has_module_permission |
| 123 | + ) |
0 commit comments