diff --git a/erpnext/tests/utils.py b/erpnext/tests/utils.py index fb204c87dca..60a55e1be99 100644 --- a/erpnext/tests/utils.py +++ b/erpnext/tests/utils.py @@ -217,6 +217,7 @@ class ERPNextTestSuite(unittest.TestCase): cls.make_item() cls.make_location() cls.make_price_list() + cls.make_shareholder() cls.update_selling_settings() cls.update_stock_settings() cls.update_system_settings() @@ -2230,6 +2231,27 @@ class ERPNextTestSuite(unittest.TestCase): else: cls.customer.append(frappe.get_doc("Customer", {"customer_name": x.get("customer_name")})) + @classmethod + def make_shareholder(cls): + records = [ + { + "doctype": "Shareholder", + "naming_series": "SH-", + "title": "Iron Man", + "company": "_Test Company", + }, + {"doctype": "Shareholder", "naming_series": "SH-", "title": "Thor", "company": "_Test Company"}, + {"doctype": "Shareholder", "naming_series": "SH-", "title": "Hulk", "company": "_Test Company"}, + ] + cls.shareholder = [] + for x in records: + if not frappe.db.exists("Shareholder", {"title": x.get("title"), "company": x.get("company")}): + cls.shareholder.append(frappe.get_doc(x).insert()) + else: + cls.shareholder.append( + frappe.get_doc("Shareholder", {"title": x.get("title"), "company": x.get("company")}) + ) + @ERPNextTestSuite.registerAs(staticmethod) @contextmanager