darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Adding more unit tests

Browse Source
This commit is contained in:
Rebecca Law
2018-02-21 11:49:21 +00:00
parent 13ef2d7bae
commit 011f93c495
2 changed files with 27 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/dao/organisation_dao.py
View File

@@ -45,14 +45,14 @@ def dao_add_service_to_organisation(service, organisation_id):
def dao_get_invited_organisation_user(user_id):
return InvitedOrganisationUser.query.filter_by(id=user_id).first()
return InvitedOrganisationUser.query.filter_by(id=user_id).one()
def dao_get_users_for_organisation(organisation_id):
return User.query.filter(
User.user_to_organisation.any(id=organisation_id),
User.state == 'active'
).all()
).order_by(User.created_at).all()
def dao_add_user_to_organisation(organisation_id, user_id):