refactor(test): make irs supplier test deterministic

This commit is contained in:
ruthra kumar
2026-01-28 17:14:58 +05:30
parent d2da518e02
commit 70059d1ec0

View File

@@ -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"