nsinnovations/erpnext
2
0
Fork 0
mirror of https://github.com/frappe/erpnext.git synced 2026-05-25 07:54:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: a more generalized way to clear old entries for test cases

Browse Source 
(cherry picked from commit ed0f9cc063)
This commit is contained in:
ruthra
2022-01-19 15:50:04 +05:30
committed by mergify-bot
parent 43a21f2836
commit 37bbdc6743
1 changed files with 36 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
erpnext/accounts/report/deferred_revenue_and_expense/test_deferred_revenue_and_expense.py
View File

@@ -17,10 +17,42 @@ from erpnext.stock.doctype.item.test_item import create_item
class TestDeferredRevenueAndExpense(unittest.TestCase): class TestDeferredRevenueAndExpense(unittest.TestCase):
@classmethod @classmethod
def setUpClass(self): def setUpClass(self):
clear_old_entries() clear_accounts_and_items()
create_company() create_company()
self.maxDiff = None
def clear_old_entries(self):
sinv = qb.DocType("Sales Invoice")
sinv_item = qb.DocType("Sales Invoice Item")
pinv = qb.DocType("Purchase Invoice")
pinv_item = qb.DocType("Purchase Invoice Item")
# delete existing invoices with deferred items
deferred_invoices = (
qb.from_(sinv)
.join(sinv_item)
.on(sinv.name == sinv_item.parent)
.select(sinv.name)
.where(sinv_item.enable_deferred_revenue == 1)
.run()
)
if deferred_invoices:
qb.from_(sinv).delete().where(sinv.name.isin(deferred_invoices)).run()
deferred_invoices = (
qb.from_(pinv)
.join(pinv_item)
.on(pinv.name == pinv_item.parent)
.select(pinv.name)
.where(pinv_item.enable_deferred_expense == 1)
.run()
)
if deferred_invoices:
qb.from_(pinv).delete().where(pinv.name.isin(deferred_invoices)).run()
def test_deferred_revenue(self): def test_deferred_revenue(self):
self.clear_old_entries()
# created deferred expense accounts, if not found # created deferred expense accounts, if not found
deferred_revenue_account = create_account( deferred_revenue_account = create_account(
account_name="Deferred Revenue", account_name="Deferred Revenue",
@@ -108,6 +140,8 @@ class TestDeferredRevenueAndExpense(unittest.TestCase):
self.assertEqual(report.period_total, expected) self.assertEqual(report.period_total, expected)
def test_deferred_expense(self): def test_deferred_expense(self):
self.clear_old_entries()
# created deferred expense accounts, if not found # created deferred expense accounts, if not found
deferred_expense_account = create_account( deferred_expense_account = create_account(
account_name="Deferred Expense", account_name="Deferred Expense",
@@ -209,15 +243,11 @@ def create_company():
company.insert() company.insert()
def clear_old_entries(): def clear_accounts_and_items():
item = qb.DocType("Item") item = qb.DocType("Item")
account = qb.DocType("Account") account = qb.DocType("Account")
customer = qb.DocType("Customer") customer = qb.DocType("Customer")
supplier = qb.DocType("Supplier") supplier = qb.DocType("Supplier")
sinv = qb.DocType("Sales Invoice")
sinv_item = qb.DocType("Sales Invoice Item")
pinv = qb.DocType("Purchase Invoice")
pinv_item = qb.DocType("Purchase Invoice Item")
qb.from_(account).delete().where( qb.from_(account).delete().where(
(account.account_name == "Deferred Revenue") (account.account_name == "Deferred Revenue")
@@ -228,26 +258,3 @@ def clear_old_entries():
).run() ).run()
qb.from_(customer).delete().where(customer.customer_name == "_Test Customer DR").run() qb.from_(customer).delete().where(customer.customer_name == "_Test Customer DR").run()
qb.from_(supplier).delete().where(supplier.supplier_name == "_Test Furniture Supplier").run() qb.from_(supplier).delete().where(supplier.supplier_name == "_Test Furniture Supplier").run()
# delete existing invoices with deferred items
deferred_invoices = (
qb.from_(sinv)
.join(sinv_item)
.on(sinv.name == sinv_item.parent)
.select(sinv.name)
.where(sinv_item.enable_deferred_revenue == 1)
.run()
)
if deferred_invoices:
qb.from_(sinv).delete().where(sinv.name.isin(deferred_invoices)).run()
deferred_invoices = (
qb.from_(pinv)
.join(pinv_item)
.on(pinv.name == pinv_item.parent)
.select(pinv.name)
.where(pinv_item.enable_deferred_expense == 1)
.run()
)
if deferred_invoices:
qb.from_(pinv).delete().where(pinv.name.isin(deferred_invoices)).run()