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="******")