diff --git a/frappe/tests/ui_test_helpers.py b/frappe/tests/ui_test_helpers.py index aaed05ca3a..4b5a854eb9 100644 --- a/frappe/tests/ui_test_helpers.py +++ b/frappe/tests/ui_test_helpers.py @@ -33,33 +33,19 @@ def create_if_not_exists(doc): @frappe.whitelist() def create_todo_records(): - if frappe.db.get_all('ToDo', {'description': 'this is first todo'}): - return + frappe.db.sql("DELETE FROM `tabToDo` WHERE name!=''") + + for number in range(1, 5): + frappe.get_doc({ + "doctype": "ToDo", + "date": add_to_date(now(), days=3), + "description": "this is {} todo".format(number) + }).insert() - frappe.get_doc({ - "doctype": "ToDo", - "date": add_to_date(now(), days=3), - "description": "this is first todo" - }).insert() - frappe.get_doc({ - "doctype": "ToDo", - "date": add_to_date(now(), days=-3), - "description": "this is second todo" - }).insert() - frappe.get_doc({ - "doctype": "ToDo", - "date": add_to_date(now(), months=2), - "description": "this is third todo" - }).insert() - frappe.get_doc({ - "doctype": "ToDo", - "date": add_to_date(now(), months=-2), - "description": "this is fourth todo" - }).insert() @frappe.whitelist() def setup_workflow(): - from frappe.workflow.doctype.workflow.test_workflow import create_todo_workflow - create_todo_workflow() - create_todo_records() - frappe.clear_cache() + from frappe.workflow.doctype.workflow.test_workflow import create_todo_workflow + create_todo_workflow() + create_todo_records() + frappe.clear_cache()