def setUp(self):
self.errors = {
field: self.REQUIRED_ERROR
for field in self.required_fields
}
self.errors.update(self.custom_errors)
self.creator = UserFactory(username="******")
self.other_user = UserFactory(username="******")
class PermissionsCheckTestCase(TestCase):
def setUp(self):
self.creator = UserFactory(is_superuser=True,
username='******')
TeamFactory(user=self.creator)
def _login_user(self):
user = UserFactory(username='******')
self.client.force_login(user)
return user
def check_permission_denied(self, method, url):
resp = None
if method == 'GET':
resp = self.client.get(url)
self.assertEqual(resp.status_code, 403)
def check_permission_for_teammate(self, method, url):
resp = None
user = self._login_user()
self.creator.team.teammates.add(user)
self.creator.save()
if method == 'GET':
resp = self.client.get(url)
self.assertEqual(resp.status_code, 200)
def check_user_redirect(self, method, url, redirect_url):
resp = None
if method == 'GET':
resp = self.client.get(url, follow=True)
elif method == 'POST':
resp = self.client.post(url, follow=True)
self.assertRedirects(resp, redirect_url)
def check_user_redirect_all_methods(self, url, redirect_url):
for method in ['GET', 'POST']:
self.check_user_redirect(method, url, redirect_url)
def check_authenticated_user_redirect_all_methods(self, url, redirect_url):
self._login_user()
for method in ['GET', 'POST']:
self.check_user_redirect(method, url, redirect_url)
self.client.logout()
def tearDown(self):
super().tearDown()
try:
os.remove(LOGGING_CSV_FILENAME)
except OSError:
pass
def _login_user(self):
user = UserFactory(username='******')
self.client.force_login(user)
return user
def setUp(self):
self.creator = UserFactory(is_superuser=True,
username='******')
TeamFactory(user=self.creator)
def setUp(self):
self.creator = UserFactory(is_superuser=True, username="******")