Shiloh/brotherton-erpnext
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: AttributeError in company transaction deletion

Browse Source 
(cherry picked from commit b127aa308eeda0246cdf885c36f0b36f270e2ae1)
This commit is contained in:
ruthra kumar 2024-01-24 11:42:37 +05:30 committed by Mergify
parent f6a779e1aa
commit 222005ca44
2 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
erpnext/setup/doctype/company/company.py
View File

@ -911,8 +911,8 @@ def generate_id_for_deletion_job(company):
@frappe.whitelist() @frappe.whitelist()
def is_deletion_job_running(company): def is_deletion_job_running(company):
job_id = generate_id_for_deletion_job(company) job_id = generate_id_for_deletion_job(company)
job_name = get_job(job_id).get_id() # job name will have site prefix
if is_job_enqueued(job_id): if is_job_enqueued(job_id):
job_name = get_job(job_id).get_id() # job name will have site prefix
frappe.throw( frappe.throw(
_("A Transaction Deletion Job: {0} is already running for {1}").format( _("A Transaction Deletion Job: {0} is already running for {1}").format(
frappe.bold(get_link_to_form("RQ Job", job_name)), frappe.bold(company) frappe.bold(get_link_to_form("RQ Job", job_name)), frappe.bold(company)

21
erpnext/setup/doctype/transaction_deletion_record/test_transaction_deletion_record.py
View File

@ -4,9 +4,10 @@
import unittest import unittest
import frappe import frappe
from frappe.tests.utils import FrappeTestCase
class TestTransactionDeletionRecord(unittest.TestCase): class TestTransactionDeletionRecord(FrappeTestCase):
def setUp(self): def setUp(self):
create_company("Dunder Mifflin Paper Co") create_company("Dunder Mifflin Paper Co")
@ -14,7 +15,7 @@ class TestTransactionDeletionRecord(unittest.TestCase):
frappe.db.rollback() frappe.db.rollback()
def test_doctypes_contain_company_field(self): def test_doctypes_contain_company_field(self):
tdr = create_transaction_deletion_request("Dunder Mifflin Paper Co") tdr = create_transaction_deletion_doc("Dunder Mifflin Paper Co")
for doctype in tdr.doctypes: for doctype in tdr.doctypes:
contains_company = False contains_company = False
doctype_fields = frappe.get_meta(doctype.doctype_name).as_dict()["fields"] doctype_fields = frappe.get_meta(doctype.doctype_name).as_dict()["fields"]
@ -27,17 +28,27 @@ class TestTransactionDeletionRecord(unittest.TestCase):
def test_no_of_docs_is_correct(self): def test_no_of_docs_is_correct(self):
for i in range(5): for i in range(5):
create_task("Dunder Mifflin Paper Co") create_task("Dunder Mifflin Paper Co")
tdr = create_transaction_deletion_request("Dunder Mifflin Paper Co") tdr = create_transaction_deletion_doc("Dunder Mifflin Paper Co")
for doctype in tdr.doctypes: for doctype in tdr.doctypes:
if doctype.doctype_name == "Task": if doctype.doctype_name == "Task":
self.assertEqual(doctype.no_of_docs, 5) self.assertEqual(doctype.no_of_docs, 5)
def test_deletion_is_successful(self): def test_deletion_is_successful(self):
create_task("Dunder Mifflin Paper Co") create_task("Dunder Mifflin Paper Co")
create_transaction_deletion_request("Dunder Mifflin Paper Co") create_transaction_deletion_doc("Dunder Mifflin Paper Co")
tasks_containing_company = frappe.get_all("Task", filters={"company": "Dunder Mifflin Paper Co"}) tasks_containing_company = frappe.get_all("Task", filters={"company": "Dunder Mifflin Paper Co"})
self.assertEqual(tasks_containing_company, []) self.assertEqual(tasks_containing_company, [])
def test_company_transaction_deletion_request(self):
from erpnext.setup.doctype.company.company import create_transaction_deletion_request
# don't reuse below company for other test cases
company = "Deep Space Exploration"
create_company(company)
# below call should not raise any exceptions or throw errors
create_transaction_deletion_request(company)
def create_company(company_name): def create_company(company_name):
company = frappe.get_doc( company = frappe.get_doc(
@ -46,7 +57,7 @@ def create_company(company_name):
company.insert(ignore_if_duplicate=True) company.insert(ignore_if_duplicate=True)
def create_transaction_deletion_request(company): def create_transaction_deletion_doc(company):
tdr = frappe.get_doc({"doctype": "Transaction Deletion Record", "company": company}) tdr = frappe.get_doc({"doctype": "Transaction Deletion Record", "company": company})
tdr.insert() tdr.insert()
tdr.submit() tdr.submit()