def test_get_award_financial_csv_sources():
original = VALUE_MAPPINGS["award_financial"]["filter_function"]
VALUE_MAPPINGS["award_financial"]["filter_function"] = MagicMock(returned_value="")
csv_sources = csv_generation.get_csv_sources(
{"download_types": ["award_financial"], "account_level": "treasury_account", "filters": {}}
)
VALUE_MAPPINGS["award_financial"]["filter_function"] = original
assert len(csv_sources) == 1
assert csv_sources[0].file_type == "treasury_account"
assert csv_sources[0].source_type == "award_financial"
def test_get_award_financial_csv_sources():
original = VALUE_MAPPINGS['award_financial']['filter_function']
VALUE_MAPPINGS['award_financial']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["award_financial"],
"account_level": "treasury_account",
"filters": {}
})
VALUE_MAPPINGS['award_financial']['filter_function'] = original
assert len(csv_sources) == 1
assert csv_sources[0].file_type == 'treasury_account'
assert csv_sources[0].source_type == 'award_financial'
def test_get_object_class_program_activity_csv_sources():
original = VALUE_MAPPINGS['object_class_program_activity']['filter_function']
VALUE_MAPPINGS['object_class_program_activity']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["object_class_program_activity"],
"account_level": "treasury_account",
"filters": {}
})
VALUE_MAPPINGS['object_class_program_activity']['filter_function'] = original
assert len(csv_sources) == 1
assert csv_sources[0].file_type == 'treasury_account'
assert csv_sources[0].source_type == 'object_class_program_activity'
def test_get_awards_csv_sources():
original = VALUE_MAPPINGS["awards"]["filter_function"]
VALUE_MAPPINGS["awards"]["filter_function"] = MagicMock(returned_value="")
csv_sources = csv_generation.get_csv_sources(
{"download_types": ["awards"], "filters": {"award_type_codes": list(award_type_mapping.keys())}}
)
assert len(csv_sources) == 2
VALUE_MAPPINGS["awards"]["filter_function"] = original
assert csv_sources[0].file_type == "d1"
assert csv_sources[0].source_type == "awards"
assert csv_sources[1].file_type == "d2"
assert csv_sources[1].source_type == "awards"
def test_get_awards_csv_sources():
original = VALUE_MAPPINGS['awards']['filter_function']
VALUE_MAPPINGS['awards']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["awards"],
"filters": {'award_type_codes': list(award_type_mapping.keys())}
})
assert len(csv_sources) == 2
VALUE_MAPPINGS['awards']['filter_function'] = original
assert csv_sources[0].file_type == 'd1'
assert csv_sources[0].source_type == 'awards'
assert csv_sources[1].file_type == 'd2'
assert csv_sources[1].source_type == 'awards'
def test_idv_orders_csv_sources():
original = VALUE_MAPPINGS["idv_orders"]["filter_function"]
VALUE_MAPPINGS["idv_orders"]["filter_function"] = MagicMock(returned_value="")
csv_sources = csv_generation.get_csv_sources(
{
"download_types": ["idv_orders"],
"filters": {"award_id": 0, "award_type_codes": tuple(set(contract_type_mapping) | set(idv_type_mapping))},
}
)
assert len(csv_sources) == 1
VALUE_MAPPINGS["idv_orders"]["filter_function"] = original
assert csv_sources[0].file_type == "d1"
assert csv_sources[0].source_type == "idv_orders"
def test_get_account_balances_csv_sources():
original = VALUE_MAPPINGS['account_balances']['filter_function']
VALUE_MAPPINGS['account_balances']['filter_function'] = MagicMock(
returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["account_balances"],
"account_level":
"treasury_account",
"filters": {}
})
VALUE_MAPPINGS['account_balances']['filter_function'] = original
assert len(csv_sources) == 1
assert csv_sources[0].file_type == 'treasury_account'
assert csv_sources[0].source_type == 'account_balances'
def test_idv_treasury_account_funding_csv_sources():
original = VALUE_MAPPINGS["idv_federal_account_funding"]["filter_function"]
VALUE_MAPPINGS["idv_federal_account_funding"]["filter_function"] = MagicMock(returned_value="")
csv_sources = csv_generation.get_csv_sources(
{
"download_types": ["idv_federal_account_funding"],
"account_level": "treasury_account",
"filters": {"award_id": 0},
}
)
assert len(csv_sources) == 1
VALUE_MAPPINGS["idv_federal_account_funding"]["filter_function"] = original
assert csv_sources[0].file_type == "treasury_account"
assert csv_sources[0].source_type == "idv_federal_account_funding"
def test_get_awards_csv_sources():
original = VALUE_MAPPINGS['awards']['filter_function']
VALUE_MAPPINGS['awards']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["awards"],
"filters": {
'award_type_codes': list(award_type_mapping.keys())
}
})
assert len(csv_sources) == 2
VALUE_MAPPINGS['awards']['filter_function'] = original
assert csv_sources[0].file_type == 'd1'
assert csv_sources[0].source_type == 'awards'
assert csv_sources[1].file_type == 'd2'
assert csv_sources[1].source_type == 'awards'
def test_idv_treasury_account_funding_csv_sources():
original = VALUE_MAPPINGS['idv_federal_account_funding']['filter_function']
VALUE_MAPPINGS['idv_federal_account_funding'][
'filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["idv_federal_account_funding"],
"account_level":
"treasury_account",
"filters": {
'award_id': 0
}
})
assert len(csv_sources) == 1
VALUE_MAPPINGS['idv_federal_account_funding']['filter_function'] = original
assert csv_sources[0].file_type == 'treasury_account'
assert csv_sources[0].source_type == 'idv_federal_account_funding'
def test_idv_transactions_csv_sources():
original = VALUE_MAPPINGS['idv_transaction_history']['filter_function']
VALUE_MAPPINGS['idv_transaction_history']['filter_function'] = MagicMock(
returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["idv_transaction_history"],
"filters": {
'award_id':
0,
'award_type_codes':
tuple(set(contract_type_mapping) | set(idv_type_mapping))
}
})
assert len(csv_sources) == 1
VALUE_MAPPINGS['idv_transaction_history']['filter_function'] = original
assert csv_sources[0].file_type == 'd1'
assert csv_sources[0].source_type == 'idv_transaction_history'
def test_get_csv_sources():
mappings = csv_generation.VALUE_MAPPINGS
original = mappings['awards']['filter_function']
mappings['awards']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["awards"],
"filters": {
'award_type_codes': list(award_type_mapping.keys())
}
})
mappings['awards']['filter_function'] = original
assert len(csv_sources) == 2
assert csv_sources[0].file_type == 'd1'
assert csv_sources[0].source_type == 'awards'
assert csv_sources[1].file_type == 'd2'
assert csv_sources[1].source_type == 'awards'
original = mappings['transactions']['filter_function']
mappings['transactions']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["transactions"],
"filters": {
'award_type_codes': list(award_type_mapping.keys())
}
})
mappings['transactions']['filter_function'] = original
assert len(csv_sources) == 2
assert csv_sources[0].file_type == 'd1'
assert csv_sources[0].source_type == 'transactions'
assert csv_sources[1].file_type == 'd2'
assert csv_sources[1].source_type == 'transactions'
original = mappings['sub_awards']['filter_function']
mappings['sub_awards']['filter_function'] = MagicMock(returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["sub_awards"],
"filters": {
'award_type_codes': list(award_type_mapping.keys())
}
})
mappings['sub_awards']['filter_function'] = original
assert len(csv_sources) == 2
assert csv_sources[0].file_type == 'd1'
assert csv_sources[0].source_type == 'sub_awards'
assert csv_sources[1].file_type == 'd2'
assert csv_sources[1].source_type == 'sub_awards'
original = mappings['account_balances']['filter_function']
mappings['account_balances']['filter_function'] = MagicMock(
returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["account_balances"],
"account_level":
"treasury_account",
"filters": {}
})
mappings['account_balances']['filter_function'] = original
assert len(csv_sources) == 1
assert csv_sources[0].file_type == 'treasury_account'
assert csv_sources[0].source_type == 'account_balances'
original = mappings['object_class_program_activity']['filter_function']
mappings['object_class_program_activity']['filter_function'] = MagicMock(
returned_value='')
csv_sources = csv_generation.get_csv_sources({
"download_types": ["object_class_program_activity"],
"account_level":
"treasury_account",
"filters": {}
})
mappings['object_class_program_activity']['filter_function'] = original
assert len(csv_sources) == 1
assert csv_sources[0].file_type == 'treasury_account'
assert csv_sources[0].source_type == 'object_class_program_activity'
