nsinnovations/erpnext
2
0
Fork 0
mirror of https://github.com/frappe/erpnext.git synced 2026-05-03 21:48:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: budget revision test cases

Browse Source
This commit is contained in:
khushi8112
2025-10-25 01:01:05 +05:30
parent 077692b57b
commit af9dc8e406
1 changed files with 116 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

117
erpnext/accounts/doctype/budget/test_budget.py
View File

@@ -3,7 +3,7 @@
import unittest
import frappe
from frappe.utils import now_datetime, nowdate
from frappe.utils import add_days, getdate, now_datetime, nowdate
from erpnext.accounts.doctype.budget.budget import (
BudgetError,
@@ -25,6 +25,10 @@ class TestBudget(ERPNextTestSuite):
def setUp(self):
frappe.db.set_single_value("Accounts Settings", "use_legacy_budget_controller", False)
self.company = "_Test Company"
self.fiscal_year = frappe.db.get_value("Fiscal Year", {}, "name")
self.account = "_Test Account Cost for Goods Sold - _TC"
self.cost_center = "_Test Cost Center - _TC"
def test_monthly_budget_crossed_ignore(self):
set_total_expense_zero(nowdate(), "cost_center")
@@ -422,6 +426,117 @@ class TestBudget(ERPNextTestSuite):
po.cancel()
jv.cancel()
def test_distribution_date_validation(self):
budget = frappe.new_doc("Budget")
budget.company = self.company
budget.fiscal_year = self.fiscal_year
budget.budget_against = "Cost Center"
budget.cost_center = self.cost_center
budget.append("accounts", {"account": self.account, "budget_amount": 100000})
start = getdate("2025-04-10")
end = getdate("2025-04-05")
budget.append(
"budget_distribution",
{
"start_date": start,
"end_date": end,
"amount": 50000,
},
)
with self.assertRaises(frappe.ValidationError):
budget.save()
def test_total_distribution_equals_budget(self):
budget = frappe.new_doc("Budget")
budget.company = self.company
budget.fiscal_year = self.fiscal_year
budget.budget_against = "Cost Center"
budget.cost_center = self.cost_center
budget.account = ("_Test Account Cost for Goods Sold - _TC",)
budget.budget_amount = 12000
budget.start_date = getdate("2025-04-01")
budget.end_date = getdate("2025-06-30")
budget.append(
"budget_distribution",
{
"start_date": getdate("2025-04-01"),
"end_date": getdate("2025-04-30"),
"amount": 6000,
},
)
budget.append(
"budget_distribution",
{
"start_date": getdate("2025-05-01"),
"end_date": getdate("2025-06-30"),
"amount": 5000,
},
)
with self.assertRaises(frappe.ValidationError):
budget.save()
def test_evenly_distribute_budget(self):
budget = frappe.new_doc("Budget")
budget.company = self.company
budget.fiscal_year = self.fiscal_year
budget.budget_against = "Cost Center"
budget.cost_center = self.cost_center
budget.distribute_evenly = 1
budget.account = ("_Test Account Cost for Goods Sold - _TC",)
budget.budget_amount = 12000
budget.budget_start_date = getdate("2025-04-01")
budget.budget_end_date = getdate("2026-03-31")
for i in range(12):
budget.append(
"budget_distribution",
{
"start_date": add_days(getdate("2025-04-01"), 30 * i),
"end_date": add_days(getdate("2025-04-30"), 30 * i),
},
)
budget.save()
budget.reload()
total = sum([d.amount for d in budget.budget_distribution])
self.assertEqual(total, 120000)
self.assertTrue(all(d.amount == 10000 for d in budget.budget_distribution))
def test_create_revised_budget(self):
budget = make_budget(budget_against="Cost Center", budget_amount=120000)
revised_name = frappe.get_doc("Budget", budget.name).revise_budget()
revised_budget = frappe.get_doc("Budget", revised_name)
self.assertNotEqual(budget.name, revised_budget.name)
self.assertEqual(revised_budget.budget_against, budget.budget_against)
self.assertEqual(revised_budget.accounts[0].budget_amount, budget.accounts[0].budget_amount)
old_budget = frappe.get_doc("Budget", budget.name)
self.assertEqual(old_budget.docstatus, 2)
def test_revision_preserves_distribution(self):
budget = make_budget(budget_against="Cost Center", budget_amount=120000)
budget.distribute_evenly = 1
budget.allocate_budget()
budget.save()
revised_name = budget.revise_budget()
revised_budget = frappe.get_doc("Budget", revised_name)
self.assertGreater(len(revised_budget.budget_distribution), 0)
total = sum(row.amount for row in revised_budget.budget_distribution)
self.assertEqual(total, revised_budget.accounts[0].budget_amount)
def set_total_expense_zero(posting_date, budget_against_field=None, budget_against_CC=None):
if budget_against_field == "project":