def test_category_federal_accounts_subawards(
mock_matviews_qs, mock_federal_account, mock_tas, mock_award, mock_financial_account, mock_transaction
):
fa = MockModel(id=10, agency_identifier="020", main_account_code="0001", account_title="Test Federal Account")
add_to_mock_objects(mock_federal_account, [fa])
tas = MockModel(treasury_account_identifier=2, federal_account_id=10)
add_to_mock_objects(mock_tas, [tas])
award = MockModel(id=3)
add_to_mock_objects(mock_award, [award])
fs = MockModel(financial_accounts_by_awards_id=4, submission_id=3, treasury_account=tas, award=award)
add_to_mock_objects(mock_financial_account, [fs])
award.financial_set = MockSet(fs)
t1 = MockModel(award=award, id=5)
t2 = MockModel(award=award, id=6)
add_to_mock_objects(mock_transaction, [t1, t2])
mock_model_1 = MockModel(
transaction=t1, recipient_hash="00000-00000-00000-00000-00000", parent_recipient_unique_id="000000", amount=1
)
mock_model_2 = MockModel(
transaction=t2, recipient_hash="00000-00000-00000-00000-00000", parent_recipient_unique_id="000000", amount=1
)
add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2])
test_payload = {
"category": "federal_account",
"filters": {"recipient_id": "00000-00000-00000-00000-00000-C"},
"subawards": True,
"page": 1,
"limit": 50,
}
spending_by_category_logic = BusinessLogic(test_payload).results()
expected_response = {
"category": "federal_account",
"limit": 50,
"page_metadata": {"page": 1, "next": None, "previous": None, "hasNext": False, "hasPrevious": False},
"results": [{"amount": 2, "code": "020-0001", "name": "Test Federal Account", "id": 10}],
}
assert expected_response == spending_by_category_logic
def test_category_federal_accounts_subawards(mock_matviews_qs,
mock_federal_account, mock_tas,
mock_award,
mock_financial_account,
mock_transaction):
fa = MockModel(id=10,
agency_identifier="020",
main_account_code="0001",
account_title="Test Federal Account")
add_to_mock_objects(mock_federal_account, [fa])
tas = MockModel(treasury_account_identifier=2, federal_account_id=10)
add_to_mock_objects(mock_tas, [tas])
award = MockModel(id=3)
add_to_mock_objects(mock_award, [award])
fs = MockModel(financial_accounts_by_awards_id=4,
submission_id=3,
treasury_account=tas,
award=award)
add_to_mock_objects(mock_financial_account, [fs])
award.financial_set = MockSet(fs)
t1 = MockModel(award=award, id=5)
t2 = MockModel(award=award, id=6)
add_to_mock_objects(mock_transaction, [t1, t2])
mock_model_1 = MockModel(transaction=t1,
recipient_hash="00000-00000-00000-00000-00000",
parent_recipient_unique_id="000000",
amount=1)
mock_model_2 = MockModel(transaction=t2,
recipient_hash="00000-00000-00000-00000-00000",
parent_recipient_unique_id="000000",
amount=1)
add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2])
test_payload = {
"category": "federal_account",
"filters": {
"recipient_id": "00000-00000-00000-00000-00000-C"
},
"subawards": True,
"page": 1,
"limit": 50,
}
spending_by_category_logic = BusinessLogic(test_payload).results()
expected_response = {
"category":
"federal_account",
"limit":
50,
"page_metadata": {
"page": 1,
"next": None,
"previous": None,
"hasNext": False,
"hasPrevious": False
},
"results": [{
"amount": 2,
"code": "020-0001",
"name": "Test Federal Account",
"id": 10
}],
}
assert expected_response == spending_by_category_logic
def test_category_federal_accounts_subawards(mock_matviews_qs,
mock_federal_account, mock_tas,
mock_award,
mock_financial_account,
mock_transaction):
fa = MockModel(id=10,
agency_identifier='020',
main_account_code='0001',
account_title='Test Federal Account')
add_to_mock_objects(mock_federal_account, [fa])
tas = MockModel(treasury_account_identifier=2, federal_account_id=10)
add_to_mock_objects(mock_tas, [tas])
award = MockModel(id=3)
add_to_mock_objects(mock_award, [award])
fs = MockModel(financial_accounts_by_awards_id=4,
submission_id=3,
treasury_account=tas,
award=award)
add_to_mock_objects(mock_financial_account, [fs])
award.financial_set = MockSet(fs)
t1 = MockModel(award=award, id=5)
t2 = MockModel(award=award, id=6)
add_to_mock_objects(mock_transaction, [t1, t2])
mock_model_1 = MockModel(transaction=t1,
recipient_hash='00000-00000-00000-00000-00000',
parent_recipient_unique_id='000000',
amount=1)
mock_model_2 = MockModel(transaction=t2,
recipient_hash='00000-00000-00000-00000-00000',
parent_recipient_unique_id='000000',
amount=1)
add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2])
test_payload = {
'category': 'federal_account',
'filters': {
'recipient_id': '00000-00000-00000-00000-00000-C',
},
'subawards': True,
'page': 1,
'limit': 50
}
spending_by_category_logic = BusinessLogic(test_payload).results()
expected_response = {
'category':
'federal_account',
'limit':
50,
'page_metadata': {
'page': 1,
'next': None,
'previous': None,
'hasNext': False,
'hasPrevious': False
},
'results': [{
'amount': 2,
'code': '020-0001',
'name': 'Test Federal Account',
'id': 10
}]
}
assert expected_response == spending_by_category_logic
