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 = MediaCorePermissionSystem(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 = MediaCorePermissionSystem(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 = MediaCorePermissionSystem(config)
return permission_system.filter_restricted_items(query, u'view',
request.perm)
