Merge pull request #119 from alphagov/update_permission_endpoints

Update permission endpoints
This commit is contained in:
Rebecca Law
2016-03-02 14:33:18 +00:00
10 changed files with 252 additions and 134 deletions

View File

@@ -334,7 +334,7 @@ def sample_permission(notify_db,
notify_db_session,
service=None,
user=None,
permission="sample permission"):
permission="manage_service"):
if user is None:
user = sample_user(notify_db, notify_db_session)
data = {
@@ -343,9 +343,14 @@ def sample_permission(notify_db,
}
if service:
data['service'] = service
p_model = Permission(**data)
db.session.add(p_model)
db.session.commit()
p_model = Permission.query.filter_by(
user=user,
service=service,
permission=permission).first()
if not p_model:
p_model = Permission(**data)
db.session.add(p_model)
db.session.commit()
return p_model
@@ -354,7 +359,7 @@ def sample_service_permission(notify_db,
notify_db_session,
service=None,
user=None,
permission="sample permission"):
permission="manage_service"):
if user is None:
user = sample_user(notify_db, notify_db_session)
if service is None:
@@ -364,7 +369,12 @@ def sample_service_permission(notify_db,
'service': service,
'permission': permission
}
p_model = Permission(**data)
db.session.add(p_model)
db.session.commit()
p_model = Permission.query.filter_by(
user=user,
service=service,
permission=permission).first()
if not p_model:
p_model = Permission(**data)
db.session.add(p_model)
db.session.commit()
return p_model