def put(task_id):
"""Update a task."""
request_json = request.get_json()
token = g.jwt_oidc_token_info
valid_format, errors = schema_utils.validate(request_json, 'task_request')
if not valid_format:
return {'message': schema_utils.serialize(errors)}, http_status.HTTP_400_BAD_REQUEST
try:
task = TaskService(TaskModel.find_by_task_id(task_id))
if task:
# Update task and its relationships
origin = request.environ.get('HTTP_ORIGIN', 'localhost')
response, status = task.update_task(task_info=request_json,
token_info=token,
origin_url=origin).as_dict(), http_status.HTTP_200_OK
else:
response, status = {'message': 'The requested task could not be found.'}, \
http_status.HTTP_404_NOT_FOUND
except BusinessException as exception:
response, status = {'code': exception.code, 'message': exception.message}, exception.status_code
return response, status
def test_update_task(session, keycloak_mock, monkeypatch): # pylint:disable=unused-argument
"""Assert that a task can be updated."""
user_with_token = TestUserInfo.user_bceid_tester
user_with_token['keycloak_guid'] = TestJwtClaims.public_bceid_user['sub']
user = factory_user_model_with_contact(user_with_token)
patch_token_info(TestJwtClaims.public_bceid_user, monkeypatch)
affidavit_info = TestAffidavit.get_test_affidavit_with_contact()
AffidavitService.create_affidavit(affidavit_info=affidavit_info)
org = OrgService.create_org(TestOrgInfo.org_with_mailing_address(),
user_id=user.id)
org_dict = org.as_dict()
assert org_dict['org_status'] == OrgStatus.PENDING_STAFF_REVIEW.value
token_info = TestJwtClaims.get_test_user(sub=user.keycloak_guid,
source=LoginSource.STAFF.value)
patch_token_info(token_info, monkeypatch)
tasks = TaskService.fetch_tasks(task_status=[TaskStatus.OPEN.value],
page=1,
limit=10)
fetched_tasks = tasks['tasks']
fetched_task = fetched_tasks[0]
task_info = {'relationshipStatus': TaskRelationshipStatus.ACTIVE.value}
task: TaskModel = TaskModel.find_by_task_id(fetched_task['id'])
task = TaskService.update_task(TaskService(task), task_info=task_info)
dictionary = task.as_dict()
user = UserModel.find_by_id(user.id)
assert dictionary['status'] == TaskStatus.COMPLETED.value
assert dictionary[
'relationship_status'] == TaskRelationshipStatus.ACTIVE.value
assert user.verified
def test_create_org_by_verified_bceid_user(session, keycloak_mock, monkeypatch): # pylint:disable=unused-argument
"""Assert that an Org can be created."""
# Steps
# 1. Create a pending affidavit
# 2. Create org
# 3. Approve Org, which will mark the affidavit as approved
# 4. Same user create new org, which should be ACTIVE.
user = factory_user_model_with_contact(user_info=TestUserInfo.user_bceid_tester)
token_info = TestJwtClaims.get_test_user(sub=user.keycloak_guid, source=LoginSource.BCEID.value)
patch_token_info(token_info, monkeypatch)
affidavit_info = TestAffidavit.get_test_affidavit_with_contact()
AffidavitService.create_affidavit(affidavit_info=affidavit_info)
org = OrgService.create_org(TestOrgInfo.org_with_mailing_address(), user_id=user.id)
org_dict = org.as_dict()
assert org_dict['org_status'] == OrgStatus.PENDING_STAFF_REVIEW.value
task_model = TaskModel.find_by_task_for_account(org_dict['id'], status=TaskStatus.OPEN.value)
assert task_model.relationship_id == org_dict['id']
assert task_model.action == TaskAction.AFFIDAVIT_REVIEW.value
task_info = {
'status': TaskStatus.OPEN.value,
'relationshipStatus': TaskRelationshipStatus.ACTIVE.value,
}
TaskService.update_task(TaskService(task_model), task_info)
org_result: OrgModel = OrgModel.find_by_org_id(org_dict['id'])
assert org_result.status_code == OrgStatus.ACTIVE.value
def test_approve_org(session, keycloak_mock, monkeypatch): # pylint:disable=unused-argument
"""Assert that an Affidavit can be approved."""
user = factory_user_model_with_contact(
user_info=TestUserInfo.user_bceid_tester)
token_info = TestJwtClaims.get_test_user(sub=user.keycloak_guid,
source=LoginSource.BCEID.value)
patch_token_info(token_info, monkeypatch)
affidavit_info = TestAffidavit.get_test_affidavit_with_contact()
AffidavitService.create_affidavit(affidavit_info=affidavit_info)
org = OrgService.create_org(TestOrgInfo.org_with_mailing_address(),
user_id=user.id)
org_dict = org.as_dict()
assert org_dict['org_status'] == OrgStatus.PENDING_STAFF_REVIEW.value
task_model = TaskModel.find_by_task_for_account(
org_dict['id'], status=TaskStatus.OPEN.value)
assert task_model.relationship_id == org_dict['id']
assert task_model.action == TaskAction.AFFIDAVIT_REVIEW.value
task_info = {
'status': TaskStatus.OPEN.value,
'relationshipStatus': TaskRelationshipStatus.ACTIVE.value,
'remarks': ['Test Remark']
}
task = TaskService.update_task(TaskService(task_model), task_info)
task_dict = task.as_dict()
affidavit = AffidavitService.find_affidavit_by_org_id(
task_dict['relationship_id'])
assert affidavit['status'] == AffidavitStatus.APPROVED.value
def get(task_id):
"""Fetch task by id."""
try:
task = TaskService(TaskModel.find_by_task_id(task_id=task_id))
response, status = task.as_dict(), http_status.HTTP_200_OK
except BusinessException as exception:
response, status = {'code': exception.code, 'message': exception.message}, exception.status_code
return response, status
def factory_task_service(user_id: int = 1, org_id: int = 1):
"""Produce a templated task service."""
task_model = factory_task_model(user_id, org_id)
service = TaskService(task_model)
return service
