Remove unused fixture and add test function to create permissions

This commit is contained in:
Katie Smith
2019-05-21 15:51:37 +01:00
parent aab8b99605
commit 844f22bcd0
2 changed files with 11 additions and 27 deletions

View File

@@ -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"

View File

@@ -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",