def test_get_internal_filings(session, client, jwt):
"""Assert that the get_completed_filings_for_colin returns completed filings with no colin ids set."""
from legal_api.models import Filing
from tests.unit.models import factory_completed_filing
# setup
identifier = 'CP7654321'
b = factory_business(identifier)
filing = factory_completed_filing(b, ANNUAL_REPORT)
assert filing.status == Filing.Status.COMPLETED.value
filing.colin_event_id = 1234
filing.save()
filings = Filing.get_completed_filings_for_colin()
# test method
# assert doesn't return completed filing with colin_event_ids set
assert len(filings) == 0
# assert returns completed filings with colin_event_id not set
filing.colin_event_id = None
filing.save()
filings = Filing.get_completed_filings_for_colin()
assert len(filings) == 1
assert filing.id == filings[0].json['filing']['header']['filingId']
assert filings[0].json['filing']['header']['colinId'] is None
# assert doesn't return non completed filings
filing.transaction_id = None
filing.save()
assert filing.status != Filing.Status.COMPLETED.value
filings = Filing.get_completed_filings_for_colin()
assert len(filings) == 0
def get(status=None):
"""Get filings to send to colin."""
filings = []
if status is None:
pending_filings = Filing.get_completed_filings_for_colin()
elif status == Filing.Status.PAID.value:
pending_filings = Filing.get_all_filings_by_status(status)
filings = [x.json for x in pending_filings]
return jsonify(filings), HTTPStatus.OK
def get(status=None):
"""Get filings to send to colin."""
pending_filings = []
if status is None:
pending_filings = Filing.get_completed_filings_for_colin()
elif status == Filing.Status.PAID.value:
pending_filings = Filing.get_all_filings_by_status(status)
filings = []
for filing in pending_filings:
filing_json = filing.filing_json
if filing_json and filing.filing_type != 'lear_epoch':
filing_json['filingId'] = filing.id
filings.append(filing_json)
return jsonify(filings), HTTPStatus.OK
def get(status=None):
"""Get filings by status formatted in json."""
pending_filings = []
filings = []
if status is None:
pending_filings = Filing.get_completed_filings_for_colin()
for filing in pending_filings:
filing_json = filing.filing_json
if filing_json and filing.filing_type != 'lear_epoch':
filing_json['filingId'] = filing.id
filings.append(filing_json)
return jsonify(filings), HTTPStatus.OK
pending_filings = Filing.get_all_filings_by_status(status)
for filing in pending_filings:
filings.append(filing.json)
return jsonify(filings), HTTPStatus.OK
def get(status=None):
"""Get filings by status formatted in json."""
pending_filings = []
filings = []
if status is None:
pending_filings = Filing.get_completed_filings_for_colin()
for filing in pending_filings:
filing_json = filing.filing_json
business = Business.find_by_internal_id(filing.business_id)
if filing_json and filing.filing_type != 'lear_epoch' and \
(filing.filing_type != 'correction' or business.legal_type != business.LegalTypes.COOP.value):
filing_json['filingId'] = filing.id
filing_json['filing']['header'][
'learEffectiveDate'] = filing.effective_date.isoformat(
)
if not filing_json['filing']['business'].get('legalName'):
business = Business.find_by_internal_id(
filing.business_id)
filing_json['filing']['business'][
'legalName'] = business.legal_name
if filing.filing_type == 'correction':
colin_ids = \
ColinEventId.get_by_filing_id(filing_json['filing']['correction']['correctedFilingId'])
if not colin_ids:
continue
filing_json['filing']['correction'][
'correctedFilingColinId'] = colin_ids[
0] # should only be 1
filings.append(filing_json)
return jsonify(filings), HTTPStatus.OK
pending_filings = Filing.get_all_filings_by_status(status)
for filing in pending_filings:
filings.append(filing.json)
return jsonify(filings), HTTPStatus.OK
def get():
"""Get filings to send to colin."""
pending_filings = Filing.get_completed_filings_for_colin()
filings = [x.json for x in pending_filings]
return jsonify(filings), HTTPStatus.OK
