From 70059d1ec03891233bdcb6731990b794dc8135f2 Mon Sep 17 00:00:00 2001 From: ruthra kumar Date: Wed, 28 Jan 2026 17:14:58 +0530 Subject: [PATCH] refactor(test): make irs supplier test deterministic --- .../united_states/test_united_states.py | 27 ++++++++++--------- 1 file changed, 14 insertions(+), 13 deletions(-) diff --git a/erpnext/regional/united_states/test_united_states.py b/erpnext/regional/united_states/test_united_states.py index 08469a7d7ee..2d5ba54c8f9 100644 --- a/erpnext/regional/united_states/test_united_states.py +++ b/erpnext/regional/united_states/test_united_states.py @@ -10,19 +10,12 @@ from erpnext.tests.utils import ERPNextTestSuite class TestUnitedStates(ERPNextTestSuite): def test_irs_1099_custom_field(self): if not frappe.db.exists("Supplier", "_US 1099 Test Supplier"): - doc = frappe.new_doc("Supplier") - doc.supplier_name = "_US 1099 Test Supplier" - doc.supplier_group = "Services" - doc.supplier_type = "Company" - doc.country = "United States" - doc.tax_id = "04-1234567" - doc.irs_1099 = 1 - doc.save() - frappe.db.commit() + make_irs_1099_supplier() supplier = frappe.get_doc("Supplier", "_US 1099 Test Supplier") self.assertEqual(supplier.irs_1099, 1) def test_irs_1099_report(self): + make_irs_1099_supplier() make_payment_entry_to_irs_1099_supplier() filters = frappe._dict({"fiscal_year": "_Test Fiscal Year 2016", "company": "_Test Company 1"}) columns, data = execute_1099_report(filters) @@ -35,11 +28,19 @@ class TestUnitedStates(ERPNextTestSuite): self.assertEqual(data[0], expected_row) -def make_payment_entry_to_irs_1099_supplier(): - frappe.db.sql("delete from `tabGL Entry` where party='_US 1099 Test Supplier'") - frappe.db.sql("delete from `tabGL Entry` where against='_US 1099 Test Supplier'") - frappe.db.sql("delete from `tabPayment Entry` where party='_US 1099 Test Supplier'") +def make_irs_1099_supplier(): + doc = frappe.new_doc("Supplier") + doc.supplier_name = "_US 1099 Test Supplier" + doc.supplier_group = "Services" + doc.supplier_type = "Company" + doc.country = "United States" + doc.tax_id = "04-1234567" + doc.irs_1099 = 1 + doc.save() + return doc + +def make_payment_entry_to_irs_1099_supplier(): pe = frappe.new_doc("Payment Entry") pe.payment_type = "Pay" pe.company = "_Test Company 1"