def get(org_id):
"""Retrieve the set of invitations for the given org."""
try:
invitation_status = request.args.get('status').upper() if request.args.get('status') else None
invitations = InvitationService.get_invitations_for_org(org_id=org_id,
status=invitation_status)
response, status = {'invitations': InvitationSchema().dump(invitations, many=True)}, http_status.HTTP_200_OK
except BusinessException as exception:
response, status = {'code': exception.code, 'message': exception.message}, exception.status_code
return response, status
def test_get_invitations_by_org_id(session, auth_mock, keycloak_mock): # pylint:disable=unused-argument
"""Find an existing invitation with the provided org id."""
with patch.object(InvitationService, 'send_invitation', return_value=None):
user_with_token = TestUserInfo.user_test
user_with_token['keycloak_guid'] = TestJwtClaims.public_user_role['sub']
user = factory_user_model(user_with_token)
org = OrgService.create_org(TestOrgInfo.org1, user_id=user.id)
org_dictionary = org.as_dict()
org_id = org_dictionary['id']
invitation_info = factory_invitation(org_dictionary['id'])
InvitationService.create_invitation(invitation_info, User(user), {}, '').as_dict()
invitations: list = InvitationService.get_invitations_for_org(org_id,
status='PENDING',
token_info=TestJwtClaims.public_user_role)
assert invitations
assert len(invitations) == 1
def test_get_invitations(session, auth_mock): # pylint:disable=unused-argument
"""Assert that invitations for an org can be retrieved."""
with patch.object(InvitationService, 'send_invitation', return_value=None):
user_with_token = TestUserInfo.user_test
user_with_token['keycloak_guid'] = TestJwtClaims.edit_role['sub']
user = factory_user_model(user_info=user_with_token)
org = OrgService.create_org(TestOrgInfo.org1, user.id)
org_dictionary = org.as_dict()
invitation_info = factory_invitation(org_dictionary['id'])
invitation = InvitationService.create_invitation(invitation_info, UserService(user), {}, '')
response = InvitationService.get_invitations_for_org(org_dictionary['id'], 'PENDING', TestJwtClaims.edit_role)
assert response
assert len(response) == 1
assert response[0].recipient_email == invitation.as_dict()['recipientEmail']
