def test_add_fiscal_due_date_when_current_date_is_fiscal():
fiscal = datetime(datetime.now().year, 3, 31, 00, 00, 00)
due_date_type = 'FIS'
period = '12'
expected_due_date = fiscal + relativedelta(months=12)
due_date = MineExpectedDocument._get_due_date_for_expected_document(
None, fiscal, due_date_type, period)
assert due_date == expected_due_date
def test_add_aniversary_due_date():
current_date = datetime.now()
expected_due_date = current_date
due_date_type = 'ANV'
period = '12'
due_date = MineExpectedDocument._get_due_date_for_expected_document(
None, current_date, due_date_type, period)
assert expected_due_date == due_date
def test_add_fiscal_due_date_with_five_year_period_this_year():
current_date = datetime(datetime.now().year, 2, 1, 00, 00, 00)
expected_due_date = datetime(current_date.year + 4, 3, 31, 00, 00, 00)
due_date_type = 'FIS'
period = '60'
due_date = MineExpectedDocument._get_due_date_for_expected_document(
None, current_date, due_date_type, period)
assert due_date == expected_due_date
def test_add_fiscal_due_date_on_new_year():
new_year = datetime(datetime.now().year, 1, 1, 00, 00, 00)
due_date_type = 'FIS'
period = '12'
expected_due_date = datetime(datetime.now().year, 3, 31, 00, 00, 00)
due_date = MineExpectedDocument._get_due_date_for_expected_document(
None, new_year, due_date_type, period)
assert due_date == expected_due_date
def test_add_fiscal_due_date_on_year_end():
end_of_year = datetime(datetime.now().year - 1, 12, 31, 23, 59, 59)
due_date_type = 'FIS'
period = '12'
expected_due_date = datetime(datetime.now().year, 3, 31, 00, 00, 00)
due_date = MineExpectedDocument._get_due_date_for_expected_document(
None, end_of_year, due_date_type, period)
assert due_date == expected_due_date
def test_add_fiscal_due_date_with_one_year_period(test_client):
current_date = datetime(datetime.now().year, 7, 1, 00, 00, 00)
expected_due_date = datetime(current_date.year + 1, 3, 31, 00, 00, 00)
due_date_type = 'FIS'
period = '12'
due_date = MineExpectedDocument._get_due_date_for_expected_document(
None, current_date, due_date_type, period)
assert due_date == expected_due_date
