def setUp(self): super(FilteringRestrictedItemsTest, self).setUp() # without explicit re-registration of the default policy unit tests # failed when running 'python setup.py test' self._register_default_policy() # get rid of default media Media.query.delete() self.private_media = Media.example(slug=u'private') self.public_media = Media.example(slug=u'public') self.permission_system = MediaDropPermissionSystem(self.pylons_config) self.media_query = Media.query user = self._create_user_without_groups() self.perm = UserPermissions(user, self.permission_system)
def perm(self): system = MediaDropPermissionSystem(self.pylons_config) system.policies = [self.policy] user = DBSession.query(User).filter(User.user_name == u'admin').one() return UserPermissions(user, system)
def viewable_media(query): permission_system = MediaDropPermissionSystem(config) return permission_system.filter_restricted_items(query, u'view', request.perm)