mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-01 07:35:34 -05:00
Remove unused fixture and add test function to create permissions
This commit is contained in:
@@ -770,33 +770,6 @@ def sample_invited_org_user(
|
||||
return create_invited_org_user(sample_organisation, sample_user)
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
def sample_permission(notify_db,
|
||||
notify_db_session,
|
||||
service=None,
|
||||
user=None,
|
||||
permission="manage_settings"):
|
||||
if user is None:
|
||||
user = create_user()
|
||||
data = {
|
||||
'user': user,
|
||||
'permission': permission
|
||||
}
|
||||
if service is None:
|
||||
service = create_service(check_if_service_exists=True)
|
||||
if service:
|
||||
data['service'] = service
|
||||
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
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
def sample_user_service_permission(
|
||||
notify_db, notify_db_session, service=None, user=None, permission="manage_settings"
|
||||
|
||||
@@ -13,6 +13,7 @@ from app.dao.notifications_dao import (
|
||||
dao_created_scheduled_notification
|
||||
)
|
||||
from app.dao.organisation_dao import dao_create_organisation
|
||||
from app.dao.permissions_dao import permission_dao
|
||||
from app.dao.service_callback_api_dao import save_service_callback_api
|
||||
from app.dao.service_data_retention_dao import insert_service_data_retention
|
||||
from app.dao.service_inbound_api_dao import save_service_inbound_api
|
||||
@@ -31,6 +32,7 @@ from app.models import (
|
||||
EmailBranding,
|
||||
LetterRate,
|
||||
Organisation,
|
||||
Permission,
|
||||
Rate,
|
||||
Service,
|
||||
ServiceEmailReplyTo,
|
||||
@@ -81,6 +83,15 @@ def create_user(
|
||||
return user
|
||||
|
||||
|
||||
def create_permissions(user, service, *permissions):
|
||||
permissions = [
|
||||
Permission(service_id=service.id, user_id=user.id, permission=p)
|
||||
for p in permissions
|
||||
]
|
||||
|
||||
permission_dao.set_user_service_permission(user, service, permissions)
|
||||
|
||||
|
||||
def create_service(
|
||||
user=None,
|
||||
service_name="Sample service",
|
||||
|
||||
Reference in New Issue
Block a user