nsinnovations/erpnext
2
0
Fork 0
mirror of https://github.com/frappe/erpnext.git synced 2026-04-14 04:15:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(test): create ERPNext specific test suite

Browse Source 
and move data initialization there
This commit is contained in:
ruthra kumar
2025-05-07 12:08:34 +05:30
parent 4a4e314899
commit 543ba51009
2 changed files with 59 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,72 +3,22 @@
import unittest
import frappe
from frappe.tests import IntegrationTestCase
from frappe.utils import now_datetime, nowdate
from erpnext.accounts.doctype.budget.budget import BudgetError, get_actual_expense
from erpnext.accounts.doctype.journal_entry.test_journal_entry import make_journal_entry
from erpnext.accounts.utils import get_fiscal_year
from erpnext.buying.doctype.purchase_order.test_purchase_order import create_purchase_order
from erpnext.tests.utils import ERPNextTestSuite
class TestBudget(IntegrationTestCase):
class TestBudget(ERPNextTestSuite):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.make_monthly_distribution()
cls.make_projects()
@classmethod
def make_monthly_distribution(cls):
records = [
{
"doctype": "Monthly Distribution",
"distribution_id": "_Test Distribution",
"fiscal_year": "_Test Fiscal Year 2013",
"percentages": [
{"month": "January", "percentage_allocation": "8"},
{"month": "February", "percentage_allocation": "8"},
{"month": "March", "percentage_allocation": "8"},
{"month": "April", "percentage_allocation": "8"},
{"month": "May", "percentage_allocation": "8"},
{"month": "June", "percentage_allocation": "8"},
{"month": "July", "percentage_allocation": "8"},
{"month": "August", "percentage_allocation": "8"},
{"month": "September", "percentage_allocation": "8"},
{"month": "October", "percentage_allocation": "8"},
{"month": "November", "percentage_allocation": "10"},
{"month": "December", "percentage_allocation": "10"},
],
}
]
cls.monthly_distribution = []
for x in records:
if not frappe.db.exists("Monthly Distribution", {"distribution_id": x.get("distribution_id")}):
cls.monthly_distribution.append(frappe.get_doc(x).insert())
else:
cls.monthly_distribution.append(
frappe.get_doc("Monthly Distribution", {"distribution_id": x.get("distribution_id")})
)
@classmethod
def make_projects(cls):
records = [
{
"doctype": "Project",
"company": "_Test Company",
"project_name": "_Test Project",
"status": "Open",
}
]
cls.projects = []
for x in records:
if not frappe.db.exists("Project", {"project_name": x.get("project_name")}):
cls.projects.append(frappe.get_doc(x).insert())
else:
cls.projects.append(frappe.get_doc("Project", {"project_name": x.get("project_name")}))
def test_monthly_budget_crossed_ignore(self):
set_total_expense_zero(nowdate(), "cost_center")

57
erpnext/tests/utils.py
View File

@@ -5,6 +5,7 @@ from typing import Any, NewType
import frappe
from frappe.core.doctype.report.report import get_report_module_dotted_path
from frappe.tests import IntegrationTestCase
ReportFilters = dict[str, Any]
ReportName = NewType("ReportName", str)
@@ -114,3 +115,59 @@ def if_lending_app_not_installed(function):
return
return wrapper
class ERPNextTestSuite(IntegrationTestCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
@classmethod
def make_monthly_distribution(cls):
records = [
{
"doctype": "Monthly Distribution",
"distribution_id": "_Test Distribution",
"fiscal_year": "_Test Fiscal Year 2013",
"percentages": [
{"month": "January", "percentage_allocation": "8"},
{"month": "February", "percentage_allocation": "8"},
{"month": "March", "percentage_allocation": "8"},
{"month": "April", "percentage_allocation": "8"},
{"month": "May", "percentage_allocation": "8"},
{"month": "June", "percentage_allocation": "8"},
{"month": "July", "percentage_allocation": "8"},
{"month": "August", "percentage_allocation": "8"},
{"month": "September", "percentage_allocation": "8"},
{"month": "October", "percentage_allocation": "8"},
{"month": "November", "percentage_allocation": "10"},
{"month": "December", "percentage_allocation": "10"},
],
}
]
cls.monthly_distribution = []
for x in records:
if not frappe.db.exists("Monthly Distribution", {"distribution_id": x.get("distribution_id")}):
cls.monthly_distribution.append(frappe.get_doc(x).insert())
else:
cls.monthly_distribution.append(
frappe.get_doc("Monthly Distribution", {"distribution_id": x.get("distribution_id")})
)
@classmethod
def make_projects(cls):
records = [
{
"doctype": "Project",
"company": "_Test Company",
"project_name": "_Test Project",
"status": "Open",
}
]
cls.projects = []
for x in records:
if not frappe.db.exists("Project", {"project_name": x.get("project_name")}):
cls.projects.append(frappe.get_doc(x).insert())
else:
cls.projects.append(frappe.get_doc("Project", {"project_name": x.get("project_name")}))