def create_roles(self): self.role_vca = CategoryUserRole(user=self.user) self.role_vca.group = self.group_vca self.role_vca.save(sync_role=True) self.role_vca.category.add(self.nested_first_level_two) self.role_vca.save(sync_role=True) self.role_all = CategoryUserRole(user=self.user) self.role_all.group = self.group_all self.role_all.save(sync_role=True) self.role_all.category.add(self.nested_second_level_two) self.role_all.save(sync_role=True)
class UserWithPermissionTestCase(NewmanTestCase): def setUp(self): super(UserWithPermissionTestCase, self).setUp() self.create_permissions() self.create_groups() self.create_roles() def create_permissions(self): self.article_ct = ContentType.objects.get_for_model(Article) for i in ['view']: setattr( self, "permission_%s_article" % i, Permission.objects.create(content_type=self.article_ct, codename='%s_article' % i, name="Can view aritcle")) def create_groups(self): self.group_vca = Group.objects.create( name=u'Permission Group: View, Change, Add') for perm in ["view", "change", "add"]: self.group_vca.permissions.add( Permission.objects.get(content_type=self.article_ct, codename="%s_article" % perm)) self.group_vca.save() self.group_all = Group.objects.create( name=u'Permission Group: Do Whatever Ya Want') for perm in ["view", "change", "add", "delete"]: self.group_all.permissions.add( Permission.objects.get(content_type=self.article_ct, codename="%s_article" % perm)) self.group_all.save() def create_roles(self): self.role_vca = CategoryUserRole(user=self.user) self.role_vca.group = self.group_vca self.role_vca.save(sync_role=True) self.role_vca.category.add(self.nested_first_level_two) self.role_vca.save(sync_role=True) self.role_all = CategoryUserRole(user=self.user) self.role_all.group = self.group_all self.role_all.save(sync_role=True) self.role_all.category.add(self.nested_second_level_two) self.role_all.save(sync_role=True) def _create_author_and_article(self): article = Article.objects.create(title=u'Testable rabbit', slug='testable-rabbit', description=u'Perex', category=self.nested_first_level_two, content=u'Long vehicle') article.authors.add(self.author) article.save() tools.assert_equals(1, Article.objects.count()) return article
class UserWithPermissionTestCase(NewmanTestCase): def setUp(self): super(UserWithPermissionTestCase, self).setUp() self.create_permissions() self.create_groups() self.create_roles() def create_permissions(self): self.article_ct = ContentType.objects.get_for_model(Article) for i in ['view']: setattr(self, "permission_%s_article" % i, Permission.objects.create(content_type=self.article_ct, codename='%s_article' % i, name="Can view aritcle")) def create_groups(self): self.group_vca = Group.objects.create(name=u'Permission Group: View, Change, Add') for perm in ["view", "change", "add"]: self.group_vca.permissions.add(Permission.objects.get(content_type=self.article_ct, codename="%s_article" % perm)) self.group_vca.save() self.group_all = Group.objects.create(name=u'Permission Group: Do Whatever Ya Want') for perm in ["view", "change", "add", "delete"]: self.group_all.permissions.add(Permission.objects.get(content_type=self.article_ct, codename="%s_article" % perm)) self.group_all.save() def create_roles(self): self.role_vca = CategoryUserRole(user=self.user) self.role_vca.group = self.group_vca self.role_vca.save(sync_role=True) self.role_vca.category.add(self.nested_first_level_two) self.role_vca.save(sync_role=True) self.role_all = CategoryUserRole(user=self.user) self.role_all.group = self.group_all self.role_all.save(sync_role=True) self.role_all.category.add(self.nested_second_level_two) self.role_all.save(sync_role=True) def _create_author_and_article(self): article = Article.objects.create(title=u'Testable rabbit', slug='testable-rabbit', description=u'Perex', category=self.nested_first_level_two, content=u'Long vehicle') article.authors.add(self.author) article.save() tools.assert_equals(1, Article.objects.count()) return article