def test_get_internal_filings(session, client, jwt):
"""Assert that the internal filings get endpoint returns all completed filings without colin ids."""
from legal_api.models import Filing
from tests.unit.models import factory_error_filing, factory_pending_filing
# setup
identifier = 'CP7654321'
b = factory_business(identifier)
factory_business_mailing_address(b)
filing1 = factory_completed_filing(b, ANNUAL_REPORT)
filing2 = factory_completed_filing(b, ANNUAL_REPORT)
filing3 = factory_pending_filing(b, ANNUAL_REPORT)
filing4 = factory_filing(b, ANNUAL_REPORT)
filing5 = factory_error_filing(b, ANNUAL_REPORT)
assert filing1.status == Filing.Status.COMPLETED.value
# completed with colin_event_id
filing2.colin_event_id = 1234
filing2.save()
assert filing2.status == Filing.Status.COMPLETED.value
assert filing2.colin_event_id is not None
# pending with no colin_event_id
assert filing3.status == Filing.Status.PENDING.value
# draft with no colin_event_id
assert filing4.status == Filing.Status.DRAFT.value
# error with no colin_event_id
assert filing5.status == Filing.Status.ERROR.value
# test endpoint returned filing1 only (completed with no colin id set)
rv = client.get(f'/api/v1/businesses/internal/filings')
assert rv.status_code == HTTPStatus.OK
assert len(rv.json) == 1
assert rv.json[0]['filing']['header']['filingId'] == filing1.id
def test_get_tasks_error_filings(session, client, jwt):
"""Assert that to-do list returns the error filings."""
from legal_api.models import Filing
from tests.unit.models import AR_FILING, factory_business_mailing_address, factory_error_filing
# setup
identifier = 'CP7654321'
b = factory_business(identifier)
factory_business_mailing_address(b)
filing = factory_error_filing(b, AR_FILING)
filing.save()
assert filing.status == Filing.Status.ERROR.value
# test endpoint returned filing in tasks call
rv = client.get(f'/api/v1/businesses/{identifier}/tasks')
assert rv.status_code == HTTPStatus.OK
assert len(rv.json['tasks']) == 1
assert rv.json['tasks'][0]['task']['filing']['header'][
'filingId'] == filing.id
def test_get_internal_filings(session, client, jwt):
"""Assert that the internal filings get endpoint returns all completed filings without colin ids."""
from legal_api.models import Filing
from legal_api.models.colin_event_id import ColinEventId
from tests.unit.models import factory_error_filing, factory_pending_filing
# setup
identifier = 'CP7654321'
b = factory_business(identifier)
factory_business_mailing_address(b)
filing1 = factory_completed_filing(b, ANNUAL_REPORT)
filing2 = factory_completed_filing(b, ANNUAL_REPORT)
filing3 = factory_pending_filing(b, ANNUAL_REPORT)
filing4 = factory_filing(b, ANNUAL_REPORT)
filing5 = factory_error_filing(b, ANNUAL_REPORT)
filing6 = factory_completed_filing(b, CORRECTION_AR)
assert filing1.status == Filing.Status.COMPLETED.value
# completed with colin_event_id
print(filing2.colin_event_ids)
assert len(filing2.colin_event_ids) == 0
colin_event_id = ColinEventId()
colin_event_id.colin_event_id = 12345
filing2.colin_event_ids.append(colin_event_id)
filing2.save()
assert filing2.status == Filing.Status.COMPLETED.value
assert filing2.colin_event_ids
# pending with no colin_event_ids
assert filing3.status == Filing.Status.PENDING.value
# draft with no colin_event_ids
assert filing4.status == Filing.Status.DRAFT.value
# error with no colin_event_ids
assert filing5.status == Filing.Status.PAID.value
# completed correction with no colin_event_ids
assert filing6.status == Filing.Status.COMPLETED.value
# test endpoint returned filing1 only (completed, no corrections, with no colin id set)
rv = client.get(f'/api/v1/businesses/internal/filings')
assert rv.status_code == HTTPStatus.OK
assert len(rv.json) == 1
assert rv.json[0]['filingId'] == filing1.id
