def test_find_by_corp_type_and_filing_type_from_new(session):
"""Assert that the fee schedule is saved to the table."""
create_linked_data(FILING_TYPE_CODE, CORP_TYPE_CODE, FEE_CODE)
fee_schedule = services.FeeSchedule()
fee_schedule.filing_type_code = FILING_TYPE_CODE
fee_schedule.corp_type_code = CORP_TYPE_CODE
fee_schedule.fee_code = FEE_CODE
fee_schedule.fee_start_date = date.today()
fee_schedule.save()
fee_schedule = services.FeeSchedule.find_by_corp_type_and_filing_type(
CORP_TYPE_CODE, FILING_TYPE_CODE, None)
assert fee_schedule.fee_schedule_id is not None
assert fee_schedule.fee_start_date == date.today()
assert fee_schedule.fee_end_date is None
assert fee_schedule.fee_code == FEE_CODE
assert fee_schedule.corp_type_code == CORP_TYPE_CODE
assert fee_schedule.filing_type_code == FILING_TYPE_CODE
assert fee_schedule.asdict() == {
'filing_type': 'TEST',
'filing_type_code': FILING_TYPE_CODE,
'filing_fees': 100,
'service_fees': 0,
'total': 100,
'tax': {
'gst': 0,
'pst': 0
},
'priority_fees': 0,
'future_effective_fees': 0,
'processing_fees': 0
}
def test_fee_schedule_with_service_fees_for_basic_user(session):
"""Assert that fee with service fees can be retrieved."""
# Create a transaction fee
tran_fee_code = 'TRAN'
corp_type_code = 'XCORP'
fee_code = 'FEE01'
fee_code_master = FeeCode(code=tran_fee_code, amount=10)
fee_code_master.save()
fee_code_master = FeeCode(code=fee_code, amount=100)
fee_code_master.save()
corp_type = CorpType(code=corp_type_code,
service_fee_code=tran_fee_code,
description='TEST')
corp_type.save()
filing_type = FilingType(code=FILING_TYPE_CODE, description='TEST')
filing_type.save()
fee_schedule = services.FeeSchedule()
fee_schedule.filing_type_code = FILING_TYPE_CODE
fee_schedule.corp_type_code = corp_type_code
fee_schedule.fee_code = fee_code
fee_schedule.fee_start_date = date.today()
fee_schedule.save()
fee_schedule = services.FeeSchedule.find_by_corp_type_and_filing_type(
corp_type=corp_type_code,
filing_type_code=FILING_TYPE_CODE,
valid_date=date.today(),
include_service_fees=False)
assert fee_schedule.service_fees == 10
def test_fee_schedule_saved_from_new(session):
"""Assert that the fee schedule is saved to the table."""
create_linked_data(FILING_TYPE_CODE, CORP_TYPE_CODE, FEE_CODE)
fee_schedule = services.FeeSchedule()
fee_schedule.filing_type_code = FILING_TYPE_CODE
fee_schedule.corp_type_code = CORP_TYPE_CODE
fee_schedule.fee_code = FEE_CODE
fee_schedule.fee_start_date = date.today()
fee_schedule.save()
fee_schedule = services.FeeSchedule.find_by_corp_type_and_filing_type(CORP_TYPE_CODE, FILING_TYPE_CODE, None)
assert fee_schedule is not None
def test_find_by_corp_type_and_filing_type_and_invalid_date(session):
"""Assert that the fee schedule is saved to the table."""
from pay_api.exceptions import BusinessException
create_linked_data(FILING_TYPE_CODE, CORP_TYPE_CODE, FEE_CODE)
fee_schedule = services.FeeSchedule()
fee_schedule.filing_type_code = FILING_TYPE_CODE
fee_schedule.corp_type_code = CORP_TYPE_CODE
fee_schedule.fee_code = FEE_CODE
fee_schedule.fee_start_date = date.today()
fee_schedule.save()
with pytest.raises(BusinessException) as excinfo:
fee_schedule = services.FeeSchedule.find_by_corp_type_and_filing_type(CORP_TYPE_CODE, FILING_TYPE_CODE,
date.today() - timedelta(1))
assert excinfo.value.code == Error.INVALID_CORP_OR_FILING_TYPE.name
