refactor(test): bare bones presets for company

This commit is contained in:
ruthra kumar
2025-05-16 15:19:31 +05:30
parent 88c6f61632
commit 0fdc961a4b
3 changed files with 191 additions and 4 deletions

View File

@@ -57,8 +57,20 @@ def clear_demo_data():
def create_demo_company():
company = frappe.db.get_all("Company")[0].name
company_doc = frappe.get_doc("Company", company)
if frappe.flags.in_test:
hash = frappe.generate_hash(length=3)
company_doc = frappe._dict(
{
"company_name": "Test Company" + " " + hash,
"abbr": "TC" + hash,
"default_currency": "INR",
"country": "India",
"chart_of_accounts": "Standard",
}
)
else:
company = frappe.db.get_all("Company")[0].name
company_doc = frappe.get_doc("Company", company).as_dict()
# Make a dummy company
new_company = frappe.new_doc("Company")

View File

@@ -21,6 +21,7 @@ class TestCompany(ERPNextTestSuite):
company.default_currency = "INR"
company.create_chart_of_accounts_based_on = "Existing Company"
company.existing_company = "_Test Company"
company.country = "India"
company.save()
expected_results = {
@@ -67,6 +68,7 @@ class TestCompany(ERPNextTestSuite):
company.default_currency = "USD"
company.create_chart_of_accounts_based_on = "Standard Template"
company.chart_of_accounts = template
company.country = country
company.save()
account_types = [
@@ -108,11 +110,11 @@ class TestCompany(ERPNextTestSuite):
max_rgt = frappe.db.sql("select max(rgt) from `tabCompany`")[0][0]
if not records:
records = self.globalTestRecords["Company"][2:]
records = self.companies[2:]
for company in records:
lft, rgt, parent_company = frappe.db.get_value(
"Company", company["company_name"], ["lft", "rgt", "parent_company"]
"Company", company.get("company_name"), ["lft", "rgt", "parent_company"]
)
if parent_company: