Added more tests

This commit is contained in:
Rebecca Law
2018-02-21 16:39:17 +00:00
parent 011f93c495
commit 70f9dfc0f6
6 changed files with 97 additions and 14 deletions

View File

@@ -1,6 +1,10 @@
from app import db
from app.dao.dao_utils import transactional
from app.models import Organisation, InvitedOrganisationUser, User
from app.models import (
Organisation,
InvitedOrganisationUser,
User
)
def dao_get_organisations():
@@ -55,10 +59,10 @@ def dao_get_users_for_organisation(organisation_id):
).order_by(User.created_at).all()
@transactional
def dao_add_user_to_organisation(organisation_id, user_id):
organisation = dao_get_organisation_by_id(organisation_id)
user = User.query.get(user_id)
user = User.query.filter_by(id=user_id).filter_by(state='active').one()
organisation.users.append(user)
db.session.add(organisation)
db.session.commit()
return user