test(fix): Add cleanup for test_doc_rename_method

This commit is contained in:
Gavin D'souza 2022-03-01 11:49:44 +05:30
parent 2b3d9cbcd8
commit 1ff0bbf6c6

View file

@ -48,14 +48,6 @@ class TestRenameDoc(unittest.TestCase):
# data generation: for base and merge tests
self.available_documents = []
self.test_doctype = "ToDo"
self.property_setter = frappe.get_doc({
"doctype":"Property Setter",
"doctype_or_field": "DocType",
"doc_type": self.test_doctype,
"property": "allow_rename",
"property_type": "Check",
"value": "1",
}).insert()
for num in range(1, 5):
doc = frappe.get_doc({
@ -93,7 +85,6 @@ class TestRenameDoc(unittest.TestCase):
# delete the documents created
for docname in self.available_documents:
frappe.delete_doc(self.test_doctype, docname)
self.property_setter.delete()
for dt in self.doctype.values():
if frappe.db.exists("DocType", dt):
@ -107,8 +98,23 @@ class TestRenameDoc(unittest.TestCase):
def setUp(self):
frappe.flags.link_fields = {}
if self._testMethodName == "test_doc_rename_method":
self.property_setter = frappe.get_doc({
"doctype":"Property Setter",
"doctype_or_field": "DocType",
"doc_type": self.test_doctype,
"property": "allow_rename",
"property_type": "Check",
"value": "1",
}).insert()
super().setUp()
def tearDown(self) -> None:
if self._testMethodName == "test_doc_rename_method":
self.property_setter.delete()
return super().tearDown()
def test_rename_doc(self):
"""Rename an existing document via frappe.rename_doc"""
old_name = choice(self.available_documents)
@ -247,9 +253,11 @@ class TestRenameDoc(unittest.TestCase):
update_linked_doctypes("User", "ToDo", "str", "str")
self.assertTrue("Function frappe.model.rename_doc.update_linked_doctypes" in stdout.getvalue())
def test_doc_rename(self):
def test_doc_rename_method(self):
name = choice(self.available_documents)
new_name = f"{name}-{frappe.generate_hash(length=4)}"
doc = frappe.get_doc(self.test_doctype, name)
doc.rename(new_name, merge=frappe.db.exists(self.test_doctype, new_name))
self.assertEqual(doc.name, new_name)
self.available_documents.append(new_name)
self.available_documents.remove(name)