def setUp(self):
super(EntitiesApiTestCase, self).setUp()
self.coll = Collection()
self.coll.title = "Test Collection"
session.add(self.coll)
session.flush()
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = True
permission.resource_id = self.coll.id
permission.resource_type = Permission.COLLECTION
session.add(permission)
self.schema_uri = 'https://schema.occrp.org/generic/organization.json#'
self.entity = {'id': 'foo', 'name': 'Foobar'}
self.config.entities.save(self.schema_uri, self.entity,
collection_id=self.coll.id)
cs = CollectionSubject(self.coll, self.entity['id'])
session.add(cs)
entity = {'id': 'test', 'name': 'Testing'}
self.config.entities.save(self.schema_uri, entity,
collection_id=self.coll.id)
cs = CollectionSubject(self.coll, entity['id'])
session.add(cs)
session.commit()
def load_ba_fixtures(config):
# This is messy. Would be cool to do it more cleanly, but how?
if not len(BA_FIXTURES['entities']):
with open(os.path.join(FIXTURES, 'ba.mapping.yaml'), 'rb') as fh:
mapping = yaml.load(fh)
mapper = Mapper(mapping, config.resolver, scope=config.base_uri)
with open(os.path.join(FIXTURES, 'ba.csv'), 'rb') as csvfh:
reader = unicodecsv.DictReader(csvfh)
for row in reader:
_, data = mapper.apply(row)
BA_FIXTURES['entities'].append(data)
source = Source.ensure({
'slug': BA_SOURCE,
'title': 'BiH Parliament',
'url': 'http://foo.ba/'
})
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = False
permission.resource_id = source.id
permission.resource_type = Permission.SOURCE
session.add(permission)
session.commit()
for entity in BA_FIXTURES['entities']:
config.entities.save(entity['$schema'], entity, source_id=source.id)
get_loom_indexer().index(source=BA_SOURCE)
def create_or_update(collection=None, source=None):
if collection is not None:
authz.require(authz.collection(authz.WRITE, collection))
if source is not None:
authz.require(authz.source(authz.WRITE, source))
resource_type = Permission.COLLECTION if collection else Permission.SOURCE
resource_id = collection or source
data = request_data()
validate(data, permissions_schema)
# check that the role exists.
rq = session.query(Role).filter(Role.id == data['role'])
if rq.first() is None:
raise BadRequest()
q = session.query(Permission)
q = q.filter(Permission.role_id == data['role'])
q = q.filter(Permission.resource_type == resource_type)
q = q.filter(Permission.resource_id == resource_id)
permission = q.first()
if permission is None:
permission = Permission()
permission.role_id = data['role']
permission.resource_type = resource_type
permission.resource_id = resource_id
permission.read = data['read']
permission.write = data['write']
session.add(permission)
session.commit()
return jsonify({
'status': 'ok',
'updated': permission
})
def setUp(self):
super(EntitiesApiTestCase, self).setUp()
self.coll = Collection()
self.coll.title = "Test Collection"
session.add(self.coll)
session.flush()
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = True
permission.resource_id = self.coll.id
permission.resource_type = Permission.COLLECTION
session.add(permission)
self.schema_uri = 'https://schema.occrp.org/generic/organization.json#'
self.entity = {'id': 'foo', 'name': 'Foobar'}
self.config.entities.save(self.schema_uri,
self.entity,
collection_id=self.coll.id)
cs = CollectionSubject(self.coll, self.entity['id'])
session.add(cs)
entity = {'id': 'test', 'name': 'Testing'}
self.config.entities.save(self.schema_uri,
entity,
collection_id=self.coll.id)
cs = CollectionSubject(self.coll, entity['id'])
session.add(cs)
session.commit()
def load_ba_fixtures(config):
# This is messy. Would be cool to do it more cleanly, but how?
if not len(BA_FIXTURES['entities']):
with open(os.path.join(FIXTURES, 'ba.mapping.yaml'), 'rb') as fh:
mapping = yaml.load(fh)
mapper = Mapper(mapping, config.resolver, scope=config.base_uri)
with open(os.path.join(FIXTURES, 'ba.csv'), 'rb') as csvfh:
reader = unicodecsv.DictReader(csvfh)
for row in reader:
_, data = mapper.apply(row)
BA_FIXTURES['entities'].append(data)
source = Source.ensure({
'slug': BA_SOURCE,
'title': 'BiH Parliament',
'url': 'http://foo.ba/'
})
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = False
permission.resource_id = source.id
permission.resource_type = Permission.SOURCE
session.add(permission)
session.commit()
for entity in BA_FIXTURES['entities']:
config.entities.save(entity['$schema'], entity, source_id=source.id)
get_loom_indexer().index(source=BA_SOURCE)
def setUp(self):
super(PermissionsApiTestCase, self).setUp()
self.coll = Collection()
self.coll.title = "Test Collection"
session.add(self.coll)
session.flush()
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = True
permission.resource_id = self.coll.id
permission.resource_type = Permission.COLLECTION
session.add(permission)
session.flush()
session.commit()
def setUp(self):
super(PermissionsApiTestCase, self).setUp()
self.coll = Collection()
self.coll.title = "Test Collection"
session.add(self.coll)
session.flush()
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = True
permission.resource_id = self.coll.id
permission.resource_type = Permission.COLLECTION
session.add(permission)
session.flush()
session.commit()
def setUp(self):
super(SourcesApiTestCase, self).setUp()
self.source = Source()
self.source.slug = "test"
self.source.title = "Test Collection"
self.source.url = "http://test.com/"
session.add(self.source)
session.flush()
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = True
permission.resource_id = self.source.id
permission.resource_type = Permission.SOURCE
session.add(permission)
session.commit()
def setUp(self):
super(SourcesApiTestCase, self).setUp()
self.source = Source()
self.source.slug = "test"
self.source.title = "Test Collection"
self.source.url = "http://test.com/"
session.add(self.source)
session.flush()
permission = Permission()
permission.role_id = Role.SYSTEM_USER
permission.read = True
permission.write = True
permission.resource_id = self.source.id
permission.resource_type = Permission.SOURCE
session.add(permission)
session.commit()
def create():
authz.require(authz.logged_in())
data = request_data()
validate(data, collections_schema)
collection = Collection()
collection.title = data.get('title')
session.add(collection)
update_subjects(collection, data)
session.flush()
permission = Permission()
permission.resource_id = collection.id
permission.resource_type = Permission.COLLECTION
permission.read = True
permission.write = True
permission.role_id = request.auth_user
session.add(permission)
session.commit()
return jsonify({'status': 'ok', 'data': collection}, status=201)